Update codriao_tb_module.py

codriao_tb_module.py

@@ -6,16 +6,21 @@ if TYPE_CHECKING:
 import sys
 import os
 
-# Ensure Python can find the TB processors
-sys.path.append(os.path.dirname(os.path.abspath(__file__)))
+# Ensure Python can detect the correct module location
+BASE_DIR = os.path.dirname(os.path.abspath(__file__))
+sys.path.append(BASE_DIR)  # Adds the directory to Python’s path
 
 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__)
+    # Try importing from 'modules' if files are inside it
+    sys.path.append(os.path.join(BASE_DIR, "modules"))
+    try:
+        from modules.tb_image_processor import TBImageProcessor
+        from modules.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 inside 'modules/'.")
 
 class CodriaoHealthModule:
     """Embedded compassionate TB detection within Codriao's architecture"""