.

app.py

@@ -91,11 +91,6 @@ def create_ui(examples_path: str):
                     video_description = gr.Markdown("", elem_id="video_desc")
                     highlight_types = gr.Markdown("", elem_id="highlight_types")
 
-        def progress_callback(current, total):
-            """Callback to update progress percentage"""
-            percentage = int((current / total) * 100)
-            return f"Processing segments... {percentage}% complete"
-
         @spaces.GPU
         def on_process(video):
             if not video:
@@ -120,6 +115,11 @@ def create_ui(examples_path: str):
                     ]
                     return
 
+                current_status = ""
+                def progress_callback(current, total):
+                    nonlocal current_status
+                    current_status = f"Processing segments... {int((current/total) * 100)}% complete"
+
                 # Make accordion visible as soon as processing starts
                 yield [
                     "Loading model...",
@@ -130,8 +130,12 @@ def create_ui(examples_path: str):
                 ]
 
                 model, processor = load_model()
-                detector = BatchedVideoHighlightDetector(model, processor, batch_size=8, progress_callback=lambda current, total: print(f"Progress: {current}/{total}")
-)
+                detector = BatchedVideoHighlightDetector(
+                    model, 
+                    processor, 
+                    batch_size=8,
+                    progress_callback=progress_callback
+                )
 
                 yield [
                     "Analyzing video content...",
@@ -155,26 +159,30 @@ def create_ui(examples_path: str):
                 highlights = detector.determine_highlights(video_desc)
                 formatted_highlights = f"#Highlights to search for: {highlights[:500] + '...' if len(highlights) > 500 else highlights}"
                 
-                # Get total number of segments for progress tracking
-                segments = get_fixed_30s_segments(video)
-                total_segments = len(segments)
-                
-                # Process segments in batches with progress updates
-                for i in range(0, total_segments, detector.batch_size):
-                    current_batch = i + detector.batch_size
-                    progress_msg = progress_callback(min(current_batch, total_segments), total_segments)
-                    
-                    yield [
-                        progress_msg,
-                        formatted_desc,
-                        formatted_highlights,
-                        gr.update(visible=False),
-                        gr.update(visible=True)
-                    ]
+                yield [
+                    "Starting highlight detection...",
+                    formatted_desc,
+                    formatted_highlights,
+                    gr.update(visible=False),
+                    gr.update(visible=True)
+                ]
 
                 with tempfile.NamedTemporaryFile(suffix='.mp4', delete=False) as tmp_file:
                     temp_output = tmp_file.name
-                detector.create_highlight_video(video, temp_output)
+
+                    # This will now call our progress_callback during processing
+                    detector.create_highlight_video(video, temp_output)
+                    
+                    # Keep yielding progress updates while processing
+                    while current_status:
+                        yield [
+                            current_status,
+                            formatted_desc,
+                            formatted_highlights,
+                            gr.update(visible=False),
+                            gr.update(visible=True)
+                        ]
+                        time.sleep(0.1)  # Small delay to prevent too frequent updates
                 
                 yield [
                     "Processing complete!",