Update app.py

app.py

@@ -66,6 +66,7 @@ def generate_image(prompt, image=None, output_filename="generated_image"):
         config=generate_content_config,
     )
 
+    full_text_response = "" # For debugging text truncation
     # Process the response
     for chunk in response:
         if not chunk.candidates or not chunk.candidates[0].content or not chunk.candidates[0].content.parts:
@@ -79,10 +80,13 @@ def generate_image(prompt, image=None, output_filename="generated_image"):
             # Convert binary data to PIL Image
             img = Image.open(io.BytesIO(inline_data.data))
             return img, f"Image saved as {filename}"
-        else:
-            return None, chunk.text
+        elif chunk.text:
+            full_text_response += chunk.text # Append chunk text for full response
+            print("Chunk Text Response:", chunk.text) # Debugging chunk text
+
+    print("Full Text Response from Gemini:", full_text_response) # Debugging full text
+    return None, full_text_response # Return full text response
 
-    return None, "No image generated"
 
 # Function to handle chat interaction
 def chat_handler(prompt, user_image, chat_history, output_filename="generated_image"):
@@ -114,6 +118,7 @@ def chat_handler(prompt, user_image, chat_history, output_filename="generated_im
         # Create thumbnail for chatbot
         thumbnail_size = (100, 100) # Define thumbnail size
         thumbnail = img.copy()
+        thumbnail = thumbnail.convert("RGB") # Force RGB mode for thumbnail
         thumbnail.thumbnail(thumbnail_size)
 
         # Convert thumbnail to base64 for chatbot display
@@ -121,6 +126,7 @@ def chat_handler(prompt, user_image, chat_history, output_filename="generated_im
         thumbnail.save(buffered, format="PNG")
         thumbnail_base64 = base64.b64encode(buffered.getvalue()).decode()
         thumbnail_data_uri = f"data:image/png;base64,{thumbnail_base64}"
+        print("Thumbnail Data URI:", thumbnail_data_uri) # Print to console
         assistant_message_content = gr.HTML(f'<img src="{thumbnail_data_uri}" alt="Generated Image Thumbnail">') # Use gr.HTML
     else:
         assistant_message_content = status # If no image, send text status