Update app.py

app.py

@@ -2,6 +2,44 @@ import datetime
 from pathlib import Path
 import gradio as gr
 import random
+import os
+import shutil
+from huggingface_hub import snapshot_download
+
+# --- LOGIKA AUTO-DOWNLOAD BERT (Hanya Folder BERT) ---
+def download_bert_assets():
+    print("Checking for BERT assets...")
+    REPO_ID = "Plana-Archive/Plana-TTS"
+    SOURCE_SUBFOLDER = "sbv2-chupa-demo/bert"
+    # Target folder di root Space harus bernama 'bert'
+    DEST_FOLDER = "./bert"
+    
+    if not os.path.exists(DEST_FOLDER):
+        try:
+            print(f"Downloading BERT assets from {REPO_ID}...")
+            # Mengunduh folder bert ke folder sementara
+            temp_dir = snapshot_download(
+                repo_id=REPO_ID,
+                allow_patterns=[f"{SOURCE_SUBFOLDER}/**/*"],
+                token=os.getenv("HF_TOKEN")
+            )
+            
+            # Memindahkan isi dari folder hasil download ke root/bert
+            src_path = os.path.join(temp_dir, SOURCE_SUBFOLDER)
+            if os.path.exists(src_path):
+                shutil.copytree(src_path, DEST_FOLDER)
+                print("✅ BERT assets downloaded and linked successfully.")
+            else:
+                print("⚠️ BERT path not found in downloaded snapshot.")
+        except Exception as e:
+            print(f"❌ Failed to download BERT: {e}")
+    else:
+        print("✅ BERT assets already exist.")
+
+# Jalankan download sebelum import model
+download_bert_assets()
+
+# --- LANJUT KE LOGIKA ASLI APP.PY ---
 from style_bert_vits2.constants import (
     DEFAULT_LENGTH,
     DEFAULT_LINE_SPLIT,
@@ -250,4 +288,4 @@ if __name__ == "__main__":
     device = "cuda" if torch.cuda.is_available() else "cpu"
     model_holder = TTSModelHolder(Path("model_assets"), device)
     app = create_inference_app(model_holder)
-    app.launch(inbrowser=True)
+    app.launch(inbrowser=True)