Update app.py

app.py

@@ -475,62 +475,63 @@ with gr.Blocks(css=css, title="LTX-2 Video Distilled 🎥🔈") as demo:
         </div>
         """
     )
-    with gr.Row():
-        with gr.Column(elem_id="step-column"):
-            input_image = gr.Image(
-                label="Input Image (Optional)",
-                type="pil",
-                height=512)
-
-            prompt = gr.Textbox(
-                label="Prompt",
-                value="Make this image come alive with cinematic motion, smooth animation",
-                lines=3,
-                max_lines=3,
-                placeholder="Describe the motion and animation you want..."
-            )
-            duration = gr.Slider(
-                label="Duration (seconds)",
-                minimum=1.0,
-                maximum=10.0,
-                value=3.0,
-                step=0.1
-            )
-            enhance_prompt = gr.Checkbox(
-                    label="Enhance Prompt",
-                    value=True,
-                    visible=False
-                )
-
-            with gr.Accordion("Advanced Settings", open=False):
-                seed = gr.Slider(
-                    label="Seed",
-                    minimum=0,
-                    maximum=MAX_SEED,
-                    value=DEFAULT_SEED,
-                    step=1
+    with gr.Column(elem_id="col-container"):
+        with gr.Row():
+            with gr.Column(elem_id="step-column"):
+                input_image = gr.Image(
+                    label="Input Image (Optional)",
+                    type="pil",
+                    height=512)
+    
+                prompt = gr.Textbox(
+                    label="Prompt",
+                    value="Make this image come alive with cinematic motion, smooth animation",
+                    lines=3,
+                    max_lines=3,
+                    placeholder="Describe the motion and animation you want..."
                 )
-
-                randomize_seed = gr.Checkbox(
-                    label="Randomize Seed",
-                    value=True
+                duration = gr.Slider(
+                    label="Duration (seconds)",
+                    minimum=1.0,
+                    maximum=10.0,
+                    value=3.0,
+                    step=0.1
                 )
-
-                with gr.Row():
-                    width = gr.Number(
-                        label="Width",
-                        value=DEFAULT_1_STAGE_WIDTH,
-                        precision=0
+                enhance_prompt = gr.Checkbox(
+                        label="Enhance Prompt",
+                        value=True,
+                        visible=False
                     )
-                    height = gr.Number(
-                        label="Height",
-                        value=DEFAULT_1_STAGE_HEIGHT,
-                        precision=0
+    
+                with gr.Accordion("Advanced Settings", open=False):
+                    seed = gr.Slider(
+                        label="Seed",
+                        minimum=0,
+                        maximum=MAX_SEED,
+                        value=DEFAULT_SEED,
+                        step=1
                     )
-
-        with gr.Column(elem_id="step-column"):
-            output_video = gr.Video(label="Generated Video", autoplay=True, height=512)
-            generate_btn = gr.Button("🤩 Generate Video", variant="primary", elem_classes="button-gradient")
+    
+                    randomize_seed = gr.Checkbox(
+                        label="Randomize Seed",
+                        value=True
+                    )
+    
+                    with gr.Row():
+                        width = gr.Number(
+                            label="Width",
+                            value=DEFAULT_1_STAGE_WIDTH,
+                            precision=0
+                        )
+                        height = gr.Number(
+                            label="Height",
+                            value=DEFAULT_1_STAGE_HEIGHT,
+                            precision=0
+                        )
+    
+            with gr.Column(elem_id="step-column"):
+                output_video = gr.Video(label="Generated Video", autoplay=True, height=512)
+                generate_btn = gr.Button("🤩 Generate Video", variant="primary", elem_classes="button-gradient")
 
     generate_btn.click(
         fn=generate_video,