fix: unable to import improved_diarization.py on HF Spaces

src/diarization.py

@@ -20,14 +20,39 @@ from typing import List, Tuple, Optional, Callable, Dict, Any
 import streamlit as st
 import logging
 
-# Import the improved diarization pipeline
+# Import the improved diarization pipeline (robust: search repo tree)
 try:
-    import sys
-    sys.path.append('/home/luigi/VoxSum')
-    from improved_diarization import enhance_diarization_pipeline
+    from importlib import import_module
+    # Try direct import first (works when repo root is in PYTHONPATH)
+    try:
+        mod = import_module('improved_diarization')
+    except Exception:
+        # Search up to 6 parent directories for improved_diarization.py
+        repo_root = None
+        current = Path(__file__).resolve()
+        for parent in list(current.parents)[:6]:
+            candidate = parent / 'improved_diarization.py'
+            if candidate.exists():
+                repo_root = parent
+                break
+
+        if repo_root is None:
+            # Fallback to CWD
+            cwd_candidate = Path.cwd() / 'improved_diarization.py'
+            if cwd_candidate.exists():
+                repo_root = Path.cwd()
+
+        if repo_root is not None:
+            import sys
+            sys.path.insert(0, str(repo_root))
+            mod = import_module('improved_diarization')
+        else:
+            raise ImportError('improved_diarization module not found in repository tree')
+
+    enhance_diarization_pipeline = getattr(mod, 'enhance_diarization_pipeline')
     ENHANCED_DIARIZATION_AVAILABLE = True
     print("✅ Enhanced diarization pipeline loaded successfully")
-except ImportError as e:
+except Exception as e:
     ENHANCED_DIARIZATION_AVAILABLE = False
     logging.warning(f"Enhanced diarization not available - using fallback: {e}")