Update app.py

app.py

@@ -35,14 +35,14 @@ def process_image(prompt, image, style, upscale_factor, inpaint):
 
         # Check if the result is valid
         if result and hasattr(result, 'images') and len(result.images) > 0:
-            return result.images[0]
+            return result.images[0], None  # Return image and no error
         else:
-            return "Error: No image returned from model."
+            return None, "Error: No image returned from model."  # Return no image and an error message
 
     except Exception as e:
         error_message = f"Error in process_image function: {e}"
         print(error_message)
-        return error_message
+        return None, error_message  # Return no image and the error message
 
 # Define the Gradio interface
 with gr.Blocks() as demo:
@@ -54,13 +54,14 @@ with gr.Blocks() as demo:
             upscale_input = gr.Slider(minimum=1, maximum=4, step=1, label="Upscale Factor")
             inpaint_input = gr.Checkbox(label="Enable Inpainting")
             
-            output = gr.Image(label="Generated Image", type="pil")
+            output_image = gr.Image(label="Generated Image", type="pil")
+            error_output = gr.Textbox(label="Error Details", lines=4, placeholder="Error details will appear here")
 
             generate_button = gr.Button("Generate Image")
             generate_button.click(
                 process_image,
                 inputs=[prompt_input, image_input, style_input, upscale_input, inpaint_input],
-                outputs=output
+                outputs=[output_image, error_output]  # Handle both image and error output
             )
 
 # Launch the interface