update app.py - error handling

app.py

@@ -8,7 +8,7 @@ import openai
 def generate_ideogram_image(api_key, prompt, aspect_ratio="ASPECT_1_1"):
     """Generate image using Ideogram API"""
     if not api_key or not prompt:
-        return None, "Please provide both API key and prompt"
+        return None, "Please provide both API key and prompt", None
     
     url = "https://api.ideogram.ai/generate"
     headers = {
@@ -31,21 +31,20 @@ def generate_ideogram_image(api_key, prompt, aspect_ratio="ASPECT_1_1"):
             result = response.json()
             if result.get("data"):
                 image_url = result["data"][0]["url"]
-                # Download and return the image
                 img_response = requests.get(image_url)
                 image = Image.open(io.BytesIO(img_response.content))
-                return image, "Image generated successfully!"
+                return image, "Image generated successfully!", image
             else:
-                return None, "No image data received"
+                return None, "No image data received", None
         else:
-            return None, f"Error: {response.status_code} - {response.text}"
+            return None, f"Error: {response.status_code} - {response.text}", None
     except Exception as e:
-        return None, f"Error generating image: {str(e)}"
+        return None, f"Error generating image: {str(e)}", None
 
 def generate_dalle_image(api_key, prompt, size="1024x1024"):
     """Generate image using DALL-E 3 API"""
     if not api_key or not prompt:
-        return None, "Please provide both API key and prompt"
+        return None, "Please provide both API key and prompt", None
     
     try:
         client = openai.OpenAI(api_key=api_key)
@@ -57,20 +56,19 @@ def generate_dalle_image(api_key, prompt, size="1024x1024"):
             n=1,
         )
         
-        # Download the image from URL
         image_url = response.data[0].url
         img_response = requests.get(image_url)
         image = Image.open(io.BytesIO(img_response.content))
-        return image, "Image generated successfully with DALL-E 3!"
+        return image, "Image generated successfully with DALL-E 3!", image
     
     except Exception as e:
-        return None, f"Error generating image: {str(e)}"
+        return None, f"Error generating image: {str(e)}", None
 
 def process_uploaded_image(image):
     """Process uploaded image"""
     if image is None:
-        return None, "No image uploaded"
-    return image, "Image uploaded successfully!"
+        return None, "No image uploaded", None
+    return image, "Image uploaded successfully!", image
 
 # Text Overlay Functions
 def add_text_to_image(img, pattern, line1, line2, line3, font_size, color):
@@ -224,17 +222,17 @@ def add_text_to_image(img, pattern, line1, line2, line3, font_size, color):
 
         if not text_drawn:
             print("No text was drawn")
-            return img, "No text to add (all lines were empty)"
+            return img, "No text to add (all lines were empty)", None
         
         print("=== Text drawing completed ===")
         print(f"Returning image with mode: {image.mode}, size: {image.size}")
-        return image, f"Text added! Pattern: {pattern}, Font: {font_size}px, Color: {original_color}, Lines drawn: {sum(1 for line in lines if line and line.strip())}"
+        return image, f"Text added! Pattern: {pattern}, Font: {font_size}px, Color: {original_color}, Lines drawn: {sum(1 for line in lines if line and line.strip())}", None
         
     except Exception as e:
         print(f"CRITICAL ERROR in add_text_to_image: {e}")
         import traceback
         traceback.print_exc()
-        return None, f"Error adding text: {str(e)}"
+        return None, f"Error adding text: {str(e)}", None
 
 def test_text_overlay():
     """Test function to verify text overlay works"""
@@ -250,16 +248,16 @@ def test_text_overlay():
             font = ImageFont.load_default()
             draw.text((200, 150), "TEST", fill='white', font=font, anchor="mm")
             print("Basic text drawing works!")
-            return test_img, "Test successful"
+            return test_img, "Test successful", None
         except Exception as e:
             print(f"Basic text drawing failed: {e}")
             draw.text((50, 100), "TEST", fill='white')
             print("Fallback text drawing works!")
-            return test_img, "Test with fallback"
+            return test_img, "Test with fallback", None
             
     except Exception as e:
         print(f"Test failed completely: {e}")
-        return None, f"Test failed: {e}"
+        return None, f"Test failed: {e}", None
 
 # Create the Gradio Interface
 with gr.Blocks(title="AI Image Generator & Text Overlay") as demo: