Update chatbot/plugins/chat.py

chatbot/plugins/chat.py

@@ -24,6 +24,7 @@ import asyncio
 import io
 import os
 import re
+import aiohttp
 from PIL import Image
 
 from pyrogram import *
@@ -43,6 +44,10 @@ from openai import AsyncOpenAI as openai
 import akenoai.openai as akeno
 import akenoai.logger as akeno_log
 
+from chatbot.plugins.user_database import users_collection
+from chatbot.plugins.keyboards import get_language_keyboard
+from chatbot.plugins.languages import LANGUAGES
+
 from . import force_sub
 
 
@@ -63,6 +68,27 @@ DISABLE_COMMAND = [
     "ask"
 ]
 
+LANGUAGES_DEFAULT = "en"
+
+async def translate(text, target_lang):
+    API_URL = "https://translate.googleapis.com/translate_a/single"
+    HEADERS = {"User-Agent": "Mozilla/5.0"}
+    params = {
+        "client": "gtx",
+        "sl": "auto",
+        "tl": target_lang,
+        "dt": "t",
+        "q": text,
+    }
+    async with aiohttp.ClientSession() as session:
+        async with session.post(API_URL, headers=HEADERS, params=params) as response:
+            if response.status == 200:
+                translation = await response.json()
+                translated_text = "".join([item[0] for item in translation[0]])
+                return translated_text
+            else:
+                return None
+
 GEMINI_START_TEXT = """
 Hey! {name}
 
@@ -188,6 +214,9 @@ async def on_mode(client: Client, cb: CallbackQuery):
 async def askcmd(client: Client, message: Message):
     pro = await message.reply("Processing your request...", quote=True)
     chat_user = await db.get_chatbot(message.chat.id)
+    user_id = message.from_user.id
+    user = await users_collection.find_one({"user_id": user_id})
+    lang = user.get("language") if user else LANGUAGES_DEFAULT
     if not chat_user:
         await pro.edit_text("Ok disable")
         return
@@ -208,7 +237,7 @@ async def askcmd(client: Client, message: Message):
             model="gpt-4o",
             messages=backup_chat
         )
-        output = response.choices[0].message.content
+        output = await translate(response.choices[0].message.content, lang)
         if len(output) > 4096:
             with open("chat.txt", "w+", encoding="utf8") as out_file:
                 out_file.write(output)
@@ -257,6 +286,9 @@ async def askcmd(client: Client, message: Message):
 async def chatbot_talk(client: Client, message: Message):
     chat_user = await db.get_chatbot(message.chat.id)
     pro = await message.reply("Processing your request...", quote=True)
+    user_id = message.from_user.id
+    user = await users_collection.find_one({"user_id": user_id})
+    lang = user.get("language") if user else LANGUAGES_DEFAULT
     if not chat_user:
         await pro.edit_text("Ok disable")
         return
@@ -285,7 +317,7 @@ async def chatbot_talk(client: Client, message: Message):
                 model="gpt-4o",
                 messages=backup_chat
             )
-            output = response.choices[0].message.content
+            output = await translate(response.choices[0].message.content, lang)
             if len(output) > 4096:
                 with open("chat.txt", "w+", encoding="utf8") as out_file:
                     out_file.write(output)