Pratyush101 commited on
Commit
2ac10ec
·
verified ·
1 Parent(s): 2485090

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +19 -4
app.py CHANGED
@@ -118,10 +118,25 @@ def video_frame_callback(frame: av.VideoFrame) -> av.VideoFrame:
118
  if 80 < angleKneeL < 110 and stage == 'up':
119
  stage = 'down'
120
 
121
- # Overlay feedback messages
122
- overlay_text = f"Correct: {correct} | Stage: {stage}"
123
- cv2.rectangle(image, (0, 0), (500, 80), (245, 117, 16), -1)
124
- cv2.putText(image, overlay_text, (10, 60), cv2.FONT_HERSHEY_SIMPLEX, 1, (255, 255, 255), 2, cv2.LINE_AA)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
125
 
126
  mp_drawing.draw_landmarks(image, results.pose_landmarks, mp_pose.POSE_CONNECTIONS,mp_drawing.DrawingSpec(color=(255, 175, 0), thickness=2, circle_radius=2),mp_drawing.DrawingSpec(color=(0, 255, 200), thickness=2, circle_radius=2))
127
 
 
118
  if 80 < angleKneeL < 110 and stage == 'up':
119
  stage = 'down'
120
 
121
+ #Render Counter to our camera screen
122
+ #Setup Status box
123
+ cv2.rectangle(image,(0,0),(500,80),(245,117,16),-1)
124
+
125
+ #REP data
126
+ cv2.putText(image,'Left',(10,12),
127
+ cv2.FONT_HERSHEY_SIMPLEX,0.5,(0,0,0),1,cv2.LINE_AA)
128
+
129
+ cv2.putText(image,str(correct),
130
+ (10,60),cv2.FONT_HERSHEY_SIMPLEX,2,(255,255,255),2,cv2.LINE_AA)
131
+ #Stage data for left leg
132
+
133
+ cv2.putText(image,'STAGE',(230,12),
134
+ cv2.FONT_HERSHEY_SIMPLEX,0.5,(0,0,0),1,cv2.LINE_AA)
135
+
136
+ cv2.putText(image,stage,
137
+ (230,60),cv2.FONT_HERSHEY_SIMPLEX,1,(255,255,255),1,cv2.LINE_AA)
138
+
139
+
140
 
141
  mp_drawing.draw_landmarks(image, results.pose_landmarks, mp_pose.POSE_CONNECTIONS,mp_drawing.DrawingSpec(color=(255, 175, 0), thickness=2, circle_radius=2),mp_drawing.DrawingSpec(color=(0, 255, 200), thickness=2, circle_radius=2))
142