test

src/app.py

@@ -53,29 +53,28 @@ def on_process(video):
         
         # Get duration and calculate total segments
         duration = get_video_duration_seconds(video)
-        total_segments = int(duration / 10)  # Using default 10-second segments
+        total_segments = (int(duration) + 9) // 10  # Ceiling division for 10-second segments
         
-        # Process all segments at once
+        # Process video segments
         yield [
-            f"Processing video... (This may take a while for {total_segments} segments)",
+            f"Processing video... (This will process {total_segments} segments)",
             "",
             gr.update(visible=True)
         ]
         
-        # Get all segments in one call
-        all_segments = analyzer.process_video(video)
-        
-        # Update progress as we format the results
-        for i, segment in enumerate(all_segments):
+        # Process segments and show progress
+        segments = []
+        for i, segment in enumerate(analyzer.process_video(video)):
+            segments.append(segment)
             progress = int((i + 1) / total_segments * 100)
             
             # Format current segments
             formatted_desc = "### Video Analysis by Segments:\n\n"
-            for j in range(i + 1):  # Only show processed segments
-                formatted_desc += f"**[{all_segments[j]['timestamp']}]** {all_segments[j]['description']}\n\n"
+            for seg in segments:
+                formatted_desc += f"**[{seg['timestamp']}]** {seg['description']}\n\n"
             
             yield [
-                f"Formatting results... {progress}% complete",
+                f"Processing segments... {progress}% complete",
                 formatted_desc,
                 gr.update(visible=True)
             ]

src/video_processor/processor.py

@@ -109,11 +109,19 @@ Be specific about visual details but stay concise."""}
             
             # Get video duration
             duration = get_video_duration_seconds(video_path)
+            segments_processed = 0
+            total_segments = int(duration / segment_length)
+            logger.info(f"Processing {total_segments} segments for video of length {duration:.2f} seconds")
             
             # Process video in segments
-            for start_time in range(0, int(duration), segment_length):
+            for segment_idx in range(total_segments):
+                start_time = segment_idx * segment_length
                 end_time = min(start_time + segment_length, duration)
                 
+                # Skip if we've reached the end
+                if start_time >= duration:
+                    break
+                
                 # Create segment
                 segment_path = os.path.join(temp_dir, f"segment_{start_time}.mp4")
                 cmd = [
@@ -121,7 +129,7 @@ Be specific about visual details but stay concise."""}
                     "-y",
                     "-i", video_path,
                     "-ss", str(start_time),
-                    "-t", str(segment_length),
+                    "-t", str(end_time - start_time),  # Duration of this segment
                     "-c:v", "libx264",
                     "-preset", "ultrafast",
                     "-pix_fmt", "yuv420p",
@@ -140,6 +148,8 @@ Be specific about visual details but stay concise."""}
                 
                 # Clean up segment file
                 os.remove(segment_path)
+                
+                logger.info(f"Processed segment {segment_idx + 1}/{total_segments} ({start_time}-{end_time}s)")
             
             # Clean up temp directory
             os.rmdir(temp_dir)