Initial Commit

app.py

@@ -197,41 +197,104 @@ class FluxInterface:
 interface = FluxInterface()
 
 # Create Gradio interface
-with gr.Blocks(title="Qwen2vl-Flux Demo") as demo:
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("""
     # 🎨 Qwen2vl-Flux Image Variation Demo
     Upload an image and get AI-generated variations. You can optionally add a text prompt to guide the generation.
     """)
     
     with gr.Row():
-        with gr.Column():
-            input_image = gr.Image(label="Upload Image", type="pil")
-            prompt = gr.Textbox(label="Optional Text Prompt(should be as long as possible)", placeholder="Enter text prompt here (optional)")
+        with gr.Column(scale=1):
+            input_image = gr.Image(
+                label="Upload Image",
+                type="pil",
+                height=384,
+                width=384,
+                tool="select"
+            )
+            prompt = gr.Textbox(
+                label="Optional Text Prompt",
+                placeholder="Enter text prompt here (optional)",
+                lines=2
+            )
             
-            with gr.Row():
-                guidance = gr.Slider(minimum=1, maximum=10, value=3.5, label="Guidance Scale")
-                steps = gr.Slider(minimum=1, maximum=50, value=28, label="Number of Steps")
-                num_images = gr.Slider(minimum=1, maximum=4, value=2, step=1, label="Number of Images")
+            with gr.Group():
+                with gr.Row(equal_height=True):
+                    with gr.Column(scale=1):
+                        guidance = gr.Slider(
+                            minimum=1,
+                            maximum=10,
+                            value=3.5,
+                            step=0.5,
+                            label="Guidance Scale",
+                            info="Higher values follow prompt more closely"
+                        )
+                    with gr.Column(scale=1):
+                        steps = gr.Slider(
+                            minimum=1,
+                            maximum=50,
+                            value=28,
+                            step=1,
+                            label="Steps",
+                            info="More steps = better quality but slower"
+                        )
+                
+                with gr.Row(equal_height=True):
+                    with gr.Column(scale=1):
+                        num_images = gr.Slider(
+                            minimum=1,
+                            maximum=4,
+                            value=2,
+                            step=1,
+                            label="Number of Images",
+                            info="Generate multiple variations"
+                        )
+                    with gr.Column(scale=1):
+                        seed = gr.Number(
+                            label="Random Seed",
+                            value=None,
+                            precision=0,
+                            info="Optional, for reproducibility"
+                        )
             
-            seed = gr.Number(label="Random Seed (optional)", precision=0)
-            submit_btn = gr.Button("Generate Variations", variant="primary")
+            submit_btn = gr.Button(
+                "Generate Variations",
+                variant="primary",
+                scale=1
+            )
             
-        with gr.Column():
-            output_gallery = gr.Gallery(label="Generated Variations", columns=2, show_label=True)
+        with gr.Column(scale=1):
+            output_gallery = gr.Gallery(
+                label="Generated Variations",
+                columns=2,
+                rows=2,
+                height=768,
+                object_fit="contain",
+                show_label=True
+            )
     
+    gr.Markdown("""
+    ### Tips:
+    - Upload any image to get started
+    - Add a text prompt to guide the generation in a specific direction
+    - Adjust guidance scale to control how closely the output follows the prompt
+    - Increase steps for higher quality (but slower) generation
+    - Use the same seed to reproduce results
+    """)
+
     # Set up the generation function
     submit_btn.click(
         fn=interface.generate,
-        inputs=[input_image, prompt, guidance, steps, num_images, seed],
-        outputs=output_gallery,
+        inputs=[
+            input_image,
+            prompt,
+            guidance,
+            steps,
+            num_images,
+            seed
+        ],
+        outputs=output_gallery
     )
-    
-    gr.Markdown("""
-    ### Notes:
-    - Higher guidance scale values result in outputs that more closely follow the prompt
-    - More steps generally produce better quality but take longer
-    - Set a seed for reproducible results
-    """)
 
 # Launch the app
 if __name__ == "__main__":