Update chatbot/plugins/chat.py

chatbot/plugins/chat.py

@@ -482,10 +482,56 @@ async def chatbot_talk(client: Client, message: Message):
 
     if message.photo:
         file_photo = None
+        captions = ""
+        file_path = "gemini-native-image.png"
         try:
             caption = message.caption or "What this?"
             await client.send_chat_action(message.chat.id, enums.ChatAction.UPLOAD_PHOTO)
             await asyncio.sleep(1.5)
+
+            if re.findall(r"\b(picture)\b", caption, re.IGNORECASE):
+                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],
+                        config=types.GenerateContentConfig(
+                            response_modalities=['TEXT', 'IMAGE']
+                        )
+                    )
+                    for part in response.candidates[0].content.parts:
+                        if part.text is not None:
+                            captions += part.text
+                        elif part.inline_data is not None:
+                            image = Image.open(BytesIO(part.inline_data.data))
+                            image.save(file_path)
+                    keyboard_like = create_keyboard(user_id=message.from_user.id)
+                    await message.reply_photo(
+                        file_path,
+                        caption=captions,
+                        reply_markup=keyboard_like
+                    )
+                    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:
+                    LOGS.error(f"Error: Gemini Edit Image: {str(e)}")
+                    return await message.reply_text("Server busy try again later")
+                finally:
+                    if file_path:
+                        try:
+                            os.remove(file_path)
+                        except:
+                            pass
+    
             if re.findall(r"\b(pro:editimage)\b", caption, re.IGNORECASE):
                 await db.backup_chatbot.update_one(
                     {"user_id": message.from_user.id},