_

main.py → chatbot/plugins/chat.py

@@ -2,27 +2,17 @@ import asyncio
 import logging
 import os
 
-from pyrogram import Client, filters, idle
+from pyrogram import Client, filters
 from pyrogram.types import Message
 
-# Your other imports
-from dotenv import load_dotenv
 from database import db
 from logger import LOGS
-from RyuzakiLib import GeminiLatest  # and other imports as needed
+from RyuzakiLib import GeminiLatest
 import google.generativeai as genai
 from google.api_core.exceptions import InvalidArgument
 
-# Initialize Pyrogram Client
-client = Client(
-    "chatbotai",
-    api_id=API_ID,
-    api_hash=API_HASH,
-    bot_token=BOT_TOKEN
-)
 
-# Define your handler
-@client.on_message(
+@Client.on_message(
     filters.incoming
     & (
         filters.text
@@ -229,13 +219,4 @@ async def chatbot_talk(client: Client, message: Message):
                 backup_chat.append({"role": "model", "parts": [{"text": output}]})
                 await db._update_chatbot_chat_in_db(message.from_user.id, backup_chat)
             except Exception as e:
-                await message.reply_text(str(e))
-
-
-if __name__ == "__main__":
-    try:
-        client.run()
-    except (KeyboardInterrupt, SystemExit):
-        LOGS.info("Bot has been terminated by the user.")
-    except Exception as e:
-        LOGS.error(f"Unexpected error: {e}")
+                await message.reply_text(str(e))