Fix import issues on Hugging Face space mock fallback

.gitignore

@@ -34,4 +34,7 @@ record/
 
 # Key / certificate files
 *.pem
-*.key
+*.key
+
+# Project retrospective tutorial
+learn.md

reachy_mini/__init__.py

@@ -6,14 +6,19 @@ parent_dir = os.path.dirname(current_dir)
 
 real_reachy_mini_pkg = None
 original_path = list(sys.path)
+original_module = sys.modules.pop('reachy_mini', None)
 try:
     sys.path = [p for p in sys.path if os.path.abspath(p) not in (parent_dir, current_dir)]
     import reachy_mini as rmp
-    real_reachy_mini_pkg = rmp
+    # Verify it is the real reachy_mini package with the required class
+    if hasattr(rmp, 'ReachyMini'):
+        real_reachy_mini_pkg = rmp
 except ImportError:
     pass
 finally:
     sys.path = original_path
+    if original_module is not None:
+        sys.modules['reachy_mini'] = original_module
 
 if real_reachy_mini_pkg is not None:
     globals().update({k: v for k, v in real_reachy_mini_pkg.__dict__.items() if not k.startswith('__')})

reachy_mini/media/media_manager.py

@@ -6,14 +6,24 @@ parent_dir = os.path.dirname(current_dir) # dev/camping_ai/
 
 real_media_manager = None
 original_path = list(sys.path)
+original_module = sys.modules.pop('reachy_mini.media.media_manager', None)
+original_parent_media = sys.modules.pop('reachy_mini.media', None)
+original_parent = sys.modules.pop('reachy_mini', None)
 try:
     sys.path = [p for p in sys.path if os.path.abspath(p) not in (parent_dir, current_dir)]
     from reachy_mini.media import media_manager as rmm
-    real_media_manager = rmm
+    if hasattr(rmm, 'MediaManager'):
+        real_media_manager = rmm
 except ImportError:
     pass
 finally:
     sys.path = original_path
+    if original_parent is not None:
+        sys.modules['reachy_mini'] = original_parent
+    if original_parent_media is not None:
+        sys.modules['reachy_mini.media'] = original_parent_media
+    if original_module is not None:
+        sys.modules['reachy_mini.media.media_manager'] = original_module
 
 if real_media_manager is not None:
     globals().update({k: v for k, v in real_media_manager.__dict__.items() if not k.startswith('__')})

reachy_mini/utils.py

@@ -6,14 +6,21 @@ parent_dir = os.path.dirname(current_dir) # dev/camping_ai/
 
 real_utils = None
 original_path = list(sys.path)
+original_module = sys.modules.pop('reachy_mini.utils', None)
+original_parent = sys.modules.pop('reachy_mini', None)
 try:
     sys.path = [p for p in sys.path if os.path.abspath(p) not in (parent_dir, current_dir)]
     from reachy_mini import utils as ru
-    real_utils = ru
+    if hasattr(ru, 'create_head_pose'):
+        real_utils = ru
 except ImportError:
     pass
 finally:
     sys.path = original_path
+    if original_parent is not None:
+        sys.modules['reachy_mini'] = original_parent
+    if original_module is not None:
+        sys.modules['reachy_mini.utils'] = original_module
 
 if real_utils is not None:
     globals().update({k: v for k, v in real_utils.__dict__.items() if not k.startswith('__')})