Update Akeno/utils/chat.py

Akeno/utils/chat.py

@@ -29,6 +29,43 @@ a short and concise answer. Give short step-by-step reasoning if required.
 Today is {dt.now():%A %d %B %Y %H:%M}
 """
 
+create_pyrogram_base = f"""
+to create pyrogram library [PYTHON]
+
+- replace other custom commands
+- add comment #
+- do not use java, c++, javascript
+- do not use telethon
+
+from pyrogram import Client, filters
+from pyrogram.erros import *
+from Akeno.utils.handler import *
+from config import *
+
+@Akeno(
+    ~filters.scheduled
+    & filters.command(["custom"], CMD_HANDLER)
+    & filters.me
+    & ~filters.forwarded
+)
+async def example(client: Client, message: Message):
+    pass
+
+Today is {dt.now():%A %d %B %Y %H:%M}
+"""
+
+async def chat_message_codegen(question):
+    clients_x = Clients_g4f()
+    response = clients_x.chat.completions.create(
+        model="gpt-4o",
+        messages=[
+            {"role": "system", "content": create_pyrogram_base},
+            {"role": "user", "content": question}
+        ],
+    )
+    messager = response.choices[0].message.content
+    return messager
+
 async def chat_message(question):
     clients_x = Clients_g4f()
     response = clients_x.chat.completions.create(