Hugging Face's logo

Spaces:
amarkc
/
yoga-pose-detection-correction
Runtime error

App Files Community
yoga-pose-detection-correction
File size: 2,451 Bytes 
c8f49a6
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
 
def init():
    angles_dict = {
        "armpit_left": 0,
        "armpit_right": 1,
        "elbow_left": 2,
        "elbow_right": 3,
        "hip_left": 4,
        "hip_right": 5,
        "knee_left": 6,
        "knee_right": 7,
        "ankle_left": 8,
        "ankle_right": 9,
    }
    return angles_dict


def error_margin(control, value):
    if int(value) in range(control - 20, control + 21):
        return True
    return False


def check_joint(angles, joint_name, threshold, body_position):
    angles_dict = init()
    joint_index = angles_dict[joint_name]

    if error_margin(threshold, angles[joint_index]):
        return None

    if angles[joint_index] > threshold:
        return f"Bring {' '.join(joint_name.split('_')[::-1])} closer to {body_position}."
    elif angles[joint_index] < threshold:
        return f"Put {' '.join(joint_name.split('_')[::-1])} further away from {body_position}."

    return None


def check_pose_angle(pose_index, angles, df):
    result = []

    result.append(check_joint(
        angles,
        "armpit_right",
        int(df.loc[pose_index, "armpit_left"]),
        "body"
    ))
    result.append(check_joint(
        angles,
        "armpit_left",
        int(df.loc[pose_index, "armpit_right"]),
        "body"
    ))
    result.append(check_joint(
        angles,
        "elbow_right",
        int(df.loc[pose_index, "elbow_left"]),
        "arm"
    ))
    result.append(check_joint(
        angles,
        "elbow_left",
        int(df.loc[pose_index, "elbow_right"]),
        "arm"
    ))
    result.append(check_joint(
        angles,
        "hip_right",
        int(df.loc[pose_index, "hip_left"]),
        "pelvis"
    ))
    result.append(check_joint(
        angles,
        "hip_left",
        int(df.loc[pose_index, "hip_right"]),
        "pelvis"
    ))
    result.append(check_joint(
        angles,
        "knee_right",
        int(df.loc[pose_index, "knee_left"]),
        "calf"
    ))
    result.append(check_joint(
        angles,
        "knee_left",
        int(df.loc[pose_index, "knee_right"]),
        "calf"
    ))
    result.append(check_joint(
        angles,
        "ankle_right",
        int(df.loc[pose_index, "ankle_left"]),
        "foot"
    ))
    result.append(check_joint(
        angles,
        "ankle_left",
        int(df.loc[pose_index, "ankle_right"]),
        "foot"
    ))

    return [message for message in result if message is not None]