Update chatbot/plugins/chat.py

chatbot/plugins/chat.py

@@ -31,22 +31,15 @@ from google.api_core.exceptions import InvalidArgument
 async def chatbot_talk(client: Client, message: Message):
     try:
         genai.configure(api_key=GOOGLE_API_KEY)
-
-        # Handling Photo Messages
         if message.photo:
             try:
                 file_path = await message.download()
                 caption = message.caption or "What's this?"
                 x = GeminiLatest(api_keys=GOOGLE_API_KEY)
-                
-                # Send initial processing message
                 ai_reply = await message.reply_text("Processing...")
-                
                 backup_chat = await db._get_chatbot_chat_from_db(message.from_user.id)
                 backup_chat.append({"role": "user", "parts": [{"text": caption}]})
-                
                 response_reads = x.get_response_image(caption, file_path)
-                
                 if len(response_reads) > 4096:
                     with open("chat.txt", "w+", encoding="utf8") as out_file:
                         out_file.write(response_reads)
@@ -58,10 +51,8 @@ async def chatbot_talk(client: Client, message: Message):
                     os.remove("chat.txt")
                 else:
                     await ai_reply.edit_text(response_reads)
-                
                 backup_chat.append({"role": "model", "parts": [{"text": response_reads}]})
                 await db._update_chatbot_chat_in_db(message.from_user.id, backup_chat)
-                
                 os.remove(file_path)
             except InvalidArgument as e:
                 await ai_reply.edit_text(f"Error: {e}")
@@ -69,7 +60,6 @@ async def chatbot_talk(client: Client, message: Message):
                 await ai_reply.edit_text(f"Error: {e}")
             return
 
-        # Handling Audio or Voice Messages
         if message.audio or message.voice:
             try:
                 ai_reply = await message.reply_text("Processing...")
@@ -86,18 +76,14 @@ async def chatbot_talk(client: Client, message: Message):
                 )
                 backup_chat = await db._get_chatbot_chat_from_db(message.from_user.id)
                 backup_chat.append({"role": "user", "parts": [{"text": caption}]})
-                
                 await ai_reply.edit_text("Uploading file...")
                 audio_file = genai.upload_file(path=audio_file_name)
-                
                 while audio_file.state.name == "PROCESSING":
                     await asyncio.sleep(10)
                     audio_file = genai.get_file(audio_file.name)
-                
                 if audio_file.state.name == "FAILED":
                     await ai_reply.edit_text(f"Error: {audio_file.state.name}")
                     return
-                
                 response = model.generate_content(
                     [audio_file, caption],
                     request_options={"timeout": 600}
@@ -125,7 +111,6 @@ async def chatbot_talk(client: Client, message: Message):
                 await ai_reply.edit_text(f"Error: {e}")
             return
 
-        # Handling Video Messages
         if message.video:
             try:
                 ai_reply = await message.reply_text("Processing...")
@@ -142,14 +127,11 @@ async def chatbot_talk(client: Client, message: Message):
                 )
                 backup_chat = await db._get_chatbot_chat_from_db(message.from_user.id)
                 backup_chat.append({"role": "user", "parts": [{"text": caption}]})
-                
                 await ai_reply.edit_text("Uploading file...")
                 video_file = genai.upload_file(path=video_file_name)
-                
                 while video_file.state.name == "PROCESSING":
                     await asyncio.sleep(10)
                     video_file = genai.get_file(video_file.name)
-                
                 if video_file.state.name == "FAILED":
                     await ai_reply.edit_text(f"Error: {video_file.state.name}")
                     return
@@ -172,7 +154,6 @@ async def chatbot_talk(client: Client, message: Message):
                 
                 backup_chat.append({"role": "model", "parts": [{"text": response.text}]})
                 await db._update_chatbot_chat_in_db(message.from_user.id, backup_chat)
-                
                 video_file.delete()
                 os.remove(video_file_name)
             except InvalidArgument as e:
@@ -181,7 +162,6 @@ async def chatbot_talk(client: Client, message: Message):
                 await ai_reply.edit_text(f"Error: {e}")
             return
 
-        # Handling Text Messages
         if message.text:
             try:
                 query = message.text.strip()
@@ -201,11 +181,9 @@ async def chatbot_talk(client: Client, message: Message):
                 )
                 backup_chat = await db._get_chatbot_chat_from_db(message.from_user.id)
                 backup_chat.append({"role": "user", "parts": [{"text": query_base}]})
-                
                 chat_session = model_flash.start_chat(history=backup_chat)
                 response_data = chat_session.send_message(query_base)
                 output = response_data.text
-                
                 if len(output) > 4096:
                     with open("chat.txt", "w+", encoding="utf8") as out_file:
                         out_file.write(output)
@@ -216,8 +194,7 @@ async def chatbot_talk(client: Client, message: Message):
                     os.remove("chat.txt")
                 else:
                     await message.reply_text(output)
-                
                 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))
+                return await message.reply_text(str(e))