Update chatbot/plugins/chat.py

chatbot/plugins/chat.py

@@ -931,10 +931,60 @@ 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":
+                return await message.reply_text(
+                    "You can't --stream-image, it has to be long OR short first\n"
+                )
+
+            stream_image_now = re.sub(r"--stream-image", "", caption).strip()
+            if stream_image_now:
+                STREAM_IMAGE_PATH = await message.download()
+                try:
+                    backup_chat = await db._get_chatbot_chat_from_db(message.from_user.id)
+                    backup_chat.append({"role": "user", "parts": [{"text": caption}]})
+                    with open(STREAM_IMAGE_PATH, 'rb') as f:
+                        image_bytes = f.read()
+                    async for chunk in await gen.aio.models.generate_content_stream(
+                        model='gemini-2.0-flash-001',
+                        contents=[
+                            str(caption),
+                            ty.Part.from_bytes(data=image_bytes, mime_type="image/jpeg")
+                        ]
+                    ):
+                        captions += chunk.text
+
+                    if captions == "":
+                        return await message.reply_text("Server busy try again later")
+
+                    keyboard = create_keyboard(user_id=message.from_user.id)
+                    await message.reply_text(
+                        captions,
+                        disable_web_page_preview=True,
+                        reply_markup=keyboard
+                    )
+
+                    await db.backup_chatbot.update_one(
+                        {"user_id": message.from_user.id},
+                        {"$set": {"translate_text": captions}},
+                        upsert=True
+                    )
+                    backup_chat.append({"role": "model", "parts": [{"text": captions}]})
+                    await db._update_chatbot_chat_in_db(message.from_user.id, backup_chat)
+                    await client.send_chat_action(message.chat.id, enums.ChatAction.CANCEL)
+                    return
+                except Exception as e:
+                    return await message.reply_text("Server busy try again later")
+                finally:
+                    if STREAM_IMAGE_PATH:
+                        try:
+                            os.remove(STREAM_IMAGE_PATH)
+                        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 write **picture** small, it has to be long OR short first\n"
+                        "You can't **picture** small, it has to be long OR short first\n"
                         "OR Caption Too many `MAX_CAPTION: 300`"
                     )
                 try:
@@ -1162,7 +1212,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"]:
+            if query_base in ["/", "/help", "about:", "enabled:", "disabled:"]:
                 await client.send_chat_action(message.chat.id, enums.ChatAction.CANCEL)
                 return