Update codriao_tb_module.py

codriao_tb_module.py

@@ -3,8 +3,17 @@ import logging
 from typing import TYPE_CHECKING
 if TYPE_CHECKING:
     from CodriaoCore import CodriaoCore
-from tb_image_processor import TBImageProcessor
-from tb_audio_processor import TBAudioProcessor
+import sys
+import os
+
+# Ensure Python can find the TB processors
+sys.path.append(os.path.dirname(os.path.abspath(__file__)))
+
+try:
+    from tb_image_processor import TBImageProcessor
+    from tb_audio_processor import TBAudioProcessor
+except ModuleNotFoundError:
+    raise ImportError("Could not locate tb_image_processor.py or tb_audio_processor.py. Ensure they are in the root directory or 'modules/'.")
 
 logger = logging.getLogger(__name__)