From 42cc5c4f265a614a655d93de1f7a50cdd7591e0e Mon Sep 17 00:00:00 2001
From: thedragonsinn <98635854+thedragonsinn@users.noreply.github.com>
Date: Sat, 3 May 2025 21:17:27 +0530
Subject: [PATCH] `ai:` fix image generation.

---
 app/plugins/ai/gemini_core.py | 18 ++++++++++--------
 1 file changed, 10 insertions(+), 8 deletions(-)

diff --git a/app/plugins/ai/gemini_core.py b/app/plugins/ai/gemini_core.py
index c311c60..e001ed4 100644
--- a/app/plugins/ai/gemini_core.py
+++ b/app/plugins/ai/gemini_core.py
@@ -69,17 +69,19 @@ def get_response_content(
         parts[0]
     except (AttributeError, IndexError, TypeError):
         LOGGER.info(response)
-        return "Query failed... Try again", None
+        return "`Query failed... Try again`", None
 
-    try:
-        image_data = io.BytesIO(parts[0].inline_data.data)
-        image_data.name = "photo.jpg"
-    except (AttributeError, IndexError):
-        image_data = None
-
-    text = "\n".join([part.text for part in parts if part.text])
+    image_data = None
+    text = ""
     sources = ""
 
+    for part in parts:
+        if part.text:
+            text += part + "\n"
+        if part.inline_data:
+            image_data = io.BytesIO(part.inline_data.data)
+            image_data.name = "photo.jpg"
+
     if add_sources:
         try:
             hrefs = [