UI

Modules/PoseEstimation/pose_agent.py

@@ -82,7 +82,7 @@ def check_knee_angle(json_path: str) -> bool:
 def check_squat(file_name: str) -> str:
     """
     Checks if the squat is correct.
-    This function uses the check_knee_angle tool to check if the squat is correct. If the minimum knee angle is smaller than 90 degrees, the squat is considered correct, because it means the user is going deep enough.
+    This function uses the check_knee_angle tool to check if the squat is correct. it checks if the user is going deep enough.
 
 
     Args:
@@ -109,7 +109,7 @@ prompt = ChatPromptTemplate.from_messages(
     [
         (
             "system",
-            "You are a helpful assistant. Make sure to use the check_squat tool if the user wants to check his movement. Also explain your response",
+            "You are a helpful assistant. Make sure to use the check_squat tool if the user wants to check his movement. Also explain your response, giving advices",
         ),
         ("placeholder", "{chat_history}"),
         ("human", "{input}"),

Modules/PoseEstimation/pose_estimator.py

@@ -23,7 +23,7 @@ id_joints_dict = {0: 'nose',
         16: 'right_ankle'}
 joints_id_dict = {v: k for k, v in id_joints_dict.items()}
 
-model = YOLO('yolov8n-pose.pt')
+model = YOLO('yolov8m-pose.pt')
 
 def get_keypoints_from_keypoints(video_path):
     save_folder='tmp'