Update Akeno/plugins/chatgpt.py

Akeno/plugins/chatgpt.py

@@ -41,6 +41,13 @@ async def mistraai(messagestr):
         return None
     return response.json()
 
+async def mode_web_gpt(query):
+    url = f"https://akeno.randydev.my.id/api/akeno-ai-web?query={query}"
+    response = requests.get(url)
+    if response.status_code != 200:
+        return None
+    return response.json()
+
 async def chatgptold(messagestr):
     url = "https://akeno.randydev.my.id/ryuzaki/chatgpt-old"
     payload = {"query": messagestr}
@@ -73,6 +80,40 @@ def akeno_ai_send_message(message):
     full_content = ''.join(contents)
     return full_content
 
+
+@Akeno(
+    ~filters.scheduled
+    & filters.command(["askweb"], CMD_HANDLER)
+    & filters.me
+    & ~filters.forwarded
+)
+async def askweb(client: Client, message: Message):
+    if len(message.command) > 1:
+        prompt = message.text.split(maxsplit=1)[1]
+    elif message.reply_to_message:
+        prompt = message.reply_to_message.text
+    else:
+        return await message.reply_text("Give ask from Web")
+    try:
+        response = await mode_web_gpt(prompt)
+        if not response:
+           return await message.reply_text("No response")
+        output = response["randydev"].get("message") 
+        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
+            )
+            os.remove("chat.txt")
+        else:
+            await message.reply_text(output, reply_to_message_id=ReplyCheck(message))
+    except Exception as e:
+        LOGS.error(str(e))
+        return await message.reply_text(str(e))
+
+
 @Akeno(
     ~filters.scheduled
     & filters.command(["askakeno"], CMD_HANDLER)