updated

akn/Gemini/gemini.py

@@ -315,34 +315,45 @@ async def chatbot_talk(client: Client, message: Message):
         command = parts[0].lower()
         pic_query = parts[1].strip() if len(parts) > 1 else ""
         new_check_flux_matches = re.findall(r"\bimage\b", query_base)
+        if query_base == "image":
+            return await message.reply_text("i don't have, what do you mean by image?")
         try:
+            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")
-                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..."
+                try:
+                    backup_chat.append({"role": "user", "parts": [{"text": query_base}]})
+                    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,
+                        params_data={"query": query_base},
                     )
-                )
-                await new_js.delete()
-                return
+                    file_path = "randydev.jpg"
+                    with open(file_path, "wb") as f:
+                        f.write(response_js)
+                    ok = await message.edit_text("Uploading image...")
+                    await message.reply_photo(
+                        file_path,
+                        progress=progress,
+                        progress_args=(
+                            "Please wait, it's still being processed",
+                            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 ok.delete()
+                    return
+                except ImageProcessFailed as e:
+                    return await message.edit_text("The server failed to process your image")
+                except Exception as e:
+                    LOGS.error(f"Error new_check_flux_matches {str(e)}")
+                    return await message.edit_text("Try again error image")
     
             model_flash = genai.GenerativeModel(
                 model_name="gemini-1.5-flash"
             )
-            backup_chat = await db._get_chatbot_chat_from_db(message.from_user.id)
             backup_chat.append({"role": "user", "parts": [{"text": query_base}]})
             chat_session = model_flash.start_chat(history=backup_chat)
             response_data = chat_session.send_message(query_base)