Update app.py

app.py

@@ -7,6 +7,7 @@ import math
 from tensorflow.keras.models import Model
 from tensorflow.keras.layers import (LSTM, Dense, Dropout, Input, Flatten, 
                                      Bidirectional, Permute, multiply)
+from moviepy.editor import VideoClip
 
 # Load the pose estimation model from Mediapipe
 mp_pose = mp.solutions.pose 
@@ -69,10 +70,7 @@ class VideoProcessor:
             temp_file.write(video_file.read())
         # Now we can open the video file using cv2.VideoCapture()
         cap = cv2.VideoCapture(filename)
-        output_filename = "processed_video.mp4"
-        frame_width = int(cap.get(3))
-        frame_height = int(cap.get(4))
-        out = cv2.VideoWriter(output_filename, cv2.VideoWriter_fourcc(*'mp4v'), 30, (frame_width,frame_height))
+        frames = []
         while cap.isOpened():
             ret, frame = cap.read()
             if not ret:
@@ -80,13 +78,11 @@ class VideoProcessor:
             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)
+            frames.append(processed_frame)
         cap.release()
-        out.release()
         # Remove the temporary file
         os.remove(filename)
-        print("Processed video saved at:", output_filename)  # Debug print
-        return output_filename
+        return frames
     
     def process_frame(self, frame, results):
         # Process the frame using the `process` function
@@ -269,8 +265,14 @@ def main():
     video_file = st.file_uploader("Upload a video file", type=["mp4", "avi"])
     if video_file is not None:
         video_processor = VideoProcessor()
-        processed_video_file = video_processor.process_video(video_file)
-        st.video(processed_video_file)
+        frames = video_processor.process_video(video_file)
+        
+        def make_frame(t):
+            frame_index = int(t * 30)  # Assuming 30 frames per second
+            return frames[frame_index]
+        
+        video = VideoClip(make_frame, duration=len(frames) / 30)
+        st.video(video)
 
 if __name__ == "__main__":
     main()