Update lipnet/preprocessing.py

lipnet/preprocessing.py

@@ -1,3 +1,4 @@
+import os
 import logging
 from pathlib import Path
 from typing import Iterable, List, Optional
@@ -6,24 +7,36 @@ import cv2
 import numpy as np
 import tensorflow as tf
 
+# ------------------------------------------------------------------
+# MediaPipe import (robust + debuggable)
+# ------------------------------------------------------------------
 _mp_import_error = None
 mp_solutions = None
+
 try:
-    from mediapipe import solutions as mp_solutions  # type: ignore
-except Exception as exc:  # pragma: no cover
+    import mediapipe as mp
+    mp_solutions = mp.solutions
+except Exception as exc:
+    _mp_import_error = exc
+    mp_solutions = None
+
+# Optional debug (enable with DEBUG_MEDIAPIPE=1)
+if os.getenv("DEBUG_MEDIAPIPE", "0") == "1":
     try:
-        import mediapipe.solutions as mp_solutions  # type: ignore
-    except Exception as inner_exc:  # pragma: no cover
-        _mp_import_error = inner_exc
-        mp_solutions = None
-    else:
-        _mp_import_error = exc
-
+        import mediapipe as mp
+        print("mediapipe version:", getattr(mp, "__version__", "unknown"))
+        print("mediapipe file:", getattr(mp, "__file__", "unknown"))
+        print("has solutions:", hasattr(mp, "solutions"))
+    except Exception as dbg_exc:
+        print("mediapipe debug import failed:", dbg_exc)
+
+# ------------------------------------------------------------------
+# Local imports
+# ------------------------------------------------------------------
 from . import config
 
 logger = logging.getLogger(__name__)
 
-
 class VideoPreprocessor:
     """
     Handles frame extraction and normalization from either a video file