Update app.py

app.py

@@ -37,12 +37,10 @@ LOAD_DETOKENIZER = os.getenv("KIMI_LOAD_DETOKENIZER", "1") == "1"
 # Kimi-Audio install (once per Space container)
 # ---------------------------------------------------------------------------
 
-
 def _ensure_kimia_infer():
     """Clone Moonshot repo with submodules and install editable."""
     try:
         from kimia_infer.api.kimia import KimiAudio  # type: ignore
-
         return KimiAudio
     except ImportError:
         pass
@@ -51,31 +49,49 @@ def _ensure_kimia_infer():
         print(f"📦 Cloning Kimi-Audio → {KIMI_AUDIO_DIR}")
         subprocess.run(
             [
-                "git",
-                "clone",
-                "--recursive",
-                "--depth",
-                "1",
-                KIMI_AUDIO_REPO,
-                KIMI_AUDIO_DIR,
+                "git", "clone", "--recursive", "--depth", "1",
+                KIMI_AUDIO_REPO, KIMI_AUDIO_DIR,
             ],
             check=True,
         )
 
+    # Safety net: make the source tree importable even if the editable
+    # install's .pth is not picked up by the current process.
+    if KIMI_AUDIO_DIR not in sys.path:
+        sys.path.insert(0, KIMI_AUDIO_DIR)
+
     print("🎵 Installing kimia_infer (editable)...")
-    subprocess.run(
+    # Use --no-deps so flash-attn does not try to rebuild.
+    # (We will handle flash-attn separately if needed.)
+    result = subprocess.run(
         [
             sys.executable, "-m", "pip", "install", "-q",
             "-e", KIMI_AUDIO_DIR,
-            "--no-build-isolation",   # ← critical
+            "--no-deps",
+            "--no-build-isolation",
         ],
-        check=True,
+        capture_output=True,
+        text=True,
     )
-
-    from kimia_infer.api.kimia import KimiAudio  # type: ignore
-
-    return KimiAudio
-
+    if result.returncode != 0:
+        print("pip install failed:")
+        print(result.stdout)
+        print(result.stderr)
+        # Still try the import via sys.path — often enough for kimia_infer
+        # because most of it is pure Python.
+
+    # Final import attempt
+    try:
+        from kimia_infer.api.kimia import KimiAudio  # type: ignore
+        print("✅ kimia_infer imported successfully")
+        return KimiAudio
+    except ImportError as e:
+        # Extra debug
+        print(f"❌ Still cannot import kimia_infer: {e}")
+        print(f"sys.path[0:5] = {sys.path[:5]}")
+        print(f"Contents of {KIMI_AUDIO_DIR}:")
+        subprocess.run(["ls", "-la", KIMI_AUDIO_DIR], check=False)
+        raise
 
 def _apply_transformers_patches() -> None:
     """Bridge Kimi-Audio remote code with common HF Space transformers builds."""