Update app.py

app.py

@@ -60,32 +60,31 @@ class VideoProcessor:
         self.pose = mp_pose.Pose(min_detection_confidence=0.5, min_tracking_confidence=0.5)
 
     def process_video(self, video_file):
-        # Get the filename from the file object
-        filename = "temp_video.mp4"
-        # Open the file and read its contents
-        with open(filename, 'wb') as temp_file:
-            temp_file.write(video_file.read())
-        # Process the video and save the processed video to a new file
-        output_filename = "processed_video.mp4"
-        cap = cv2.VideoCapture(filename)
-        frame_width = int(cap.get(3))
-        frame_height = int(cap.get(4))
-        out = cv2.VideoWriter(output_filename, cv2.VideoWriter_fourcc(*'h264'), 30, (frame_width, frame_height))
+        cap = cv2.VideoCapture(video_file.name)  # Open the video file
+        height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
+        width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
+        fps = cap.get(cv2.CAP_PROP_FPS)
+
+        frames = []
+
         while cap.isOpened():
             ret, frame = cap.read()
             if not ret:
                 break
-            frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
-            results = self.pose.process(frame_rgb)
-            processed_frame = self.process_frame(frame, results)
-            out.write(processed_frame)
+            rgb_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+            # Process the frame (replace this with your processing logic)
+            processed_frame = self.process_frame(rgb_frame)
+            frames.append(processed_frame)  # Append processed frame
+
         cap.release()
-        out.release()
-        # Remove the temporary file
-        os.remove(filename)
-        # Return the path to the processed video file
+
+        # Write processed frames to video
+        output_filename = "processed_video.mp4"
+        mediapy.write_video(output_filename, frames, fps=fps)
+
         return output_filename
 
+
     
     def process_frame(self, frame, results):
         # Process the frame using the `process` function