ui: input top-left, floorplan top-right, output image bottom

app.py

@@ -210,16 +210,16 @@ with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column(scale=1):
             image_input = gr.Image(label="Upload a room image", type="numpy")
+        with gr.Column(scale=1):
             floor_plan = FloorPlan(
                 value=DEFAULT_FLOOR_PLAN,
                 label="Floor Plan",
                 interactive=True,
             )
-            generate_button = gr.Button("Generate New Image based on Floor plan", variant="primary")
-        with gr.Column(scale=1):
-            topview_before = gr.Image(label="Top View — Original", interactive=False)
-            topview_after = gr.Image(label="Top View — New Layout (Step A)", interactive=False)
-            image_output = gr.Image(label="Final Photorealistic Output (Step B)", interactive=False)
+    generate_button = gr.Button("Generate New Image based on Floor plan", variant="primary")
+    topview_before = gr.Image(visible=False)
+    topview_after = gr.Image(visible=False)
+    image_output = gr.Image(label="Final Photorealistic Output", interactive=False)
 
     # Wire it: input -> [floor_plan, initial_layout_state, topview_state, topview_before]
     image_input.change(