Update app.py

app.py

@@ -288,8 +288,56 @@ def generate_video(
     width: int = DEFAULT_1_STAGE_WIDTH ,
     progress=gr.Progress(track_tqdm=True)
 ):
+    """
+    Generate a short cinematic video from a text prompt and optional input image using the LTX-2 distilled pipeline.
+
+    This function performs text-to-video or image-to-video generation by:
+    1. Optionally enhancing the input prompt using a Gemma-based text encoder.
+    2. Encoding the prompt into video and audio conditioning embeddings.
+    3. Running the LTX-2 distilled DiT-based video generation pipeline with optional image conditioning.
+    4. Producing an MP4 video file at the requested resolution, duration, and random seed.
+
+    Args:
+        input_image (PIL.Image.Image | None):
+            Optional input image used for image-to-video generation. If provided,
+            the image is injected at the first frame with full strength to guide motion.
+        prompt (str):
+            Text description of the scene, motion, and cinematic style to generate.
+        duration (float):
+            Desired video length in seconds. Internally converted to number of frames
+            using a fixed frame rate (24 FPS).
+        enhance_prompt (bool, optional):
+            Whether to enhance the prompt using an AI prompt enhancer before encoding.
+            Defaults to True.
+        seed (int, optional):
+            Base random seed for reproducibility. Ignored if `randomize_seed` is True.
+            Defaults to 42.
+        randomize_seed (bool, optional):
+            If True, a random seed is generated for each run. Defaults to True.
+        height (int, optional):
+            Output video height in pixels. Defaults to the model’s 1-stage height.
+        width (int, optional):
+            Output video width in pixels. Defaults to the model’s 1-stage width.
+        progress (gr.Progress, optional):
+            Gradio progress tracker for visualizing generation progress.
 
-    """Generate a video based on the given parameters."""
+    Returns:
+        tuple:
+            - str: Path to the generated MP4 video file.
+            - int: The seed used for the generation.
+
+    Notes:
+        - The function uses a fixed frame rate of 24 FPS.
+        - Prompt embeddings are generated externally and passed directly to the pipeline
+          to avoid reloading the text encoder.
+        - GPU memory is explicitly cleared after generation to reduce VRAM pressure.
+        - If an input image is provided, it is temporarily saved to disk for processing.
+
+    Raises:
+        Exception:
+            Any errors during prompt encoding, embedding loading, or video generation
+            are caught and printed with a full traceback for debugging.
+    """
     try:
         # Randomize seed if checkbox is enabled
         current_seed = random.randint(0, MAX_SEED) if randomize_seed else int(seed)
@@ -578,4 +626,4 @@ with gr.Blocks(css=css, title="LTX-2 Video Distilled 🎥🔈") as demo:
 
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(ssr_mode=False, mcp_server=True)