Added Button to Control Export to GUI

HPEModule.py

@@ -16,7 +16,7 @@ mp_pose = mp.solutions.pose
 class HPEModule(IHPEModule):
 
     # This method starts HPE using a camera specified by its name
-    def startHPEwithCamera(self, camera_name):
+    def startHPEwithCamera(self, camera_name, export_landmarks):
         out = None
         # check if the camera_name is a file path or not
         if os.path.isfile(camera_name):
@@ -39,7 +39,8 @@ class HPEModule(IHPEModule):
 
         # set the window name using the camera name
         window_name = f"Pose Estimation on Camera {camera_name}"
-
+        print(export_landmarks.get())
+        if export_landmarks.get() is True:
             # create the root element
             root = ET.Element("Landmarks")
 
@@ -49,6 +50,7 @@ class HPEModule(IHPEModule):
 
             framecounter = 0
 
+
         # start pose detection
         with mp_pose.Pose(min_detection_confidence=0.5, min_tracking_confidence=0.5) as pose:
 
@@ -75,6 +77,7 @@ class HPEModule(IHPEModule):
                 # Extract landmarks
                 try:
                     landmarks = results.pose_landmarks.landmark
+                    if export_landmarks.get() is True:
                         for j, landmark in enumerate(landmarks):
                             frame_element = ET.SubElement(root.find(f'landmark{j}'), f'frame{framecounter}')
                             frame_element.set("X", str(landmark.x))
@@ -100,6 +103,7 @@ class HPEModule(IHPEModule):
                 if cv2.getWindowProperty(window_name, cv2.WND_PROP_VISIBLE) < 1:
                     break
 
+        if export_landmarks.get() is True:
             xml_string = ET.tostring(root, encoding="UTF-8")
             parsed_xml = minidom.parseString(xml_string)
             pretty_xml_string = parsed_xml.toprettyxml(indent="    ")

IHPEModule.py

@@ -5,5 +5,5 @@ class IHPEModule(ABC):
 
     # This method starts HPE using a camera specified by its name
     @abstractmethod
-    def startHPEwithCamera(self, camera_name):
+    def startHPEwithCamera(self, camera_name, export_landmarks):
         pass

PoseEstimationGUI.py

@@ -12,7 +12,7 @@ class PoseEstimationGUI:
         self.mp4_file_path = None
         self.root = tk.Tk()
         self.root.title("Pose Estimation GUI")
-        self.root.geometry("400x250")
+        self.root.geometry("400x400")
         self.root.config(bg="#F9F9F9")
 
         self.camera_options = []  # Initialize camera options list
@@ -42,13 +42,21 @@ class PoseEstimationGUI:
                                       font=("Arial", 12, "bold"), command=self.start_pose_estimation)
         self.start_button.pack(pady=(20, 10))
 
+        # Create a checkbox to enable/disable Export of Landmarks
+        self.export_Landmarks = tk.BooleanVar()
+        self.export_Landmarks.set(False)
+        export_Landmarks_checkbox = tk.Checkbutton(self.root, text="Export Landmarks", variable=self.export_Landmarks,
+                                             bg="#F9F9F9",
+                                             font=("Arial", 12, "bold"), onvalue=True, offvalue=False)
+        export_Landmarks_checkbox.pack()
+
         self.root.mainloop()
 
     def start_pose_estimation(self):
         if self.mp4_file_path is not None:
             # Start pose estimation on the selected mp4 file
             pose_estimator = HPEModule()
-            pose_thread_file = threading.Thread(target=pose_estimator.startHPEwithCamera, args=(self.mp4_file_path,))
+            pose_thread_file = threading.Thread(target=pose_estimator.startHPEwithCamera, args=(self.mp4_file_path, self.export_Landmarks,))
             self.mp4_file_path = None
             self.file_selected_label.config(text="No File selected", fg="red")
             pose_thread_file.start()
@@ -57,7 +65,7 @@ class PoseEstimationGUI:
             # print(self.selected_camera.get())
             pose_estimator = HPEModule()
             camera = int(self.selected_camera.get())
-            pose_thread_camera = threading.Thread(target=pose_estimator.startHPEwithCamera, args=(camera,))
+            pose_thread_camera = threading.Thread(target=pose_estimator.startHPEwithCamera, args=(camera, self.export_Landmarks,))
             pose_thread_camera.start()
 
     def select_file(self):