better?

app.py

@@ -90,92 +90,95 @@ with gr.Blocks(css=css) as demo:
     with gr.Sidebar():
         gr.Markdown("# Inference Provider")
         gr.Markdown(
-            "This Space showcases the black-forest-labs/FLUX.1-dev model, served by the nebius API. Sign in with your Hugging Face account to use this API."
+            "This Space showcases the black‑forest‑labs/FLUX.1‑dev model, served by the nebius API. Sign in with your Hugging Face account to use this API."
         )
-        button = gr.LoginButton("Sign in")
-        button.click(fn=get_token, inputs=[], outputs=[])
+        gr.LoginButton("Sign in").click(fn=get_token, inputs=[], outputs=[])
 
-    with gr.Row():
-        
-        with gr.Tab("Generate Image"):
-                gr.Markdown(
-                    """# FLUX.1 [schnell] with fal‑ai through HF Inference Providers ⚡\nLearn more about HF Inference Providers [here](https://huggingface.co/docs/inference-providers/index)"""
-                )
+    with gr.Column(elem_id="col-container"):
+        gr.Markdown(
+            """# FLUX.1 [schnell] with fal‑ai through HF Inference Providers ⚡\nLearn more about HF Inference Providers [here](https://huggingface.co/docs/inference-providers/index)"""
+        )
 
-                with gr.Row():
-                    prompt = gr.Text(
-                        label="Prompt",
-                        show_label=False,
-                        max_lines=1,
-                        placeholder="Enter your prompt",
-                        container=False,
-                    )
-                    run_button = gr.Button("Run", scale=0)
-
-                result = gr.Image(label="Result", show_label=False, format="png")
-
-                with gr.Accordion("Advanced Settings", open=False):
-                    seed = gr.Slider(
-                        label="Seed",
-                        minimum=0,
-                        maximum=MAX_SEED,
-                        step=1,
-                        value=42,
-                    )
-                    with gr.Row():
-                        width = gr.Slider(
-                            label="Width",
-                            minimum=256,
-                            maximum=MAX_IMAGE_SIZE,
-                            step=32,
-                            value=1024,
-                        )
-                        height = gr.Slider(
-                            label="Height",
-                            minimum=256,
-                            maximum=MAX_IMAGE_SIZE,
-                            step=32,
-                            value=1024,
-                        )
-                    num_inference_steps = gr.Slider(
-                        label="Number of inference steps",
-                        minimum=1,
-                        maximum=50,
-                        step=1,
-                        value=25,
-                    )
-
-                gr.Examples(
-                    examples=examples,
-                    fn=generate,
-                    inputs=[prompt],
-                    outputs=[result, seed],
-                    cache_examples="lazy",
-                )
+        with gr.Row():
+            prompt = gr.Text(
+                label="Prompt",
+                show_label=False,
+                max_lines=1,
+                placeholder="Enter your prompt",
+                container=False,
+            )
+            run_button = gr.Button("Run", scale=0)
+
+        result = gr.Image(label="Result", show_label=False, format="png")
+        download_btn = gr.DownloadButton(
+            label="Download result",
+            visible=False,  
+            value=None,
+            variant="primary",
+        )
 
-                gr.on(
-                    triggers=[run_button.click, prompt.submit],
-                    fn=generate,
-                    inputs=[prompt, seed, width, height, num_inference_steps],
-                    outputs=[result, seed],
-                    show_api=True,
-                )
+        seed_number = gr.Number(label="Seed", precision=0, value=42, interactive=False)
 
-        with gr.Tab("Download Image"):
-            gr.Markdown("## Download an image from a URL and make it available as a file")
-            url_input = gr.Text(label="Image URL", placeholder="https://example.com/cool-image.png")
-            filename_input = gr.Text(
-                label="Save as (filename)", value="downloaded_image.png", placeholder="downloaded_image.png"
+        with gr.Accordion("Advanced Settings", open=False):
+            seed_slider = gr.Slider(
+                label="Seed",
+                minimum=0,
+                maximum=MAX_SEED,
+                step=1,
+                value=42,
             )
-            download_btn = gr.Button("Download")
-            file_output = gr.File(label="Downloaded file")
-
-            download_btn.click(
-                fn=download_image_locally,
-                inputs=[url_input, filename_input],
-                outputs=[file_output],
-                show_api=True,
+            with gr.Row():
+                width_slider = gr.Slider(
+                    label="Width",
+                    minimum=256,
+                    maximum=MAX_IMAGE_SIZE,
+                    step=32,
+                    value=1024,
+                )
+                height_slider = gr.Slider(
+                    label="Height",
+                    minimum=256,
+                    maximum=MAX_IMAGE_SIZE,
+                    step=32,
+                    value=1024,
+                )
+            steps_slider = gr.Slider(
+                label="Number of inference steps",
+                minimum=1,
+                maximum=50,
+                step=1,
+                value=25,
             )
 
+        gr.Examples(
+            examples=examples,
+            fn=generate,
+            inputs=[prompt],
+            outputs=[result, seed_number, download_btn],
+            cache_examples="lazy",
+        )
+
+        run_button.click(
+            fn=generate,
+            inputs=[prompt, seed_slider, width_slider, height_slider, steps_slider],
+            outputs=[result, seed_number, download_btn],
+            show_api=True,  # expose as MCP tool
+        )
+
+        result.change(
+            lambda _: gr.DownloadButton.update(visible=True),
+            inputs=[result],
+            outputs=[download_btn],
+        )
+
+        gr.Interface(
+            fn=download_image_locally,
+            inputs=[gr.Text(label="Image URL"), gr.Text(label="Filename")],
+            outputs=[gr.File()],
+            show_api=True,
+            render=False,  # do not show in UI
+        )
+
+
 if __name__ == "__main__":
-    demo.launch(mcp_server=True)
+    demo.launch(mcp_server=True)