import cv2
import mediapipe as mp

mp_drawing = mp.solutions.drawing_utils
mp_pose = mp.solutions.pose

cap = cv2.VideoCapture(0)
ratio = cap.get(cv2.CAP_PROP_FRAME_WIDTH) / cap.get(cv2.CAP_PROP_FRAME_HEIGHT)
WIDTH = 400
HEIGHT = int(WIDTH / ratio)

pose = mp_pose.Pose(min_detection_confidence = 0.5, min_tracking_confidence = 0.5)
                    
while cap.isOpened():
    ret, frame = cap.read()

    if ret:
        frame = cv2.resize(frame, (WIDTH, HEIGHT))
        frame = cv2.rotate(frame, rotateCode = 1)
        frame = cv2.flip(frame, 1)

        frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
        frame.flags.writeable = False
        results = pose.process(frame)
        frame.flags.writeable = True
        frame = cv2.cvtColor(frame, cv2.COLOR_RGB2BGR)
        mp_drawing.draw_landmarks(frame,
            results.pose_landmarks,
            mp_pose.POSE_CONNECTIONS)
        cv2.imshow("MediaPipe Pose", frame)

    if cv2.waitKey(1) == 27:
        break

cap.release()
cv2.destroyAllWindows()
