main.py

import cv2
import mediapipe as mp
import numpy as np

mp_pose = mp.solutions.pose
pose = mp_pose.Pose()

mp_drawing = mp.solutions.drawing_utils

cap = cv2.VideoCapture(0)

width, height = 1920, 1080

cap.set(cv2.CAP_PROP_FRAME_WIDTH, width)
cap.set(cv2.CAP_PROP_FRAME_HEIGHT, height)

pose_bg = np.zeros((height, width, 3), dtype=np.uint8)
landmarks_present = 0

while cap.isOpened():
    ret, frame = cap.read()
    if not ret:
        break

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

    if results.pose_landmarks:
        if landmarks_present == 0:
            mp_drawing.draw_landmarks(pose_bg, results.pose_landmarks, mp_pose.POSE_CONNECTIONS)
            landmarks_present = 1
        else:
            pose_bg = np.zeros((height, width, 3), dtype=np.uint8)
            mp_drawing.draw_landmarks(pose_bg, results.pose_landmarks, mp_pose.POSE_CONNECTIONS)

    cv2.imshow('Body Tracking', pose_bg)
    cv2.imshow('Camera Footage', frame)

    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

cap.release()
cv2.destroyAllWindows()