Update chatbot/plugins/chat.py

chatbot/plugins/chat.py

@@ -61,6 +61,15 @@ trans = SyncTranslator()
 js = AkenoXJs(DifferentAPIDefault()).connect()
 
 message_memory_state = {}
+DISABLED_COMMANDS =  [
+    "/",
+    "/help",
+    "about:",
+    "enabled:",
+    "disabled:",
+    "--stream-image",
+    "--edit-image"
+]
 
 def obfuscate(word):
     return word.lower()\
@@ -931,9 +940,9 @@ async def chatbot_talk(client: Client, message: Message):
             if regex_all_blacklist(caption):
                 return await message.reply_text("You been blocked blacklisted")
 
-            if caption == "--stream-image":
+            if caption == "--stream-image" or caption == "--edit-image" or len(caption) > 350:
                 return await message.reply_text(
-                    "You can't --stream-image, it has to be long OR short first\n"
+                    "You can't type, it has to be long or short"
                 )
 
             stream_image_now = re.sub(r"--stream-image", "", caption).strip()
@@ -947,7 +956,7 @@ async def chatbot_talk(client: Client, message: Message):
                     async for chunk in await gen.aio.models.generate_content_stream(
                         model='gemini-2.0-flash-001',
                         contents=[
-                            str(caption),
+                            str(stream_image_now),
                             ty.Part.from_bytes(data=image_bytes, mime_type="image/jpeg")
                         ]
                     ):
@@ -981,19 +990,15 @@ async def chatbot_talk(client: Client, message: Message):
                         except:
                             pass
 
-            if re.findall(r"\b(This is a picture of me)\b", caption, re.IGNORECASE):
-                if caption == "picture" or len(caption) > 300:
-                    return await message.reply_text(
-                        "You can't **picture** small, it has to be long OR short first\n"
-                        "OR Caption Too many `MAX_CAPTION: 300`"
-                    )
+            edit_image_now = re.sub(r"--edit-image", "", caption).strip()
+            if edit_image_now:
                 try:
                     backup_chat = await db._get_chatbot_chat_from_db(message.from_user.id)
                     backup_chat.append({"role": "user", "parts": [{"text": caption}]})
                     images = Image.open(await message.download())
                     response = await gen.aio.models.generate_content(
                         model="gemini-2.0-flash-exp-image-generation",
-                        contents=[str(caption), images],
+                        contents=[str(edit_image_now), images],
                         config=ty.GenerateContentConfig(
                             response_modalities=['TEXT', 'IMAGE']
                         )
@@ -1212,7 +1217,7 @@ async def chatbot_talk(client: Client, message: Message):
             if regex_all_blacklist(query_base):
                 return await message.reply_text("You been blocked blacklisted")
 
-            if query_base in ["/", "/help", "about:", "enabled:", "disabled:"]:
+            if query_base in DISABLED_COMMANDS:
                 await client.send_chat_action(message.chat.id, enums.ChatAction.CANCEL)
                 return