Image process fix

akn/Gemini/gemini.py

@@ -320,9 +320,9 @@ async def chatbot_talk(client: Client, message: Message):
             new_js = None
             backup_chat = await db._get_chatbot_chat_from_db(message.from_user.id)
             if new_check_flux_matches:
+                new_js = await message.reply_text("Please wait, it's still being processed")
                 try:
                     backup_chat.append({"role": "user", "parts": [{"text": query_base}]})
-                    new_js = await message.reply_text("Please wait, it's still being processed")
                     response_js = await js.image.create(
                         "black-forest-labs/flux-1-schnell",
                         image_read=True,
@@ -346,7 +346,29 @@ async def chatbot_talk(client: Client, message: Message):
                     await new_js.delete()
                     return
                 except ImageProcessFailed as e:
-                    return await new_js.edit_text("The server failed to process your image")
+                    await new_js.edit_text("Please wait, Error 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...")
+                    await message.reply_photo(
+                        file_path,
+                        progress=progress,
+                        progress_args=(
+                            new_js,
+                            time.time(),
+                            "Uploading image..."
+                        )
+                    )
+                    backup_chat.append({"role": "model", "parts": [{"text": f"IMAGE OUTPUT: {query_base}"}]})
+                    await db._update_chatbot_chat_in_db(message.from_user.id, backup_chat)
+                    await new_js.delete()
+                    return
                 except Exception as e:
                     LOGS.error(f"Error new_check_flux_matches {str(e)}")
                     return await new_js.edit_text("Try again error image")