Update app.py

app.py

@@ -27,7 +27,7 @@ def generate_image(api_key, prompt, image_size='landscape_4_3', num_images=1):
             images.append(img)
         return images
     except Exception as e:
-        return [f"Error: {e}"]
+        return [Image.new('RGB', (300, 300), color='red')]  # Return a red image on error
 
 with gr.Blocks() as demo:
     gr.Markdown("# FLUX1.1 [pro] Text-to-Image Generator")
@@ -43,7 +43,7 @@ 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")
+    output_gallery = gr.Gallery(label="Generated Images").style(columns=2, rows=2, height="auto")
 
     generate_btn.click(
         fn=generate_image,
@@ -51,4 +51,5 @@ with gr.Blocks() as demo:
         outputs=output_gallery
     )
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()