Update app.py

app.py

@@ -15,6 +15,7 @@ def generate_edit(prompt, pil_image, api_key, model="gemini-2.0-flash-exp"):
     with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp_img:
         image_path = tmp_img.name
         pil_image.save(image_path)
+
     files = [client.files.upload(file=image_path)]
     contents = [
         types.Content(
@@ -33,8 +34,10 @@ def generate_edit(prompt, pil_image, api_key, model="gemini-2.0-flash-exp"):
         response_modalities=["image", "text"],
         response_mime_type="text/plain",
     )
+
     text_response = ""
     image_out_path = None
+
     with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp_out:
         out_path = tmp_out.name
         for chunk in client.models.generate_content_stream(
@@ -54,29 +57,24 @@ def generate_edit(prompt, pil_image, api_key, model="gemini-2.0-flash-exp"):
     del files
     return image_out_path, text_response
 
-def process_image_and_prompt(pil_image, prompt, api_key, progress_callback=None):
+def process_image_and_prompt(pil_image, prompt, api_key):
     try:
-        if progress_callback:
-            progress_callback("Generating…")
         image_path, text_out = generate_edit(prompt, pil_image, api_key)
         if image_path:
             img = Image.open(image_path)
             if img.mode == "RGBA":
                 img = img.convert("RGB")
-            if progress_callback:
-                progress_callback("Done ✓")
-            return img, "Image generated successfully!", None
+            return img # Return only the image on success
         else:
-            if progress_callback:
-                progress_callback("Failed to generate image")
-            return None, f"⚠️ {text_out.strip()}", None
+            # If no image generated, raise an error for Gradio popup
+            raise gr.Error(f"⚠️ Image generation failed: {text_out.strip() if text_out.strip() else 'No specific error message.'}")
     except Exception as e:
-        if progress_callback:
-            progress_callback("Error")
-        return None, f"❌ Generation failed: {str(e)}", None
+        # Catch any other exceptions and re-raise as Gradio error
+        raise gr.Error(f"❌ Generation failed: {str(e)}")
 
 def reset_inputs(api_key_value=None):
-    return None, "", api_key_value or "", ""
+    # Reset all inputs, keeping API key unchanged
+    return None, "", api_key_value or ""
 
 # Styles
 css_style = """
@@ -106,10 +104,6 @@ body, .app-container { background: var(--bg); color: var(--text); }
 @media (max-width: 1100px) {
   .layout-two-col { grid-template-columns: 1fr; }
 }
-.input-area, .output-area { background: #202533; border-radius: 12px; padding: 12px; }
-#status-container { display: flex; flex-direction: column; gap: 6px; padding: 6px; background: #1b1f2b; border-radius: 8px; }
-#status-label { font-weight: 700; color: #a6d1ff; }
-#status-text { height: 6em; max-height: 8em; overflow: auto; white-space: pre-wrap; word-break: break-word; padding: 6px 8px; border: none; background: transparent; color: #cbd5e1; font-weight: 600; }
 
 #output-viewport { display: flex; justify-content: center; align-items: center; min-height: 260px; }
 #output-image { display: flex; justify-content: center; align-items: center; }
@@ -122,10 +116,11 @@ with gr.Blocks(css=css_style) as app:
     <div class='header-block'>
       <div class='header-gradient'>
         <h1 class='header-title'>🖼️ Image Editor <span style="font-size:1.1em;">(Powered by Gemini)</span> 🔮</h1>
-        <div class='header-subtitle'>Step-by-step prompts with a persistent status banner and progress feedback</div>
+        <div class='header-subtitle'>Step-by-step prompts for image editing</div>
       </div>
     </div>
     """)
+
     with gr.Row():
         with gr.Column(scale=3, elem_classes="sidebar"):
             gr.Markdown(
@@ -158,28 +153,25 @@ with gr.Blocks(css=css_style) as app:
             with gr.Row():
                 submit_btn = gr.Button("Generate Edit", elem_classes="gradient-button")
                 reset_btn = gr.Button("Reset Inputs")
+            
+            # Note: Status bar elements removed as requested. Errors will now show as Gradio popups.
 
-            # Status area (enhanced)
-            with gr.Column():
-                gr.HTML("<div id='status-container'><div id='status-label'>Status</div></div>")
-                status_bar = gr.Textbox(label="", interactive=False, lines=6, elem_id="status-text")
-
-            def on_submit(pil_img, prompt, key, progress=None):
+            def on_submit(pil_img, prompt, key):
                 if not key or key.strip() == "":
                     raise gr.Error("Gemini API Key is required!")
-                img, stat, _ = process_image_and_prompt(pil_img, prompt, key, progress)
-                return img, stat
+                # process_image_and_prompt now raises gr.Error directly for failures
+                return process_image_and_prompt(pil_img, prompt, key)
 
             submit_btn.click(
                 fn=on_submit,
                 inputs=[image_input, prompt_input, api_key_input],
-                outputs=[output_image, status_bar]
+                outputs=[output_image] # Only output the image
             )
 
             reset_btn.click(
                 fn=reset_inputs,
                 inputs=[api_key_input],
-                outputs=[image_input, prompt_input, api_key_input, status_bar]
+                outputs=[image_input, prompt_input, api_key_input] # Remove status_bar from outputs
             )
 
 app.launch()