pushup-counter/pushup.py at main · 04092000f/pushup-counter · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
import streamlit as st
import cv2
import mediapipe as mp
import numpy as np
import tempfile
import os
import time

st.set_page_config(page_title="Push-Up Counter", layout="centered")
st.title("Push-Up Counter from Video")

def calculate_angle(a, b, c):
    a, b, c = np.array(a), np.array(b), np.array(c)
    radians = np.arctan2(c[1]-b[1], c[0]-b[0]) - np.arctan2(a[1]-b[1], a[0]-b[0])
    angle = abs(radians * 180 / np.pi)
    return 360 - angle if angle > 180 else angle

uploaded_file = st.file_uploader("Upload your push-up video", type=["mp4", "mov", "avi"])
if uploaded_file:
    temp_input = tempfile.NamedTemporaryFile(delete=False)
    temp_input.write(uploaded_file.read())
    input_path = temp_input.name

    output_path = tempfile.NamedTemporaryFile(suffix=".mp4", delete=False).name

    cap = cv2.VideoCapture(input_path)
    width, height = int(cap.get(3)), int(cap.get(4))
    fps = int(cap.get(cv2.CAP_PROP_FPS))
    total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))

    # Use x264-compatible codec (if available)
    fourcc = cv2.VideoWriter_fourcc(*'avc1')  # Try 'H264' if 'avc1' doesn't work
    writer = cv2.VideoWriter(output_path, fourcc, fps, (width, height))

    mp_pose = mp.solutions.pose
    mp_drawing = mp.solutions.drawing_utils
    pushup_count = 0
    direction = 0

    progress_bar = st.progress(0)
    progress_text = st.empty()
    st.info("Processing your video...")

    start_time = time.time()
    frame_counter = 0

    with mp_pose.Pose(min_detection_confidence=0.5, min_tracking_confidence=0.5) as pose:
        while cap.isOpened():
            ret, frame = cap.read()
            if not ret:
                break

            image = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
            results = pose.process(image)
            image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)

            try:
                landmarks = results.pose_landmarks.landmark
                shoulder = [landmarks[mp_pose.PoseLandmark.LEFT_SHOULDER.value].x,
                            landmarks[mp_pose.PoseLandmark.LEFT_SHOULDER.value].y]
                elbow = [landmarks[mp_pose.PoseLandmark.LEFT_ELBOW.value].x,
                         landmarks[mp_pose.PoseLandmark.LEFT_ELBOW.value].y]
                wrist = [landmarks[mp_pose.PoseLandmark.LEFT_WRIST.value].x,
                         landmarks[mp_pose.PoseLandmark.LEFT_WRIST.value].y]
                angle = calculate_angle(shoulder, elbow, wrist)

                if angle > 160 and direction == 1:
                    pushup_count += 1
                    direction = 0
                elif angle < 90 and direction == 0:
                    direction = 1

                cv2.putText(image, f'Angle: {int(angle)}', (width - 320, 80),
                            cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 255), 3)
            except:
                pass

            mp_drawing.draw_landmarks(image, results.pose_landmarks, mp_pose.POSE_CONNECTIONS)
            cv2.putText(image, f'Push-Ups: {pushup_count}', (width - 320, 50),
                        cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 0), 4)

            writer.write(image)
            frame_counter += 1

            progress = frame_counter / total_frames
            progress_bar.progress(min(progress, 1.0))
            elapsed = time.time() - start_time
            avg_time_per_frame = elapsed / frame_counter if frame_counter else 0
            eta = (total_frames - frame_counter) * avg_time_per_frame
            progress_text.markdown(
                f"Processed {frame_counter}/{total_frames} frames "
                f"Elapsed: {elapsed:.1f}s | ETA: {eta:.1f}s"
            )

    cap.release()
    writer.release()

    st.success(f"Done! Total push-ups detected: {pushup_count}")

    with open(output_path, 'rb') as video_file:
        video_bytes = video_file.read()
        st.video(video_bytes)
        st.download_button("Download H.264 Video (.mp4)", video_bytes, file_name="pushups_h264.mp4", mime="video/mp4")