Update app.py

app.py

@@ -16,53 +16,53 @@ def push_up_count_logic(nose, right_wrist, right_shoulder):
     else:
         return 0
 
-def main():
-    st.title("Push-up Counter using Streamlit and webrtc-streamer")
+# Define initial values
+push_up_start = 0
+push_up_count = 0
 
-    webrtc_ctx = webrtc_streamer(
-        key="example",
-        mode=WebRtcMode.SENDRECV,
-        video_transformer_factory=None,
-        async_processing=True,
-    )
+def video_frame_callback(frame):
+    global push_up_start, push_up_count
 
-    if not webrtc_ctx.state.playing:
-        return
+    # Dummy values for demonstration
+    nose = (100, 100)
+    right_wrist = (200, 200)
+    right_shoulder = (150, 150)
 
-    push_up_start = 0
-    push_up_count = 0
+    push_up_count += push_up_count_logic(nose, right_wrist, right_shoulder)
 
-    while True:
-        if webrtc_ctx.video_receiver:
-            frame = webrtc_ctx.video_receiver.recv()
-            image = frame.to_ndarray(format="bgr24")
+    # Display the push-up count on the video frame
+    image = frame.to_ndarray(format="bgr24")
+    font = cv2.FONT_HERSHEY_SIMPLEX
+    org = (50, 100)
+    font_scale = 1
+    color = (255, 0, 0)
+    thickness = 2
+    image = cv2.putText(
+        image,
+        f"Push-up count: {push_up_count}",
+        org,
+        font,
+        font_scale,
+        color,
+        thickness,
+        cv2.LINE_AA,
+    )
 
-            # Dummy values for demonstration
-            nose = (100, 100)
-            right_wrist = (200, 200)
-            right_shoulder = (150, 150)
+    return av.VideoFrame.from_ndarray(image, format="bgr24")
 
-            push_up_count += push_up_count_logic(nose, right_wrist, right_shoulder)
+def main():
+    st.title("AI Push-up Counter")
 
-            # Display the push-up count on the video frame
-            font = cv2.FONT_HERSHEY_SIMPLEX
-            org = (50, 100)
-            font_scale = 1
-            color = (255, 0, 0)
-            thickness = 2
-            image = cv2.putText(
-                image,
-                f"Push-up count: {push_up_count}",
-                org,
-                font,
-                font_scale,
-                color,
-                thickness,
-                cv2.LINE_AA,
-            )
+    webrtc_ctx = webrtc_streamer(
+        key="pushup",
+        mode=WebRtcMode.SENDRECV,
+        # video_transformer_factory=None,
+        video_frame_callback=video_frame_callback,
+        # async_processing=True,
+    )
 
-            # Show the video frame with push-up count
-            st.image(image, channels="BGR", use_column_width=True)
+    if not webrtc_ctx.state.playing:
+        return
 
 if __name__ == "__main__":
-    main()
+    main()