Update app.py

app.py

@@ -44,7 +44,7 @@ def analyze_video(video_path, confidence=0.5):
         return None, "Error: Could not open video.", [], None, None, None, None, "Error: Could not open video.", None
 
     # Early frame check and limit
-    if int(cap.get(cv2.CAP_PROP_FRAME_COUNT)) > 600:  # Limit to 10 seconds at 60 FPS
+    if int(cap.get(cv2.CAP_PROP_FRAME_COUNT)) > 300:  # Limit to 5 seconds at 60 FPS
         cap.set(cv2.CAP_PROP_POS_FRAMES, 0)
     width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
     height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
@@ -70,8 +70,8 @@ def analyze_video(video_path, confidence=0.5):
             log.write(f"Resolution: {new_width}x{new_height}, FPS: {fps}, Confidence threshold: {confidence}\n\n")
 
             frame_count = 0
-            sample_rate = 2  # Faster processing with every 2nd frame
-            while cap.isOpened() and frame_count < 600:  # Limit to 10 seconds
+            sample_rate = 4  # Increased to process every 4th frame for faster output
+            while cap.isOpened() and frame_count < 300:  # Limit to 5 seconds
                 ret, frame = cap.read()
                 if not ret:
                     break
@@ -120,8 +120,8 @@ def analyze_video(video_path, confidence=0.5):
                             color = (255, 255, 0)  # Yellow for traffic lights
                         else:
                             color = (255, 255, 255)  # White for others
-                        cv2.rectangle(frame, (x1, y1), (x2, y2), color, 3)  # Reduced to 3px border
-                        cv2.putText(frame, f"{label} {conf:.2f}", (x1, y1 - 20), cv2.FONT_HERSHEY_SIMPLEX, 1.0, color, 2)  # Reduced to 1.0 font size
+                        cv2.rectangle(frame, (x1, y1), (x2, y2), color, 3)  # 3px border
+                        cv2.putText(frame, f"{label} {conf:.2f}", (x1, y1 - 20), cv2.FONT_HERSHEY_SIMPLEX, 1.0, color, 2)  # 1.0 font size
 
                         if label == "person":
                             for other_box in boxes:
@@ -219,7 +219,7 @@ h3 {color: #34495E; font-size: 18px; padding: 5px;}
 
 with gr.Blocks(css=css) as iface:
     gr.Markdown("# Road Safety AI Video Analysis")
-    gr.Markdown("Upload a traffic video for fast HD analysis (<1 min) with charts and downloadable PDF/ZIP.")
+    gr.Markdown("Upload a traffic video for fast HD analysis (<30 sec) with charts and downloadable PDF/ZIP.")
     
     with gr.Row():
         video_input = gr.Video(label="Upload Video", height=360)
@@ -233,7 +233,7 @@ with gr.Blocks(css=css) as iface:
     
     with gr.Row():
         with gr.Column(scale=1):
-            video_output = gr.Video(label="Annotated Video Preview (HD)", height=360)  # Adjusted to 360px for better proportion
+            video_output = gr.Video(label="Annotated Video Preview (HD)", height=360)
             summary_output = gr.Textbox(label="Analysis Summary", lines=15)
         with gr.Column(scale=1):
             screenshot_output = gr.Gallery(label="Top 15 HD Incident Screenshots", columns=3, rows=5, height=400)