Upload app.py with huggingface_hub

app.py

@@ -14,31 +14,66 @@ from huggingface_hub import hf_hub_download, snapshot_download
 from pathlib import Path
 
 # Try to import f5_tts - handle different possible locations
-try:
-    from f5_tts.api import F5TTS
-    from f5_tts.infer.utils_infer import preprocess_ref_audio_text
-except ImportError:
-    # Try adding local paths
-    current_dir = os.path.dirname(__file__)
-    possible_paths = [
-        os.path.join(current_dir, "vakya_model"),
-        os.path.join(current_dir, "f5_tts"),
-        os.path.join(current_dir, "..", "vakya_model"),
-    ]
-    for path in possible_paths:
-        if os.path.exists(path):
-            sys.path.insert(0, path)
-            try:
-                from f5_tts.api import F5TTS
-                from f5_tts.infer.utils_infer import preprocess_ref_audio_text
-                break
-            except ImportError:
-                continue
+# The f5_tts directory should be in the same directory as app.py
+current_dir = os.path.dirname(os.path.abspath(__file__))
+
+# Add current directory to path (so we can import f5_tts if it's in the same dir)
+if current_dir not in sys.path:
+    sys.path.insert(0, current_dir)
+
+# Also check for f5_tts in common locations
+possible_parent_paths = [
+    current_dir,  # Same directory as app.py
+    os.path.join(current_dir, ".."),  # Parent directory
+    "/app",  # Common HF Spaces location
+]
+
+f5_tts_imported = False
+for parent_path in possible_parent_paths:
+    parent_path = os.path.abspath(parent_path)
+    f5_tts_path = os.path.join(parent_path, "f5_tts")
+    
+    if os.path.exists(f5_tts_path) and os.path.isdir(f5_tts_path):
+        # Add parent directory to path (not f5_tts itself)
+        if parent_path not in sys.path:
+            sys.path.insert(0, parent_path)
+        try:
+            from f5_tts.api import F5TTS
+            from f5_tts.infer.utils_infer import preprocess_ref_audio_text
+            f5_tts_imported = True
+            print(f"✅ Successfully imported f5_tts from {parent_path}")
+            break
+        except ImportError as e:
+            print(f"⚠️  Tried {parent_path}, but import failed: {e}")
+            continue
+
+if not f5_tts_imported:
+    # Try direct import (in case it's installed as a package)
+    try:
+        from f5_tts.api import F5TTS
+        from f5_tts.infer.utils_infer import preprocess_ref_audio_text
+        f5_tts_imported = True
+        print("✅ Successfully imported f5_tts (installed package)")
+    except ImportError:
+        pass
+
+if not f5_tts_imported:
+    # Print debug information
+    print(f"❌ Current directory: {current_dir}")
+    print(f"❌ Python path: {sys.path[:5]}")
+    print(f"❌ Checking for f5_tts in: {current_dir}")
+    if os.path.exists(os.path.join(current_dir, "f5_tts")):
+        print(f"   ✅ f5_tts directory exists at: {os.path.join(current_dir, 'f5_tts')}")
+        print(f"   📁 Contents: {os.listdir(os.path.join(current_dir, 'f5_tts'))[:10]}")
     else:
-        raise ImportError(
-            "Could not import f5_tts. Please ensure the model code is available. "
-            "You may need to include the f5_tts directory in your Space or install it as a package."
-        )
+        print(f"   ❌ f5_tts directory NOT found")
+    
+    raise ImportError(
+        f"Could not import f5_tts. Please ensure the model code is available.\n"
+        f"Current directory: {current_dir}\n"
+        f"Looking for f5_tts in: {current_dir}\n"
+        f"Python path: {sys.path[:3]}"
+    )
 
 # Model configuration
 MODEL_REPO_ID = "ashishkblink/vakya2.0"