Update app.py

app.py

@@ -53,15 +53,15 @@ def update_progress(progress: PipelineProgress) -> None:
 
 
 def process_video(
-    video_file: str,
-    api_key: str,
-    domain: str,
-    num_clips: int,
-    clip_duration: float,
-    reference_image: Optional[str],
-    custom_prompt: str,
-    progress: gr.Progress = gr.Progress(),
-) -> Tuple[List[str], str, str]:
+    video_file,
+    api_key,
+    domain,
+    num_clips,
+    clip_duration,
+    reference_image,
+    custom_prompt,
+    progress=None,
+):
     """
     Main processing function for Gradio interface.
 
@@ -84,14 +84,21 @@ def process_video(
 
     log_messages = []
 
-    def log(msg: str) -> None:
+    def log(msg):
         log_messages.append(f"[{time.strftime('%H:%M:%S')}] {msg}")
         logger.info(msg)
 
+    def update_prog(val, desc=""):
+        if progress is not None:
+            try:
+                progress(val, desc=desc)
+            except:
+                pass
+
     try:
         # Validate inputs
         log("Validating inputs...")
-        progress(0.02, desc="Validating inputs...")
+        update_prog(0.02, "Validating inputs...")
 
         if not video_file:
             return [], "❌ Error: No video file provided", "\n".join(log_messages)
@@ -130,12 +137,12 @@ def process_video(
         log(f"Output directory: {output_dir}")
 
         # Initialize pipeline with progress callback
-        def progress_callback(p: PipelineProgress):
+        def progress_callback(p):
             update_progress(p)
-            progress(p.progress, desc=p.message)
+            update_prog(p.progress, p.message)
 
         log("Initializing pipeline...")
-        progress(0.05, desc="Initializing pipeline...")
+        update_prog(0.05, "Initializing pipeline...")
 
         _current_pipeline = PipelineOrchestrator(
             progress_callback=progress_callback
@@ -338,16 +345,19 @@ def create_interface() -> gr.Blocks:
         # Event handlers
         def on_process(video, api_key, domain, num_clips, duration, ref_img, prompt, progress=gr.Progress()):
             """Handle process button click."""
-            clips, status, logs = process_video(
-                video, api_key, domain, num_clips, duration, ref_img, prompt, progress
-            )
+            try:
+                clips, status, logs = process_video(
+                    video, api_key, domain, num_clips, duration, ref_img, prompt, progress
+                )
 
-            # Convert clip paths to gallery format
-            gallery_items = []
-            for clip_path in clips:
-                gallery_items.append((clip_path, f"Clip {len(gallery_items)+1}"))
+                # Convert clip paths to gallery format
+                gallery_items = []
+                for clip_path in clips:
+                    gallery_items.append((clip_path, f"Clip {len(gallery_items)+1}"))
 
-            return status, gallery_items, logs
+                return status, gallery_items, logs
+            except Exception as e:
+                return f"❌ Error: {str(e)}", [], f"Error: {str(e)}"
 
         process_btn.click(
             fn=on_process,
@@ -361,7 +371,6 @@ def create_interface() -> gr.Blocks:
                 custom_prompt,
             ],
             outputs=[status_output, clip_gallery, log_output],
-            show_progress=True,
         )
 
     return demo