Update chatbot/plugins/chat.py

chatbot/plugins/chat.py

@@ -117,6 +117,60 @@ async def rmchatbot_user(client: Client, message: Message):
     await db.remove_chatbot(message.chat.id)
     await message.reply_text("ok stopped GPT")
 
+@Client.on_message(
+    ~filters.scheduled
+    & filters.command(["ask"])
+    & ~filters.forwarded
+)
+async def rmchatbot_user(client: Client, message: Message):
+   chat_user = await db.get_chatbot(message.chat.id)
+    if not chat_user:
+        return
+    query = message.text.split(" ", 1)[1] if len(message.command) > 1 else None 
+
+    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)
+    try:
+        backup_chat = await db._get_openai_chat_from_db(message.from_user.id)
+        backup_chat.append({"role": "system", "content": BASE_PROMPT})
+        backup_chat.append({"role": "user", "content": query_base})
+        response = await akeno.OpenAI.run(
+            ...,
+            openai_meta=openai,
+            model="gpt-4o-mini-2024-07-18",
+            messages=backup_chat
+        )
+        output = response
+        if len(output) > 4096:
+            with open("chat.txt", "w+", encoding="utf8") as out_file:
+                out_file.write(output)
+            await message.reply_document(
+                document="chat.txt",
+                disable_notification=True
+            )
+            await pro.delete()
+            os.remove("chat.txt")
+        else:
+            await pro.edit_text(output, disable_web_page_preview=True)
+        backup_chat.append({"role": "assistant", "content": output})
+        user_detail = (
+            f"**Akeno GPT Bot**\n"
+            f"**User Username**: @{message.from_user.username if message.from_user else None}\n"
+            f"**User ID**: `{message.from_user.id}`\n"
+            f"**Chat Title**: `{message.chat.title if message.chat else None}`\n"
+            f"**Chat ID**: `{message.chat.id if message.chat else None}`\n"
+        )
+        response_log = await send_log(user_detail)
+        if response_log is None:
+            LOGS.warning("Error response")
+        LOGS.info(response_log)
+        await db._update_openai_chat_in_db(message.from_user.id, backup_chat)
+        await client.send_chat_action(message.chat.id, enums.ChatAction.CANCEL)
+        return
+    except Exception as e:
+        return await pro.edit_text(f"Error: {e}") 
+
 @Client.on_message(
     filters.incoming
     & (