Update app.py

app.py

@@ -122,7 +122,8 @@ def process_video_for_canny(video, width, height):
      
     return canny_video
 
-def process_input_video(reference_video, width, height):
+def process_input_video(reference_video, width, height,
+    progress=gr.Progress(track_tqdm=True)):
     """
     Process the input video for canny edges and return both processed video and preview.
     """
@@ -173,7 +174,7 @@ def generate_video(
     seed=0,
     randomize_seed=False,
     control_type="canny",
-    progress=gr.Progress()
+    progress=gr.Progress(track_tqdm=True)
 ):
     try:
         # Initialize models if needed
@@ -195,7 +196,6 @@ def generate_video(
         temporal_compression = pipeline.vae_temporal_compression_ratio
         num_frames = ((num_frames - 1) // temporal_compression) * temporal_compression + 1
         
-        progress(0.1, desc="Preparing processed video...")
         
         # Use pre-processed video frames if available (for canny), otherwise process on-demand
         print("######## control_video ", control_video)
@@ -213,7 +213,6 @@ def generate_video(
         processed_video = read_video(processed_video)
         print(type(processed_video))
         
-        progress(0.2, desc="Preparing generation parameters...")
         
         # Calculate downscaled dimensions
         downscale_factor = 2 / 3
@@ -223,7 +222,6 @@ def generate_video(
             downscaled_height, downscaled_width, pipeline.vae_temporal_compression_ratio
         )
         
-        progress(0.3, desc="Generating video at lower resolution...")
         
         # 1. Generate video at smaller resolution
         latents = pipeline(
@@ -241,7 +239,6 @@ def generate_video(
             output_type="latent",
         ).frames
 
-        progress(0.6, desc="Upscaling video...")
         
         # 2. Upscale generated video
         upscaled_height, upscaled_width = downscaled_height * 2, downscaled_width * 2
@@ -250,7 +247,6 @@ def generate_video(
             output_type="latent"
         ).frames
 
-        progress(0.8, desc="Final denoising and processing...")
         
         # 3. Denoise the upscaled video
         final_video_frames_np = pipeline(
@@ -270,7 +266,6 @@ def generate_video(
             output_type="np",
         ).frames[0]
 
-        progress(0.9, desc="Finalizing output...")
         
 
         # Export to temporary file
@@ -281,8 +276,7 @@ def generate_video(
                 progress((frame_idx + 1) / len(video_uint8_frames), desc="Encoding video frames...")
                 writer.append_data(frame_data)
 
-        
-        progress(1.0, desc="Complete!")
+
         
         return output_filename, seed