Update main.py

main.py

@@ -1,6 +1,7 @@
 # main.py
 
 import os
+from pathlib import Path
 from gui import get_gui
 from utils import download_manager
 
@@ -12,21 +13,21 @@ UVR_MODELS = [
     "UVR-MDX-NET-Inst_HQ_4.onnx",
 ]
 
-BASE_DIR = os.path.dirname(os.path.abspath(__file__))
-mdxnet_models_dir = os.path.join(BASE_DIR, "mdx_models")
+BASE_DIR = Path(__file__).resolve().parent
+mdxnet_models_dir = BASE_DIR / "mdx_models"
 
-# Ensure mdx_models is a directory, not a file
-if os.path.exists(mdxnet_models_dir) and not os.path.isdir(mdxnet_models_dir):
-    os.remove(mdxnet_models_dir)
-os.makedirs(mdxnet_models_dir, exist_ok=True)
+# Ensure mdx_models is a directory
+if mdxnet_models_dir.exists() and not mdxnet_models_dir.is_dir():
+    mdxnet_models_dir.unlink()
+mdxnet_models_dir.mkdir(parents=True, exist_ok=True)
 
 # Debug: confirm visibility of data.json
-json_path = os.path.join(mdxnet_models_dir, "data.json")
+json_path = mdxnet_models_dir / "data.json"
 print("🔍 Checking for data.json at:", json_path)
-print("📁 mdx_models contents:", os.listdir(mdxnet_models_dir))
+print("📁 mdx_models contents:", list(mdxnet_models_dir.iterdir()))
 print("📂 Current working directory:", os.getcwd())
 
-if not os.path.exists(json_path):
+if not json_path.exists():
     print("❌ data.json NOT FOUND! Check if it's committed or misplaced.")
 else:
     print("✅ data.json found.")
@@ -35,7 +36,7 @@ else:
 def download_models():
     for model in UVR_MODELS:
         url = os.path.join(MDX_DOWNLOAD_LINK, model)
-        download_manager(url, mdxnet_models_dir)
+        download_manager(url, str(mdxnet_models_dir))
 
 
 if __name__ == "__main__":
@@ -49,4 +50,4 @@ if __name__ == "__main__":
         show_error=True,
         quiet=False,
         debug=False,
-    )
+    )