Update osc_module.py

Author: SentryCoderDev

modules/osc_module.py

@@ -6,42 +6,48 @@
 
 ANIMATION_DIR = 'modules/animations'
 
-def load_animation(animation_name):
-    """Load animation data from a JSON file."""
-    file_path = os.path.join(ANIMATION_DIR, f'{animation_name}.json')
-    if not os.path.isfile(file_path):
-        print(f"Animation file '{file_path}' does not exist.")
-        return None
-    with open(file_path, 'r') as file:
-        return json.load(file)
-
-def handle_prediction(left, right):
-    """Determine which animation to execute based on prediction values."""
-    if left > 0.6:
-        animation_name = 'Llft'  # Load 'Llft' animation for left
-    elif right > 0.6:
-        animation_name = 'Rlft'  # Load 'Rlft' animation for right
-    else:
-        animation_name = 'lower'  # Load 'lower' animation
-
-    # Send the animation action to the Animate class via PubSub
-    pub.sendMessage('animate', action=animation_name)
-
-def start_osc_server():
-    """Initialize and start the OSC server to handle messages."""
-    def callback(left, right):
-        """Handle OSC messages and execute animations based on predictions."""
-        print("Left prediction : ", round(left, 2))
-        print("Right prediction : ", round(right, 2))
-        handle_prediction(left, right)
-
-    osc = OSCThreadServer()
-    osc.listen(address='127.0.0.1', port=9002, default=True)
-    osc.bind(b'/neuropype', callback)
-
-    print("OSC server started.")
-    while True:
-        sleep(1)
+class StartOSCServer:
+    def __init__(self, address='127.0.0.1', port=9002):
+        self.osc = OSCThreadServer()
+        self.address = address
+        self.port = port
+
+    def load_animation(self, animation_name):
+        """Load animation data from a JSON file."""
+        file_path = os.path.join(ANIMATION_DIR, f'{animation_name}.json')
+        if not os.path.isfile(file_path):
+            print(f"Animation file '{file_path}' does not exist.")
+            return None
+        with open(file_path, 'r') as file:
+            return json.load(file)
+
+    def handle_prediction(self, left, right):
+        """Determine which animation to execute based on prediction values."""
+        if left > 0.6:
+            animation_name = 'Llft'  # Load 'Llft' animation for left
+        elif right > 0.6:
+            animation_name = 'Rlft'  # Load 'Rlft' animation for right
+        else:
+            animation_name = 'lower'  # Load 'lower' animation
+
+        # Send the animation action to the Animate class via PubSub
+        pub.sendMessage('animate', action=animation_name)
+
+    def start_server(self):
+        """Initialize and start the OSC server to handle messages."""
+        def callback(left, right):
+            """Handle OSC messages and execute animations based on predictions."""
+            print("Left prediction : ", round(left, 2))
+            print("Right prediction : ", round(right, 2))
+            self.handle_prediction(left, right)
+
+        self.osc.listen(address=self.address, port=self.port, default=True)
+        self.osc.bind(b'/neuropype', callback)
+
+        print("OSC server started.")
+        while True:
+            sleep(1)
 
 if __name__ == "__main__":
-    start_osc_server()
+    osc_server = StartOSCServer()
+    osc_server.start_server()