Update Akeno/plugins/chatgpt.py

Akeno/plugins/chatgpt.py

@@ -24,7 +24,7 @@ from pyrogram import Client, filters
 from pyrogram.types import *
 from RyuzakiLib import FaceAI, FullStackDev, GeminiLatest, RendyDevChat
 
-from Akeno.utils.chat import chat_message
+from Akeno.utils.chat import *
 from Akeno.utils.tools import *
 from Akeno.utils.database import db
 from Akeno.utils.handler import *
@@ -423,6 +423,36 @@ async def chatgpt(client: Client, message: Message):
         LOGS.error(str(e))
         return await message.reply_text(str(e))
 
+@Akeno(
+    ~filters.scheduled
+    & filters.command(["askcode"], CMD_HANDLER)
+    & filters.me
+    & ~filters.forwarded
+)
+async def chatgptcode(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 CHATGPT-4O")
+    try:
+        messager = await chat_message_codegen(prompt)
+        if len(messager) > 4096:
+            with open("chat.txt", "w+", encoding="utf8") as out_file:
+                out_file.write(messager)
+            await message.reply_document(
+                document="chat.txt",
+                disable_notification=True
+            )
+            os.remove("chat.txt")
+        else:
+            await message.reply_text(messager, reply_to_message_id=ReplyCheck(message))
+    except Exception as e:
+        LOGS.error(str(e))
+        return await message.reply_text(str(e))
+
+
 module = modules_help.add_module("chatgpt", __file__)
 module.add_command("askf", "to read in the picture")
 module.add_command("askm", "to give random image questions")