update app.py - fix of errors

app.py

@@ -35,7 +35,8 @@ def generate_ideogram_image(api_key, prompt, aspect_ratio="ASPECT_1_1"):
                 image_url = result["data"][0]["url"]
                 img_response = requests.get(image_url)
                 image = Image.open(io.BytesIO(img_response.content))
-                return image, "Image generated successfully!", image
+                safe_copy = safe_image_copy(image)
+                return image, "Image generated successfully!", safe_copy
             else:
                 return None, "No image data received", None
         else:
@@ -61,7 +62,8 @@ def generate_dalle_image(api_key, prompt, size="1024x1024"):
         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!", image
+        safe_copy = safe_image_copy(image)
+        return image, "Image generated successfully with DALL-E 3!", safe_copy
     
     except Exception as e:
         return None, f"Error generating image: {str(e)}", None
@@ -70,7 +72,12 @@ def process_uploaded_image(image):
     """Process uploaded image"""
     if image is None:
         return None, "No image uploaded", None
-    return image, "Image uploaded successfully!", image
+    try:
+        safe_copy = safe_image_copy(image)
+        return image, "Image uploaded successfully!", safe_copy
+    except Exception as e:
+        print(f"Error processing uploaded image: {e}")
+        return image, f"Image uploaded with warning: {e}", image
 
 # Text Overlay Functions
 def add_text_to_image(img, pattern, line1, line2, line3, font_size, color, add_outline):
@@ -340,11 +347,6 @@ def create_pattern_template(pattern, width=200, height=120):
         positions = [(width // 2, int(height * 0.45)), (width // 2, int(height * 0.65)), (width // 2, int(height * 0.85))]
     
     # Draw white rectangles to represent text boxes
-    try:
-        font = ImageFont.load_default()
-    except:
-        font = None
-    
     for i, (x, y) in enumerate(positions):
         # Draw white rectangle as text placeholder
         box_width = 60
@@ -355,18 +357,35 @@ def create_pattern_template(pattern, width=200, height=120):
         bottom = y + box_height // 2
         
         draw.rectangle([left, top, right, bottom], fill='white', outline='white')
-        
-        # Add small text label
-        if font:
-            try:
-                label = f"Line {i+1}"
-                draw.text((x, y), label, fill='black', font=font, anchor="mm")
-            except TypeError:
-                # Fallback for older PIL
-                draw.text((x-15, y-5), f"L{i+1}", fill='black', font=font)
     
     return img
 
+# Create templates
+def get_pattern_template(pattern):
+    """Get template for pattern - create on demand to avoid initialization issues"""
+    return create_pattern_template(pattern)
+
+# Fix the image passing with better error handling
+def safe_image_copy(img):
+    """Safely copy an image to avoid corruption issues"""
+    if img is None:
+        return None
+    try:
+        # Convert to RGB and create a fresh copy
+        if hasattr(img, 'mode'):
+            # It's a PIL image
+            rgb_img = img.convert('RGB')
+            # Create a new image by copying pixel data
+            new_img = Image.new('RGB', rgb_img.size)
+            new_img.paste(rgb_img)
+            return new_img
+        else:
+            print(f"Warning: Unexpected image type: {type(img)}")
+            return img
+    except Exception as e:
+        print(f"Error copying image: {e}")
+        return img
+
 # Function to convert PIL image to base64 for Gradio
 def pil_to_base64(img):
     """Convert PIL image to base64 string for Gradio"""
@@ -496,10 +515,10 @@ with gr.Blocks(title="AI Image Generator & Text Overlay") as demo:
                         )
                     
                     with gr.Column(scale=1):
-                        # Template preview image
+                        # Template preview image - initialize with None first
                         template_preview = gr.Image(
                             label="Template Preview",
-                            value=pattern_templates["2-lines-top"],
+                            value=None,  # Start with None to avoid initialization issues
                             show_label=True,
                             interactive=False,
                             width=200,
@@ -551,7 +570,7 @@ with gr.Blocks(title="AI Image Generator & Text Overlay") as demo:
         
         # Update template when pattern changes
         pattern.change(
-            update_pattern_template,
+            get_pattern_template,
             inputs=[pattern],
             outputs=[template_preview]
         )
@@ -574,6 +593,12 @@ with gr.Blocks(title="AI Image Generator & Text Overlay") as demo:
             outputs=[text_image, status_message]
         )
 
+# Initialize template preview after the interface is created
+demo.load(
+    lambda: get_pattern_template("2-lines-top"),
+    outputs=[template_preview]
+)
+
 # Launch the application
 if __name__ == "__main__":
     demo.launch(