Update Akeno/plugins/chatgpt.py

Akeno/plugins/chatgpt.py

@@ -18,6 +18,7 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 
 import requests
+import json
 from pyrogram import *
 from pyrogram import Client, filters
 from pyrogram.types import *
@@ -47,6 +48,59 @@ async def chatgptold(messagestr):
         return None
     return response.json()
 
+def akeno_ai_send_message(message):
+    url = "https://ai.randydev.my.id/backend-api/v2/conversation"
+    payload = {
+        "api_key": "",
+        "auto_continue": True,
+        "conversation_id": "c527f012-ae20-47c1-881e-914617b5d995",
+        "id": "7411834416398216880",
+        "model": "",
+        "provider": "",
+        "messages": [{"role": "user", "content": message}]
+    }
+    response = requests.post(url, json=payload)
+    response_lines = response.text.strip().split("\n")
+    contents = []
+    for line in response_lines:
+        try:
+            json_obj = json.loads(line)
+            if json_obj.get("type") == "content":
+                contents.append(json_obj.get("content"))
+        except json.JSONDecodeError:
+            return f"Error: {line}"
+    full_content = ''.join(contents)
+    return full_content
+
+@Akeno(
+    ~filters.scheduled
+    & filters.command(["askakeno"], CMD_HANDLER)
+    & filters.me
+    & ~filters.forwarded
+)
+async def akenoai_(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 FaceAI")
+    try:
+        response = akeno_ai_send_message(message)
+        if len(response) > 4096:
+            with open("chat.txt", "w+", encoding="utf8") as out_file:
+                out_file.write(response)
+            await message.reply_document(
+                document="chat.txt",
+                disable_notification=True
+            )
+            os.remove("chat.txt")
+        else:
+            await message.reply_text(response)
+    except Exception as e:
+        LOGS.error(str(e))
+        return await message.reply_text(str(e)
+
 @Akeno(
     ~filters.scheduled
     & filters.command(["addchatbot"], CMD_HANDLER)