Update app.py

app.py

@@ -5,7 +5,8 @@ from pathlib import Path
 current_dir = Path(__file__).parent
 sys.path.insert(0, str(current_dir / "packages" / "ltx-pipelines" / "src"))
 sys.path.insert(0, str(current_dir / "packages" / "ltx-core" / "src"))
-
+import numpy as np
+import random
 import spaces
 import gradio as gr
 from typing import Optional
@@ -23,6 +24,7 @@ from ltx_pipelines.constants import (
     DEFAULT_LORA_STRENGTH,
 )
 
+MAX_SEED = np.iinfo(np.int32).max
 # Custom negative prompt
 DEFAULT_NEGATIVE_PROMPT = "shaky, glitchy, low quality, worst quality, deformed, distorted, disfigured, motion smear, motion artifacts, fused fingers, bad anatomy, weird hand, ugly, transition, static"
 
@@ -129,9 +131,7 @@ def generate_video(
     """Generate a video based on the given parameters."""
     try:
         # Randomize seed if checkbox is enabled
-        if randomize_seed:
-            import random
-            seed = random.randint(0, 1000000)
+        current_seed = random.randint(0, MAX_SEED) if randomize_seed else int(seed)
 
         # Calculate num_frames from duration (using fixed 24 fps)
         frame_rate = 24.0
@@ -160,7 +160,7 @@ def generate_video(
             prompt=prompt,
             negative_prompt=negative_prompt,
             output_path=str(output_path),
-            seed=seed,
+            seed=current_seed,
             height=height,
             width=width,
             num_frames=num_frames,
@@ -171,7 +171,7 @@ def generate_video(
             tiling_config=TilingConfig.default(),
         )
 
-        return str(output_path)
+        return str(output_path), current_seed
 
     except Exception as e:
         import traceback
@@ -274,7 +274,7 @@ with gr.Blocks(title="LTX-2 Video 🎥🔈") as demo:
             height,
             width,
         ],
-        outputs=output_video
+        outputs=[output_video,seed]
     )
 
     # Add example
@@ -288,7 +288,7 @@ with gr.Blocks(title="LTX-2 Video 🎥🔈") as demo:
         ],
         fn=generate_video,
         inputs=[input_image, prompt, duration],
-        outputs = [output_video],
+        outputs = [output_video,seed],
         label="Example",
         cache_examples=True,
         cache_mode="lazy",