Fix Media caption too long

akn/Gemini/gemini.py

@@ -336,6 +336,8 @@ async def chatbot_talk(client: Client, message: Message):
             return await ai_reply.edit_text(f"Error: {e}")
 
     if message.text:
+        query_base = None
+        start_time = time.time()
         await client.send_chat_action(message.chat.id, enums.ChatAction.TYPING)
         await asyncio.sleep(1.5)
         query = message.text.strip()
@@ -369,9 +371,11 @@ async def chatbot_talk(client: Client, message: Message):
                     await new_js.edit_text("Uploading image completed...")
                     x = GeminiLatest(api_keys=GOOGLE_API_KEY)
                     response_reads = x.get_response_image("What this?", file_path)
+                    end_time = time.time()
+                    execution_time = end_time - start_time
                     await message.reply_photo(
                         file_path,
-                        caption=response_reads,
+                        caption=response_reads + f"\nExecution time `{execution_time:.2f}` seconds",
                         progress=progress,
                         progress_args=(
                             new_js,
@@ -397,9 +401,42 @@ async def chatbot_talk(client: Client, message: Message):
                     await new_js.edit_text("Uploading image completed...")
                     x = GeminiLatest(api_keys=GOOGLE_API_KEY)
                     response_reads = x.get_response_image("What this?", file_path)
+                    end_time = time.time()
+                    execution_time = end_time - start_time
                     await message.reply_photo(
                         file_path,
-                        caption=response_reads,
+                        caption=response_reads + f"\nExecution time `{execution_time:.2f}` seconds",
+                        progress=progress,
+                        progress_args=(
+                            new_js,
+                            time.time(),
+                            "Uploading image completed..."
+                        )
+                    )
+                    backup_chat.append({"role": "model", "parts": [{"text": response_reads}]})
+                    await db._update_chatbot_chat_in_db(message.from_user.id, backup_chat)
+                    await new_js.delete()
+                    return
+                
+                except MediaCaptionTooLong as e:
+                    backup_chat.append({"role": "user", "parts": [{"text": query_base}]})
+                    await new_js.edit_text("Please wait, caption too try still processing...")
+                    response_js = await js.image.create(
+                        "black-forest-labs/flux-1-schnell",
+                        image_read=True,
+                        params_data={"query": query_base},
+                    )
+                    file_path = "randydev.jpg"
+                    with open(file_path, "wb") as f:
+                        f.write(response_js)
+                    await new_js.edit_text("Uploading image completed...")
+                    x = GeminiLatest(api_keys=GOOGLE_API_KEY)
+                    response_reads = x.get_response_image("What this?", file_path)
+                    end_time = time.time()
+                    execution_time = end_time - start_time
+                    await message.reply_photo(
+                        file_path,
+                        caption=f"Execution time `{execution_time:.2f}` seconds",
                         progress=progress,
                         progress_args=(
                             new_js,