Update chatbot/plugins/chat.py

chatbot/plugins/chat.py

@@ -659,6 +659,7 @@ async def chatbot_talk(client: Client, message: Message):
         await asyncio.sleep(1.5)
         query = message.text.strip()
         query_base = query
+        captions = ""
         file_path = "gemini-native-image.png"
         try:
             if query_base in ["/", "/help"]:
@@ -682,16 +683,17 @@ async def chatbot_talk(client: Client, message: Message):
                     )
                     for part in response.candidates[0].content.parts:
                         if part.text is not None:
-                            LOGS.info(f"Gemini Image: {part.text}")
+                            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
                     )
-                    backup_chat.append({"role": "model", "parts": [{"text": "Nothing test image"}]})
+                    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