Update app.py

app.py

@@ -49,38 +49,73 @@ def generate_image(api_key, model, prompt, image_size, num_inference_steps, guid
 
 def update_visible_components(model):
     if model == "Flux Pro":
-        return [gr.update(visible=True), gr.update(visible=True), gr.update(visible=False)]
+        return [
+            gr.update(visible=True, value=28),
+            gr.update(visible=True, value=3.5),
+            gr.update(visible=True, value="2"),
+            gr.update(visible=False)
+        ]
     elif model == "Flux Dev":
-        return [gr.update(visible=True), gr.update(visible=False), gr.update(visible=True)]
+        return [
+            gr.update(visible=True, value=28),
+            gr.update(visible=True, value=3.5),
+            gr.update(visible=False),
+            gr.update(visible=True, value=True)
+        ]
     else:  # Flux Schnell
-        return [gr.update(visible=False), gr.update(visible=False), gr.update(visible=True)]
+        return [
+            gr.update(visible=True, value=4),
+            gr.update(visible=False),
+            gr.update(visible=False),
+            gr.update(visible=True, value=True)
+        ]
+
+with gr.Blocks(theme='bethecloud/storj_theme') as demo:
+    gr.HTML("""
+    <h1 align="center">Flux AI Image Generation</h1>
+    <p align="center">
+    <a href="https://www.fal.ai/" target="_blank">[FAL AI Website]</a>
+    <a href="https://github.com/fal-ai/fal" target="_blank">[FAL AI GitHub]</a>
+    </p>
+    """)
 
-with gr.Blocks() as demo:
-    gr.Markdown("# Flux Image Generation")
-    
-    api_key = gr.Textbox(type="password", label="FAL API Key")
-    
     with gr.Row():
-        model = gr.Dropdown(choices=["Flux Pro", "Flux Dev", "Flux Schnell"], label="Model", value="Flux Pro")
-        image_size = gr.Dropdown(choices=["square_hd", "square", "portrait_4_3", "portrait_16_9", "landscape_4_3", "landscape_16_9"], label="Image Size", value="landscape_4_3")
-    
-    prompt = gr.Textbox(label="Prompt", lines=3)
-    num_inference_steps = gr.Slider(1, 100, value=28, step=1, label="Number of Inference Steps")
-    guidance_scale = gr.Slider(0, 20, value=3.5, step=0.1, label="Guidance Scale")
-    num_images = gr.Slider(1, 10, value=1, step=1, label="Number of Images")
-    safety_tolerance = gr.Dropdown(choices=["1", "2", "3", "4", "5", "6"], label="Safety Tolerance", value="2")
-    enable_safety_checker = gr.Checkbox(label="Enable Safety Checker", value=True)
-    seed = gr.Number(label="Seed", value=-1)
+        with gr.Column(scale=1):
+            api_key = gr.Textbox(type="password", label="FAL API Key")
+            model = gr.Dropdown(
+                label="Model",
+                choices=["Flux Pro", "Flux Dev", "Flux Schnell"],
+                value="Flux Pro"
+            )
+            prompt = gr.Textbox(label="Prompt", lines=3, placeholder="Add your prompt here")
+            image_size = gr.Dropdown(
+                choices=["square_hd", "square", "portrait_4_3", "portrait_16_9", "landscape_4_3", "landscape_16_9"],
+                label="Image Size",
+                value="landscape_4_3"
+            )
+            
+            with gr.Accordion("Advanced settings", open=False):
+                num_inference_steps = gr.Slider(1, 100, 28, step=1, label="Number of Inference Steps")
+                guidance_scale = gr.Slider(0, 20, 3.5, step=0.1, label="Guidance Scale")
+                num_images = gr.Slider(1, 10, 1, step=1, label="Number of Images")
+                safety_tolerance = gr.Dropdown(choices=["1", "2", "3", "4", "5", "6"], label="Safety Tolerance", value="2")
+                enable_safety_checker = gr.Checkbox(label="Enable Safety Checker", value=True)
+                seed = gr.Number(label="Seed", value=-1)
+
+            generate_btn = gr.Button("Generate Image")
 
-    generate_button = gr.Button("Generate Images")
-    output_images = gr.Gallery(label="Generated Images")
+        with gr.Column(scale=1):
+            output_gallery = gr.Gallery(label="Generated Images", elem_id="gallery", show_label=False)
 
-    model.change(update_visible_components, inputs=[model], outputs=[guidance_scale, safety_tolerance, enable_safety_checker])
+    model.change(update_visible_components, inputs=[model], outputs=[num_inference_steps, guidance_scale, safety_tolerance, enable_safety_checker])
 
-    generate_button.click(
-        generate_image,
-        inputs=[api_key, model, prompt, image_size, num_inference_steps, guidance_scale, num_images, safety_tolerance, enable_safety_checker, seed],
-        outputs=output_images
+    generate_btn.click(
+        fn=generate_image,
+        inputs=[
+            api_key, model, prompt, image_size, num_inference_steps,
+            guidance_scale, num_images, safety_tolerance, enable_safety_checker, seed
+        ],
+        outputs=[output_gallery]
     )
 
 demo.launch()