updates for errors

app.py

@@ -266,6 +266,9 @@ with gr.Blocks(title="AI Image Generator & Text Overlay") as demo:
     gr.Markdown("# 🎨 AI Image Generator & Text Overlay")
     gr.Markdown("Generate images with AI or upload your own, then add custom text overlays!")
     
+    # Shared state for the current image
+    current_image = gr.State(value=None)
+    
     # Tab 1: Image Generation
     with gr.Tab("🖼️ Image Generation", id="generation_tab"):
         gr.Markdown("## Generate or Upload Images")
@@ -323,19 +326,19 @@ with gr.Blocks(title="AI Image Generator & Text Overlay") as demo:
         ideogram_btn.click(
             generate_ideogram_image,
             inputs=[ideogram_key, prompt, ideogram_aspect],
-            outputs=[generated_image, status_message]
+            outputs=[generated_image, status_message, current_image]
         )
         
         dalle_btn.click(
             generate_dalle_image,
             inputs=[dalle_key, prompt, dalle_size],
-            outputs=[generated_image, status_message]
+            outputs=[generated_image, status_message, current_image]
         )
         
         upload_btn.click(
             process_uploaded_image,
             inputs=[upload_image],
-            outputs=[generated_image, status_message]
+            outputs=[generated_image, status_message, current_image]
         )
     
     # Tab 2: Add Text Overlay
@@ -348,12 +351,12 @@ with gr.Blocks(title="AI Image Generator & Text Overlay") as demo:
                 # Text pattern selection
                 pattern = gr.Dropdown(
                     choices=[
-                        ("2-lines-top", "2 lines - Top"),
-                        ("2-lines-center", "2 lines - Center"),
-                        ("2-lines-bottom", "2 lines - Bottom"),
-                        ("3-lines-top", "3 lines - Top"),
-                        ("3-lines-center", "3 lines - Center"),
-                        ("3-lines-bottom", "3 lines - Bottom")
+                        "2-lines-top",
+                        "2-lines-center", 
+                        "2-lines-bottom",
+                        "3-lines-top",
+                        "3-lines-center",
+                        "3-lines-bottom"
                     ],
                     value="2-lines-top",
                     label="Layout Pattern"
@@ -396,7 +399,7 @@ with gr.Blocks(title="AI Image Generator & Text Overlay") as demo:
         # Event handler for Tab 2
         add_text_btn.click(
             add_text_to_image,
-            inputs=[generated_image, pattern, line1_inp, line2_inp, line3_inp, font_size, color],
+            inputs=[current_image, pattern, line1_inp, line2_inp, line3_inp, font_size, color],
             outputs=[text_image, status_message]
         )