Update app.py

app.py

@@ -89,68 +89,79 @@ css = """
 """
 
 with gr.Blocks(css=css, theme=gr.themes.Soft(primary_hue="violet")) as demo:
-    with gr.Column(elem_id="col-container"):
-        gr.Markdown("## FIBO")
+    gr.Markdown("## FIBO")
 
-        with gr.Row():
-            with gr.Tab("generate") as tab_generate:
-                with gr.Row():
-                    prompt_generate = gr.Textbox(label="Prompt", placeholder="a man holding a goose screaming")
+    # Two-column layout
+    with gr.Row(elem_id="col-container"):
+        # ==== LEFT COLUMN ====
+        with gr.Column():
+            with gr.Row():
+                with gr.Tab("generate") as tab_generate:
+                    prompt_generate = gr.Textbox(
+                        label="Prompt",
+                        placeholder="a man holding a goose screaming"
+                    )
 
-            with gr.Tab("refine") as tab_refine:
-                with gr.Row():
-                    prompt_refine = gr.Textbox(label="Prompt", info="describe the change you want to make", placeholder="make the cat white")
+                with gr.Tab("refine") as tab_refine:
+                    prompt_refine = gr.Textbox(
+                        label="Prompt",
+                        info="describe the change you want to make",
+                        placeholder="make the cat white"
+                    )
 
-            with gr.Tab("inspire") as tab_inspire:
-                with gr.Row():
+                with gr.Tab("inspire") as tab_inspire:
                     prompt_inspire_image = gr.Image(
                         label="Inspiration Image",
                         type="pil",
                     )
-            
-        submit_btn = gr.Button("Generate", variant="primary")
-        result = gr.Image(label="output")
-        with gr.Accordion("Structured Prompt", open=False) as structured_accordion:
-            prompt_in_json = gr.JSON(label="json structured prompt")
-        with gr.Accordion("Advanced Settings", open=False):
-            with gr.Row():
-                seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0)
-                randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
-            with gr.Row():
-                guidance_scale = gr.Slider(label="guidance scale", minimum=1.0, maximum=10.0, step=0.1, value=5.0)
-                num_inference_steps = gr.Slider(
-                    label="number of inference steps", minimum=1, maximum=60, step=1, value=50
-                )
-                height = gr.Slider(label="Height", minimum=768, maximum=1248, step=32, value=1024)
-                width = gr.Slider(label="Width", minimum=832, maximum=1344, step=64, value=1024)
-            with gr.Row():
-                negative_prompt = gr.Textbox(label="negative prompt")
-                negative_prompt_json = gr.JSON(label="json negative prompt")
-
-        # Track active tab
-        current_mode = gr.State("generate")
-
-        tab_generate.select(lambda: "generate", outputs=current_mode)
-        tab_refine.select(lambda: "refine", outputs=current_mode)
-        tab_inspire.select(lambda: "inspire", outputs=current_mode)
-
-        submit_btn.click(
-            fn=infer,
-            inputs=[
-                prompt_generate,
-                prompt_refine,
-                prompt_inspire_image,
-                prompt_in_json,
-                negative_prompt,
-                seed,
-                randomize_seed,
-                width,
-                height,
-                guidance_scale,
-                num_inference_steps,
-                current_mode,
-            ],
-            outputs=[result, seed, prompt_in_json, negative_prompt_json, structured_accordion],
-        )
+
+            submit_btn = gr.Button("Generate", variant="primary")
+
+            with gr.Accordion("Advanced Settings", open=False):
+                with gr.Row():
+                    seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0)
+                    randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
+                with gr.Row():
+                    guidance_scale = gr.Slider(label="guidance scale", minimum=1.0, maximum=10.0, step=0.1, value=5.0)
+                    num_inference_steps = gr.Slider(
+                        label="number of inference steps", minimum=1, maximum=60, step=1, value=50
+                    )
+                    height = gr.Slider(label="Height", minimum=768, maximum=1248, step=32, value=1024)
+                    width = gr.Slider(label="Width", minimum=832, maximum=1344, step=64, value=1024)
+                with gr.Row():
+                    negative_prompt = gr.Textbox(label="negative prompt")
+                    negative_prompt_json = gr.JSON(label="json negative prompt")
+
+        # ==== RIGHT COLUMN ====
+        with gr.Column():
+            result = gr.Image(label="output")
+            with gr.Accordion("Structured Prompt", open=False) as structured_accordion:
+                prompt_in_json = gr.JSON(label="json structured prompt")
+
+    # Track active tab
+    current_mode = gr.State("generate")
+    tab_generate.select(lambda: "generate", outputs=current_mode)
+    tab_refine.select(lambda: "refine", outputs=current_mode)
+    tab_inspire.select(lambda: "inspire", outputs=current_mode)
+
+    submit_btn.click(
+        fn=infer,
+        inputs=[
+            prompt_generate,
+            prompt_refine,
+            prompt_inspire_image,
+            prompt_in_json,
+            negative_prompt,
+            seed,
+            randomize_seed,
+            width,
+            height,
+            guidance_scale,
+            num_inference_steps,
+            current_mode,
+        ],
+        outputs=[result, seed, prompt_in_json, negative_prompt_json, structured_accordion],
+    )
 
 demo.queue().launch()
+