_

chatbot/plugins/chat.py

@@ -26,6 +26,7 @@ import base64
 import re
 import uuid
 import shlex
+import requests
 import subprocess
 import sys
 import time
@@ -92,6 +93,23 @@ blacklist_patterns = [
     r"\bdeath\s*threat\b",
 ]
 
+def aigen_check(text: str):
+    reason = ""
+    try:
+        response = requests.get(
+            f"https://randydev-ryu-js.hf.space/api/v1/ai/akenox/aigen-v1?query={text}",
+            headers={"x-api-key": "akeno_UKQEQMt991kh2Ehh7JqJYKapx8CCyeC"}
+        ).json()
+        ok = json.loads(response["results"])
+
+        if ok.get("prompt"):
+            reason += ok["prompt"]
+        if ok.get("is_image") is True:
+            return True, reason
+        return False, reason
+    except (json.decoder.JSONDecodeError, KeyError, TypeError):
+        return False, reason
+
 def is_command_disabled(text):
   text_lower = text.lower()
   if text_lower.startswith("/") and not (
@@ -1474,12 +1492,8 @@ async def chatbot_talk(client: Client, message: Message):
                 await message.reply_text(CREDITS_DEVELOPER)
                 return
 
-            if re.findall(r"\b(image)\b", query_base, re.IGNORECASE):
-                if query_base == "image" or len(query_base) > 350:
-                    return await message.reply_text(
-                        "You can't write **image** small, it has to be long OR short first\n"
-                        "OR Text Too many `MAX_TOO_MANY_TEXT: 350`"
-                    )
+            is_image, new_prompt_gen, = aigen_check(query_base)
+            if is_image:
                 try:
                     buttons = [
                         [
@@ -1503,7 +1517,7 @@ async def chatbot_talk(client: Client, message: Message):
                     ]
                     await db.backup_chatbot.update_one(
                         {"user_id": message.from_user.id},
-                        {"$set": {"prompt_image": query_base}},
+                        {"$set": {"prompt_image": new_prompt_gen}},
                         upsert=True
                     )
                     await message.reply_photo(