Update app.py

app.py

@@ -3,6 +3,7 @@ import fal_client
 import requests
 from PIL import Image
 from io import BytesIO
+import traceback
 
 def generate_image(api_key, prompt, image_size='landscape_4_3', num_images=1):
     try:
@@ -27,9 +28,11 @@ def generate_image(api_key, prompt, image_size='landscape_4_3', num_images=1):
             images.append(img)
         return images
     except Exception as e:
-        print(f"Error: {e}")
-        return [Image.new('RGB', (300, 300), color='red') for _ in range(num_images)]  # Return red images on error
+        error_msg = f"Error: {str(e)}\n\nTraceback:\n{traceback.format_exc()}"
+        print(error_msg)
+        return [gr.update(visible=True), error_msg]  # Return error message to be displayed
 
+# Update the Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# FLUX1.1 [pro] Text-to-Image Generator")
     with gr.Row():
@@ -45,11 +48,12 @@ with gr.Blocks() as demo:
         num_images = gr.Slider(label="Number of Images", minimum=1, maximum=4, step=1, value=1)
     generate_btn = gr.Button("Generate Image")
     output_gallery = gr.Gallery(label="Generated Images", columns=2, rows=2)
+    error_output = gr.Textbox(label="Error Message", visible=False)
 
     generate_btn.click(
         fn=generate_image,
         inputs=[api_key, prompt, image_size, num_images],
-        outputs=output_gallery
+        outputs=[output_gallery, error_output]
     )
 
 if __name__ == "__main__":