Updated

chatbot/plugins/chat.py

@@ -93,22 +93,30 @@ blacklist_patterns = [
     r"\bdeath\s*threat\b",
 ]
 
-def aigen_check(text: str):
-    reason = ""
+def aigen_check(user_id: int, text: str):
+    result = {}
     try:
         response = requests.get(
-            f"https://randydev-ryu-js.hf.space/api/v1/ai/akenox/aigen-v1?query={text}",
+            f"https://randydev-ryu-js.hf.space/api/v1/ai/akenox/aigen-v2?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
+        ok = json.loads(response.get("results", "{}"))
+
+        result[user_id] = {
+            "is_image": ok.get("is_image", False),
+            "prompt": ok.get("prompt", ""),
+            "is_anti_porno": ok.get("is_anti_porno", False),
+            "reason": ok.get("reason", "")
+        }
+        return result
+    except (json.decoder.JSONDecodeError, KeyError, TypeError, requests.RequestException) as e:
+        return {user_id: {
+            "is_image": False,
+            "prompt": "",
+            "is_anti_porno": False,
+            "reason": f"Error: {str(e)}"
+        }}
 
 def is_command_disabled(text):
   text_lower = text.lower()
@@ -1509,8 +1517,12 @@ async def chatbot_talk(client: Client, message: Message):
                 await message.reply_text(CREDITS_DEVELOPER)
                 return
 
-            is_image, new_prompt_gen, = aigen_check(query_base)
-            if is_image:
+            check_data = aigen_check(message.from_user.id, query_base)
+            user_result = check_data.get(message.from_user.id, {})
+            if user_result.get("is_anti_porno"):
+                return await message.reply_text(user_result["reason"])
+                
+            if user_result.get("is_image"):
                 try:
                     buttons = [
                         [
@@ -1534,7 +1546,7 @@ async def chatbot_talk(client: Client, message: Message):
                     ]
                     await db.backup_chatbot.update_one(
                         {"user_id": message.from_user.id},
-                        {"$set": {"prompt_image": new_prompt_gen}},
+                        {"$set": {"prompt_image": user_result["prompt"]}},
                         upsert=True
                     )
                     await message.reply_photo(