Update app.py

app.py

@@ -68,71 +68,66 @@ def process_video(
     manual_height,
     manual_width,
     guidance_scale,
-    progress=gr.Progress()
+    progress=gr.Progress(track_tqdm=True)
 ):
-    """Process the video with Lucy Edit"""
-    try:
-        # Load and preprocess video
-        progress(0.2, desc="Loading video...")
+    # Load and preprocess video
+    progress(0.2, desc="Loading video...")
+    
+    # Get video dimensions
+    temp_video = load_video(video_path)
+    print(len(temp_video))
+    if temp_video and len(temp_video) > 0:
+        original_width, original_height = temp_video[0].size
         
-        # Get video dimensions
-        temp_video = load_video(video_path)
-        if temp_video and len(temp_video) > 0:
-            original_width, original_height = temp_video[0].size
-            
-            # Calculate dimensions
-            if auto_resize:
-                width, height = calculate_resolution(original_width, original_height)
-                gr.Info(f"Auto-resized from {original_width}x{original_height} to {width}x{height} (preserving aspect ratio)")
-            else:
-                width, height = manual_width, manual_height
-                if abs((original_width/original_height) - (width/height)) > 0.1:
-                    gr.Warning(f"Output aspect ratio ({width}x{height}) differs significantly from input ({original_width}x{original_height}). Video may appear stretched.")
+        # Calculate dimensions
+        if auto_resize:
+            width, height = calculate_resolution(original_width, original_height)
+            gr.Info(f"Auto-resized from {original_width}x{original_height} to {width}x{height} (preserving aspect ratio)")
         else:
-            raise ValueError("Could not load video or video is empty")
-        
-        # Convert video function
-        def convert_video(video: List[Image.Image]) -> List[Image.Image]:
-            # Ensure we don't exceed the video length
-            frames_to_load = min(len(video), num_frames)
-            video_frames = video[:frames_to_load]
-            # Resize frames
-            video_frames = [frame.resize((width, height)) for frame in video_frames]
-            return video_frames
-        
-        # Load video from file path
-        video = load_video(video_path, convert_method=convert_video)
-        
-        # Ensure we have the right number of frames
-        if len(video) < num_frames:
-            gr.Warning(f"Video has only {len(video)} frames, using all available frames.")
-            num_frames = len(video)
-        
-        # Generate edited video
-        progress(0.5, desc="Generating edited video...")
-        output = pipe(
-            prompt=prompt,
-            video=video,
-            negative_prompt=negative_prompt,
-            height=height,
-            width=width,
-            num_frames=num_frames,
-            guidance_scale=guidance_scale,
-        ).frames[0]
-        
-        # Export to temporary file
-        progress(0.9, desc="Exporting video...")
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".mp4") as tmp_file:
-            output_path = tmp_file.name
-        
-        export_to_video(output, output_path, fps=24)
-        
-        progress(1.0, desc="Complete!")
-        return output_path
-        
-    except Exception as e:
-        gr.Error(f"An error occurred: {str(e)}")
-        return None
+            width, height = manual_width, manual_height
+            if abs((original_width/original_height) - (width/height)) > 0.1:
+                gr.Warning(f"Output aspect ratio ({width}x{height}) differs significantly from input ({original_width}x{original_height}). Video may appear stretched.")
+    else:
+        raise gr.Error("Could not load video or video is empty")
+    
+    # Convert video function
+    def convert_video(video: List[Image.Image]) -> List[Image.Image]:
+        # Ensure we don't exceed the video length
+        frames_to_load = min(len(video), num_frames)
+        video_frames = video[:frames_to_load]
+        # Resize frames
+        video_frames = [frame.resize((width, height)) for frame in video_frames]
+        return video_frames
+    
+    # Load video from file path
+    video = load_video(video_path, convert_method=convert_video)
+    
+    # Ensure we have the right number of frames
+    if len(video) < num_frames:
+        gr.Warning(f"Video has only {len(video)} frames, using all available frames.")
+        num_frames = len(video)
+    
+    # Generate edited video
+    progress(0.5, desc="Generating edited video...")
+    output = pipe(
+        prompt=prompt,
+        video=video,
+        negative_prompt=negative_prompt,
+        height=height,
+        width=width,
+        num_frames=num_frames,
+        guidance_scale=guidance_scale,
+    ).frames[0]
+    
+    # Export to temporary file
+    progress(0.9, desc="Exporting video...")
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".mp4") as tmp_file:
+        output_path = tmp_file.name
+    
+    export_to_video(output, output_path, fps=24)
+    
+    progress(1.0, desc="Complete!")
+    return output_path
 
 # Create Gradio interface
 with gr.Blocks(title="Lucy Edit - Video Editing with Text") as demo: