Update log.py

log.py

@@ -7,3 +7,64 @@ logger.setLevel(logging.INFO)
 ch = logging.StreamHandler()
 ch.setFormatter(logging.Formatter("[%(levelname)s] %(message)s"))
 logger.addHandler(ch)
+
+# Auto model/data setup from anywhere
+import os
+import json
+from pathlib import Path
+from mdx_core import MDX
+
+BASE_DIR = Path(__file__).resolve().parent
+mdxnet_models_dir = BASE_DIR / "mdx_models"
+
+MODEL_PRESETS = {
+    "UVR-MDX-NET-Voc_FT.onnx": {
+        "mdx_dim_f_set": 2048,
+        "mdx_dim_t_set": 3,
+        "mdx_n_fft_scale_set": 6144,
+        "primary_stem": "Vocals",
+        "compensate": 1.035
+    },
+    "UVR_MDXNET_KARA_2.onnx": {
+        "mdx_dim_f_set": 1024,
+        "mdx_dim_t_set": 3,
+        "mdx_n_fft_scale_set": 4096,
+        "primary_stem": "Main",
+        "compensate": 1.035
+    },
+    "UVR-MDX-NET-Inst_HQ_4.onnx": {
+        "mdx_dim_f_set": 2048,
+        "mdx_dim_t_set": 3,
+        "mdx_n_fft_scale_set": 6144,
+        "primary_stem": "Instrumental",
+        "compensate": 1.0
+    },
+    "Reverb_HQ_By_FoxJoy.onnx": {
+        "mdx_dim_f_set": 2048,
+        "mdx_dim_t_set": 3,
+        "mdx_n_fft_scale_set": 6144,
+        "primary_stem": "Vocals",
+        "compensate": 1.035
+    },
+}
+
+
+def initialize_models():
+    mdxnet_models_dir.mkdir(parents=True, exist_ok=True)
+    json_path = mdxnet_models_dir / "data.json"
+
+    if json_path.exists():
+        logger.info("✅ data.json already exists.")
+        return
+
+    logger.info("🔧 Generating data.json from local ONNX models")
+    data = {}
+    for filename, params in MODEL_PRESETS.items():
+        model_path = mdxnet_models_dir / filename
+        if model_path.exists():
+            model_hash = MDX.get_hash(str(model_path))
+            data[model_hash] = params
+
+    with open(json_path, "w") as f:
+        json.dump(data, f, indent=2)
+    logger.info(f"✅ Created data.json with {len(data)} entries.")