Migrating to Postgres from sqlite3

main.py

@@ -2,20 +2,24 @@ import asyncio
 import logging
 import sys
 
-from aiogram import Bot, Dispatcher
-from aiogram.enums import ParseMode
 from dotenv import load_dotenv
 
 from bot.main_bot import run_bot
 from config.settings import get_settings, Settings
-from db.database import init_db
+from db.database_setup import init_db, init_db_connection
 
 
 async def main():
     load_dotenv()
     settings = get_settings()
 
-    await init_db()
+    session_factory = init_db_connection(settings)
+    if not session_factory:
+        logging.critical(
+            "Failed to initialize DB connection and session factory. Exiting.")
+        return
+
+    await init_db(settings, session_factory)
 
     await run_bot(settings)
 
@@ -29,3 +33,7 @@ if __name__ == "__main__":
         asyncio.run(main())
     except (KeyboardInterrupt, SystemExit):
         logging.info("Bot stopped manually")
+    except Exception as e_global:
+        logging.critical(f"Global unhandled exception in main: {e_global}",
+                         exc_info=True)
+        sys.exit(1)