Update app.py

app.py

@@ -80,6 +80,11 @@ def create_output_writer(cap, output_path):
             
     raise ValueError("Could not initialize any video codec")
 
+def frame_to_pil(frame):
+    """Convert OpenCV frame to PIL Image"""
+    rgb_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+    return Image.fromarray(rgb_frame)
+
 def process_video(video_path, progress=gr.Progress()):
     """
     Process the video file and return the path to the processed video
@@ -116,14 +121,19 @@ def process_video(video_path, progress=gr.Progress()):
             
             # Process frame
             if frame_count % process_every_n_frames == 0:
-                # Convert frame to RGB for the model
-                frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
-                
-                # Detect objects
-                detections = object_detector(frame_rgb)
+                # Convert frame to PIL Image for the model
+                pil_frame = frame_to_pil(frame)
                 
-                # Draw bounding boxes
-                frame = draw_bounding_boxes(frame, detections)
+                try:
+                    # Detect objects
+                    detections = object_detector(pil_frame)
+                    
+                    # Draw bounding boxes
+                    frame = draw_bounding_boxes(frame, detections)
+                except Exception as e:
+                    print(f"Error processing frame {frame_count}: {str(e)}")
+                    # Continue with the original frame if detection fails
+                    pass
             
             # Write the frame
             out.write(frame)
@@ -181,4 +191,4 @@ demo = gr.Interface(
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()