Update chatbot/plugins/chat.py

chatbot/plugins/chat.py

@@ -27,6 +27,7 @@ import re
 from PIL import Image
 
 from pyrogram import *
+from pyrogram import enums
 from pyrogram import Client, filters
 from pyrogram.types import *
 from pyrogram.errors import *
@@ -53,7 +54,7 @@ USER_TRACK = """
 
 UserID: {user_id}
 First Name: {first_name}
-Username: @{username}
+Username: `{username}`
 """
 
 @Client.on_message(
@@ -75,6 +76,8 @@ Username: @{username}
 async def chatbot_talk(client: Client, message: Message):
     genai.configure(api_key=GOOGLE_API_KEY)
     if message.photo:
+        await client.send_chat_action(message.chat.id, enums.ChatAction.TYPING)
+        await asyncio.sleep(1.5)
         file_path = await message.download()
         caption = message.caption or "What's this?"
         x = GeminiLatest(api_keys=GOOGLE_API_KEY)
@@ -107,6 +110,7 @@ async def chatbot_talk(client: Client, message: Message):
                 )
             backup_chat.append({"role": "model", "parts": [{"text": response_reads}]})
             await db._update_chatbot_chat_in_db(message.from_user.id, backup_chat)
+            await client.send_chat_action(message.chat.id, enums.ChatAction.CANCEL)
             os.remove(file_path)
             return
         except InvalidArgument as e:
@@ -115,6 +119,8 @@ async def chatbot_talk(client: Client, message: Message):
             return await ai_reply.edit_text(f"Error: {e}")
 
     if message.audio or message.voice:
+        await client.send_chat_action(message.chat.id, enums.ChatAction.TYPING)
+        await asyncio.sleep(1.5)
         if client.me.is_premium:
             ai_reply = await message.reply_text(f"{custom_loading}Processing...")
         else:
@@ -171,6 +177,7 @@ 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)
+            await client.send_chat_action(message.chat.id, enums.ChatAction.CANCEL)
             audio_file.delete()
             os.remove(audio_file_name)
             return
@@ -180,6 +187,8 @@ async def chatbot_talk(client: Client, message: Message):
             return await ai_reply.edit_text(f"Error: {e}")
 
     if message.video:
+        await client.send_chat_action(message.chat.id, enums.ChatAction.TYPING)
+        await asyncio.sleep(1.5)
         if client.me.is_premium:
             ai_reply = await message.reply_text(f"{custom_loading}Processing...")
         else:
@@ -233,6 +242,7 @@ 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)
+            await client.send_chat_action(message.chat.id, enums.ChatAction.CANCEL)
             video_file.delete()
             os.remove(video_file_name)
             return
@@ -242,6 +252,8 @@ async def chatbot_talk(client: Client, message: Message):
             return await ai_reply.edit_text(f"Error: {e}")
 
     if message.text:
+        await client.send_chat_action(message.chat.id, enums.ChatAction.TYPING)
+        await asyncio.sleep(1.5)
         query = message.text.strip()
         match = re.search(r"\b(Randy|Rendi)\b(.*)", query, flags=re.IGNORECASE)
         if match:
@@ -281,5 +293,6 @@ 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)
+            await client.send_chat_action(message.chat.id, enums.ChatAction.CANCEL)
         except Exception as e:
             return await message.reply_text(str(e))