Update chatbot/plugins/chat.py

chatbot/plugins/chat.py

@@ -40,6 +40,7 @@ import datetime
 from chatbot import send_log
 from openai import AsyncOpenAI as openai
 import akenoai.openai as akeno
+import akenoai.logger as akeno_log
 
 @Client.on_message(
     ~filters.scheduled
@@ -77,6 +78,7 @@ async def rmchatbot_user(client: Client, message: Message):
     & ~filters.forwarded,
     group=2,
 )
+@akeno_log.log_perfomance
 async def chatbot_talk(client: Client, message: Message):
     chat_user = await db.get_chatbot(message.chat.id)
     if not chat_user:
@@ -85,6 +87,7 @@ async def chatbot_talk(client: Client, message: Message):
         if message.reply_to_message.from_user.id != client.me.id:
             return
     if message.text:
+        pro = await message.reply("Processing your request...", quote=True)
         await client.send_chat_action(message.chat.id, enums.ChatAction.TYPING)
         await asyncio.sleep(1.5)
         query = message.text.strip()
@@ -115,9 +118,10 @@ async def chatbot_talk(client: Client, message: Message):
                     document="chat.txt",
                     disable_notification=True
                 )
+                await pro.delete()
                 os.remove("chat.txt")
             else:
-                await message.reply_text(output)
+                await pro.edit_text(output)
             backup_chat.append({"role": "assistant", "content": output})
             user_detail = (
                 f"**Akeno GPT Bot**\n"
@@ -134,4 +138,4 @@ async def chatbot_talk(client: Client, message: Message):
             await client.send_chat_action(message.chat.id, enums.ChatAction.CANCEL)
             return
         except Exception as e:
-            return await message.reply_text(f"Error: {e}")
+            return await pro.edit_text(f"Error: {e}")