Update chatbot/plugins/chat.py

chatbot/plugins/chat.py

@@ -945,7 +945,7 @@ async def chatbot_talk(client: Client, message: Message):
                     "You can't type (error 404: creativity not found)"
                 )
 
-            stream_image_now = re.sub(r"--stream-images", "", caption).strip()
+            stream_image_now = re.match(r"--stream-images\s*(.*)", caption, re.IGNORECASE)
             if stream_image_now:
                 STREAM_IMAGE_PATH = await message.download()
                 try:
@@ -956,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(stream_image_now),
+                            str(stream_image_now.group(1)),
                             ty.Part.from_bytes(data=image_bytes, mime_type="image/jpeg")
                         ]
                     ):
@@ -990,16 +990,15 @@ async def chatbot_talk(client: Client, message: Message):
                         except:
                             pass
 
-            edit_image_now = re.sub(r"--edit-images", "", caption).strip()
+            edit_image_now = re.match(r"--edit-images\s*(.*)", caption, re.IGNORECASE)
             if edit_image_now:
-                LOGS.info(f"This Enabled cleared: {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(edit_image_now), images],
+                        contents=[str(edit_image_now.group(1)), images],
                         config=ty.GenerateContentConfig(
                             response_modalities=['TEXT', 'IMAGE']
                         )