Update app.py

app.py

@@ -1,5 +1,5 @@
 import os
-# --- 1. FORCE CPU (Driver Error ကာကွယ်ရန်) ---
+# --- 1. FORCE CPU (GPU Error ရှောင်ရန်) ---
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
 
 import sys
@@ -9,7 +9,7 @@ import edge_tts
 import asyncio
 import shutil
 import zipfile
-import urllib.request
+from huggingface_hub import hf_hub_download
 
 # PyTorch CPU Mode
 torch.cuda.is_available = lambda : False
@@ -22,33 +22,34 @@ if not os.path.exists("OpenVoice"):
 
 sys.path.append(os.path.abspath("OpenVoice"))
 
-# V2 Checkpoints Folder ဖန်တီးခြင်း
+# V2 Checkpoints Folder
 os.makedirs("checkpoints_v2", exist_ok=True)
 
-# V2 Model Download Function (Official S3 Link)
+# V2 Model Download Function (Hugging Face Source)
 def download_v2_models():
     # Model ရှိမရှိ စစ်ဆေးခြင်း
     if not os.path.exists("checkpoints_v2/converter/checkpoint.pth"):
-        print("Downloading OpenVoice V2 Model from AWS...")
+        print("Downloading OpenVoice V2 Zip from Hugging Face...")
         try:
-            # 1. Zip ဖိုင်ကို ဒေါင်းလုပ်ဆွဲခြင်း
-            url = "https://myshell-public-repo-hosting.s3.amazonaws.com/openvoice/checkpoints_v2_0417.zip"
-            zip_path = "checkpoints_v2.zip"
-            urllib.request.urlretrieve(url, zip_path)
+            # AWS Link (403 Error) အစား Hugging Face Link ကို သုံးသည်
+            zip_path = hf_hub_download(
+                repo_id="myshell-ai/OpenVoice",
+                filename="checkpoints_v2_0417.zip",
+                local_dir=".",
+                local_dir_use_symlinks=False
+            )
             
-            # 2. ဖြည်ချခြင်း (Unzip)
-            print("Extracting Model...")
+            print("Extracting Model Zip...")
             with zipfile.ZipFile(zip_path, 'r') as zip_ref:
                 zip_ref.extractall(".")
             
-            # 3. Folder နာမည်ပြန်ညှိခြင်း (လိုအပ်လျှင်)
+            # Folder နာမည်ပြန်ညှိခြင်း
             if os.path.exists("checkpoints_v2_0417"):
-                # အဟောင်းဖျက်ပြီး အသစ်နေရာချ
                 if os.path.exists("checkpoints_v2"):
                     shutil.rmtree("checkpoints_v2")
                 os.rename("checkpoints_v2_0417", "checkpoints_v2")
                 
-            print("✅ V2 Model Ready!")
+            print("✅ V2 Model Extracted Successfully!")
         except Exception as e:
             print(f"Download Error: {e}")
 
@@ -65,16 +66,17 @@ except ImportError:
 
 # --- 3. Load V2 Model ---
 ckpt_converter = 'checkpoints_v2/converter'
+if not os.path.exists(f"{ckpt_converter}/config.json"):
+    print("Warning: Config not found, trying fallback path...")
+    ckpt_converter = 'OpenVoice/checkpoints_v2/converter'
 
 print(f"Loading V2 Model from {ckpt_converter}...")
 try:
-    # V2 requires 'device' argument
     tone_color_converter = ToneColorConverter(f'{ckpt_converter}/config.json', device='cpu')
     tone_color_converter.load_ckpt(f'{ckpt_converter}/checkpoint.pth')
     print("✅ V2 Model Loaded Successfully!")
 except Exception as e:
-    print(f"Model Load Error: {e}")
-    # Error တက်ရင် Variable ကို None ပေးထားမည် (UI မှာ Crash မဖြစ်အောင်)
+    print(f"CRITICAL MODEL LOAD ERROR: {e}")
     tone_color_converter = None
 
 # --- 4. Mastering Engine ---
@@ -103,18 +105,17 @@ async def run_edge_tts(text, gender):
 
 def predict(text, ref_audio, gender, mastering_style):
     if tone_color_converter is None:
-        return "Critical Error: Model not loaded. Check logs.", None
+        return "System Error: Model failed to load. Check logs.", None
     if not text: return "စာရိုက်ထည့်ပါ", None
     if not ref_audio: return "Reference Audio ထည့်ပါ", None
 
     try:
-        # Step A: Edge TTS (Myanmar)
+        # Step A: Edge TTS
         base_audio = asyncio.run(run_edge_tts(text, gender))
 
-        # Step B: OpenVoice V2 Conversion
+        # Step B: OpenVoice V2
         os.makedirs("outputs", exist_ok=True)
         
-        # VAD Handling
         try:
             target_se, _ = se_extractor.get_se(ref_audio, tone_color_converter, target_dir='outputs', vad=True)
         except:
@@ -144,7 +145,7 @@ def predict(text, ref_audio, gender, mastering_style):
 # UI
 with gr.Blocks(title="Myanmar OpenVoice V2") as demo:
     gr.Markdown("# 🇲🇲 Myanmar Voice Cloning (OpenVoice V2)")
-    gr.Markdown("V2 Model (AWS Source) + CPU Stable")
+    gr.Markdown("Corrected Download Source (No 403 Error)")
     
     with gr.Row():
         with gr.Column():