Update main.py

main.py

@@ -4,6 +4,7 @@ import cv2
 import face_detection
 import numpy as np
 import base64
+import os
 
 app = Flask(__name__)
 
@@ -48,30 +49,31 @@ def video_feed():
         return jsonify({'error': 'No video file in the request'})
 
     video_file = request.files['video']
+    video_path = "uploaded_video.mp4"
+    video_file.save(video_path)
 
-    if video_file.filename == '':
-        return jsonify({'error': 'Empty video file'})
+    vidObj = cv2.VideoCapture(video_path)
 
-    try:
-        video_bytes = video_file.read()
-        print(video_bytes[:10])
-        video_np = np.frombuffer(video_bytes, dtype=np.uint8)
-        vidcap = cv2.imdecode(video_np, cv2.IMREAD_COLOR)
-    except Exception as e:
-        return jsonify({'error': f'Error decoding video file: {str(e)}'})
+    success, image = vidObj.read()
 
-    if vidcap is None:
-        return jsonify({'error': 'Error decoding video file'})
+    while success:
+        try:
+            # Call your face detection function here
+            detect_faces_and_save(image)
+        except Exception as e:
+            return jsonify({'error': f'Error processing frames: {str(e)}'})
 
-    success, image = vidcap.read()
+        success, image = vidObj.read()
 
-    while success:
-        # Call your face detection function here
-        detect_faces_and_save(image)
+    # Close the video capture object
+    vidObj.release()
 
-        success, image = vidcap.read()
+    # Delete the video file after processing
+    os.remove(video_path)
 
     return jsonify({'result': 'Video parsed'})
+
+
 @app.route('/')
 def home():
     return jsonify(message='Welcome to the Hugging Face Space!')