Update app.py

app.py

@@ -112,6 +112,7 @@ def generate_video(
     input_image,
     prompt: str,
     duration: float,
+    enhance_prompt: bool = True,
     seed: int = 42,
     randomize_seed: bool = True,
     height: int = DEFAULT_HEIGHT,
@@ -164,7 +165,7 @@ def generate_video(
             
             result = text_encoder_client.predict(
                 prompt=prompt,
-                enhance_prompt=True,
+                enhance_prompt=enhance_prompt,
                 input_image=image_input,
                 seed=current_seed,
                 api_name="/encode_prompt"
@@ -225,14 +226,18 @@ with gr.Blocks(title="LTX-2 Video Distilled 🎥🔈") as demo:
                 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
-            )
+            with gr.Row():
+                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
+                    )
 
             generate_btn = gr.Button("Generate Video", variant="primary", size="lg")
 
@@ -271,6 +276,7 @@ with gr.Blocks(title="LTX-2 Video Distilled 🎥🔈") as demo:
             input_image,
             prompt,
             duration,
+            enhance_prompt,
             seed,
             randomize_seed,
             height,
@@ -307,4 +313,4 @@ css = '''
 .gradio-container .contain{max-width: 1200px !important; margin: 0 auto !important}
 '''
 if __name__ == "__main__":
-    demo.launch(theme=gr.themes.Citrus(), css=css, share=True)
+    demo.launch(theme=gr.themes.Citrus(), css=css)