main

gradio_app.py

@@ -159,7 +159,7 @@ def infer(prompt, sample_image, recraft_model, seed=0):
         logger.error("Models not loaded. Please load the models first.")
         return None
 
-    model_path = model_paths[selected_model]
+    model_path = model_paths[recraft_model]
     frame_num = model_path['Frame']
 
     logger.info(f"Started generating image with prompt: {prompt}")
@@ -281,35 +281,44 @@ def infer(prompt, sample_image, recraft_model, seed=0):
 # Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("## FLUX Image Generation")
-    
-    with gr.Row():
-        # Dropdown for selecting the recraft model
-        recraft_model = gr.Dropdown(
-            label="Select Recraft Model", 
-            choices=["Wood Sculpture", "LEGO", "Sketch", "Portrait"], 
-            value="Wood Sculpture"
-        )
-
-        # Input for the prompt
-        prompt = gr.Textbox(label="Prompt", placeholder="Enter your prompt here", lines=1)
-
-        # File upload for image
-        sample_image = gr.Image(label="Upload a Conditional Image", type="pil")
-
-        # Seed
-        seed = gr.Slider(0, np.iinfo(np.int32).max, step=1, label="Seed", value=0)
 
-        # Load Model Button
-        load_button = gr.Button("Load Model")
-
-        # Status message box
-        status_box = gr.Textbox(label="Status", placeholder="Model loading status", interactive=False, value="Model not loaded")
+    with gr.Row():
+        with gr.Column(scale=1):
+            # Dropdown for selecting the recraft model
+            recraft_model = gr.Dropdown(
+                label="Select Recraft Model", 
+                choices=["Wood Sculpture", "LEGO", "Sketch", "Portrait"], 
+                value="Wood Sculpture"
+            )
+        
+            # Load Model Button
+            load_button = gr.Button("Load Model")
 
-        # Run Button
-        run_button = gr.Button("Generate Image")
+        with gr.Column(scale=1):
+            # Status message box
+            status_box = gr.Textbox(label="Status", placeholder="Model loading status", interactive=False, value="Model not loaded")
+    
+    with gr.Row():
+        with gr.Column(scale=2):
+            # Input for the prompt
+            prompt = gr.Textbox(label="Prompt", placeholder="Enter your prompt here", lines=1)
 
-        # Output result
-        result_image = gr.Image(label="Generated Image")
+        with gr.Column(scale=1):
+            # Seed
+            seed = gr.Slider(0, np.iinfo(np.int32).max, step=1, label="Seed", value=0)
+    
+    with gr.Row():
+        with gr.Column(scale=2):
+            # File upload for image
+            sample_image = gr.Image(label="Upload a Conditional Image", type="pil")
+        
+        with gr.Column(scale=1):
+            # Run Button
+            run_button = gr.Button("Generate Image")
+
+        with gr.Column(scale=1):
+            # Output result
+            result_image = gr.Image(label="Generated Image")
 
     # Load model button action
     load_button.click(fn=load_target_model, inputs=[recraft_model], outputs=[status_box])