timing

src/app.py

@@ -3,6 +3,7 @@ from video_processor.processor import VideoAnalyzer, get_video_duration_seconds
 import logging
 import torch
 import spaces
+import time
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -17,6 +18,8 @@ if torch.cuda.is_available():
 
 @spaces.GPU
 def on_process(video):
+    start_time = time.time()
+    
     # Clear all components when starting new processing
     yield [
         "",  # Clear status
@@ -34,6 +37,7 @@ def on_process(video):
     
     try:
         # Initialize analyzer
+        init_start = time.time()
         yield [
             "Initializing video analyzer...",
             "",
@@ -41,10 +45,12 @@ def on_process(video):
         ]
         
         analyzer = VideoAnalyzer()
+        init_time = time.time() - init_start
+        logger.info(f"Initialization took {init_time:.2f} seconds")
         
         # Process video
         yield [
-            "Analyzing video content...",
+            f"Model initialized in {init_time:.2f}s. Starting analysis...",
             "",
             gr.update(visible=True)
         ]
@@ -57,16 +63,25 @@ def on_process(video):
         
         # Process video segments
         yield [
-            f"Processing video... (This will process {total_segments} segments)",
+            f"Processing video... (Will analyze {total_segments} segments)",
             "",
             gr.update(visible=True)
         ]
         
         # Process segments and show progress
         segments = []
+        total_processing_time = 0
+        
         for i, segment in enumerate(analyzer.process_video(video)):
+            segment_start = time.time()
             segments.append(segment)
+            segment_time = time.time() - segment_start
+            total_processing_time += segment_time
+            
             progress = int((i + 1) / total_segments * 100)
+            avg_time_per_segment = total_processing_time / (i + 1)
+            remaining_segments = total_segments - (i + 1)
+            estimated_remaining = remaining_segments * avg_time_per_segment
             
             # Format current segments
             formatted_desc = "### Video Analysis by Segments:\n\n"
@@ -74,13 +89,19 @@ def on_process(video):
                 formatted_desc += f"**[{seg['timestamp']}]** {seg['description']}\n\n"
             
             yield [
-                f"Processing segments... {progress}% complete",
+                f"Processing segments... {progress}% complete\n" +
+                f"Segment {i+1}/{total_segments} processed in {segment_time:.2f}s\n" +
+                f"Average time per segment: {avg_time_per_segment:.2f}s\n" +
+                f"Estimated time remaining: {estimated_remaining:.2f}s",
                 formatted_desc,
                 gr.update(visible=True)
             ]
         
+        total_time = time.time() - start_time
         yield [
-            "Processing complete!",
+            f"Processing complete!\n" +
+            f"Total processing time: {total_time:.2f}s\n" +
+            f"Average time per segment: {total_processing_time/total_segments:.2f}s",
             formatted_desc,
             gr.update(visible=True)
         ]