Update app.py

app.py

@@ -4,11 +4,12 @@ import requests
 from PIL import Image
 from io import BytesIO
 import traceback
+import os
 
 def generate_image(api_key, prompt, image_size='landscape_4_3', num_images=1):
     try:
-        # Set the API key for the fal_client
-        fal_client.api_key = api_key
+        # Set the API key as an environment variable
+        os.environ['FAL_KEY'] = api_key
 
         handler = fal_client.submit(
             "fal-ai/flux-pro/v1.1",
@@ -26,13 +27,12 @@ def generate_image(api_key, prompt, image_size='landscape_4_3', num_images=1):
             response = requests.get(img_url)
             img = Image.open(BytesIO(response.content))
             images.append(img)
-        return images
+        return [gr.update(value=images, visible=True), gr.update(visible=False)]
     except Exception as e:
         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
+        return [gr.update(visible=False), gr.update(value=error_msg, visible=True)]
 
-# Update the Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# FLUX1.1 [pro] Text-to-Image Generator")
     with gr.Row():