Update main.py

main.py

@@ -37,6 +37,7 @@ from config import *
 
 import google.generativeai as genai
 from google.api_core.exceptions import InvalidArgument
+import database as db
 from logger import LOGS
 
 load_dotenv()
@@ -256,6 +257,7 @@ async def chatbot_talk(client: Client, message: Message):
             return await message.reply_text(str(e))
 
 async def main():
+    await db.connect()
     await client.start()
     me = await client.get_me()
     LOGS.info(f"started bot: {me.first_name}")