Update app.py

app.py

@@ -23,6 +23,7 @@ if uploaded_file is not None:
     cap = cv2.VideoCapture(tfile.name)
     assert cap.isOpened(), "Error reading video file"
     w, h, fps = (int(cap.get(x)) for x in (cv2.CAP_PROP_FRAME_WIDTH, cv2.CAP_PROP_FRAME_HEIGHT, cv2.CAP_PROP_FPS))
+    total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
 
     # Initialize AIGym object
     gym_object = solutions.AIGym(
@@ -35,6 +36,10 @@ if uploaded_file is not None:
     # List to store processed frames
     processed_frames = []
 
+    # Streamlit progress bar
+    progress_bar = st.progress(0)
+    frame_count = 0
+
     # Process the video frame by frame
     st.write("Analyzing video... Please wait.")
     while cap.isOpened():
@@ -45,18 +50,25 @@ if uploaded_file is not None:
         results = model.track(im0, verbose=False)  # Tracking recommended
         im0 = gym_object.start_counting(im0, results)
 
+        # Resize the frame to 320x320
+        im0_resized = cv2.resize(im0, (320, 320))
+
         # Append the processed frame to the list
-        processed_frames.append(im0)
+        processed_frames.append(im0_resized)
+
+        # Update progress bar
+        frame_count += 1
+        progress_bar.progress(frame_count / total_frames)
 
     cap.release()
     cv2.destroyAllWindows()
 
     # Create a temporary file to save the processed video
-    output_video_file = tempfile.NamedTemporaryFile(delete=False, suffix='.avi')
+    output_video_file = tempfile.NamedTemporaryFile(delete=False, suffix='.mp4')
     output_video_path = output_video_file.name
 
     # Write the processed frames to a video file
-    video_writer = cv2.VideoWriter(output_video_path, cv2.VideoWriter_fourcc(*"mp4v"), fps, (w, h))
+    video_writer = cv2.VideoWriter(output_video_path, cv2.VideoWriter_fourcc(*"mp4v"), fps, (320, 320))
     for frame in processed_frames:
         video_writer.write(frame)
     video_writer.release()
@@ -68,4 +80,4 @@ if uploaded_file is not None:
     # Provide a download link for the processed video
     st.write("Download the processed video:")
     with open(output_video_path, "rb") as video_file:
-        st.download_button("Download", video_file, "workouts.avi")
+        st.download_button("Download", video_file, "workouts.mp4")