Add __init__.py files and fix sys.path for rvc_logic

pipeline/rvc_training.py

@@ -1,4 +1,5 @@
 import os
+import sys
 import shutil
 import logging
 import traceback
@@ -7,6 +8,12 @@ from huggingface_hub import HfApi
 # Set up logging
 logger = logging.getLogger(__name__)
 
+# Add the parent directory (app root) to sys.path so we can import rvc_logic
+# since this script is in the 'pipeline' subdirectory
+app_root = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
+if app_root not in sys.path:
+    sys.path.insert(0, app_root)
+
 try:
     import spaces
 except ImportError:
@@ -23,18 +30,24 @@ os.environ["URVC_AUDIO_DIR"] = os.path.abspath("rvc_audio")
 os.environ["URVC_TEMP_DIR"] = os.path.abspath("rvc_temp")
 
 # Import the core functions from our LOCAL rvc_logic
+RVC_IMPORT_ERROR = None
 try:
     from rvc_logic.core_train import prepare, extract, train
     from rvc_logic.typing_extra import TrainingSampleRate, F0Method, EmbedderModel
     RVC_LOGIC_AVAILABLE = True
 except ImportError as e:
     logger.error(f"Failed to import rvc_logic: {e}")
+    RVC_IMPORT_ERROR = str(e)
+    RVC_LOGIC_AVAILABLE = False
+except Exception as e:
+    logger.error(f"Unexpected error loading rvc_logic: {e}")
+    RVC_IMPORT_ERROR = str(e)
     RVC_LOGIC_AVAILABLE = False
 
 @spaces.GPU(duration=1000)
 def train_rvc_model(audio_path, model_name, epochs=100, progress=None):
     if not RVC_LOGIC_AVAILABLE:
-        return "Error: rvc_logic module not found in the project.", None
+        return f"Error: rvc_logic module failed to load. Reason: {RVC_IMPORT_ERROR}", None
     
     if not audio_path:
         return "Error: Please upload an audio file.", None

requirements.txt

@@ -12,6 +12,7 @@ numpy<2.0
 soxr
 ffmpeg-python>=0.2.0
 pedalboard
+lazy-loader
 
 # Demucs (stem separation)
 demucs