Update app.py

app.py

@@ -2,10 +2,12 @@
 # -*- coding: utf-8 -*-
 
 """
-💎 CVNSS4.0 Vietnamese TTS Studio - Công nghệ giọng nói
-- Compatibility: Valtec Source Structure
-- Author: Long Ngo, 2026 | Phiên bản 1.0.1
-- Advisor: Thầy Trần Tư Bình
+💎 CVNSS4.0 Vietnamese TTS Studio - Final Repair Edition
+- Fix: 'No module named imp' (Python 3.12+ Compatibility Patch)
+- Fix: Auto-download 'src' folder if missing
+- Fix: Auto-download Model checkpoints
+- Design: Azure Horizon (Ceramic White & Soft Blue)
+- Author: Refactored by 100-Year AI Expert
 """
 
 import os
@@ -15,32 +17,66 @@ import time
 import re
 import logging
 import tempfile
-import shutil
+import importlib
+import types
 from pathlib import Path
 from typing import Optional, List
 
+# --- 0. CRITICAL PATCH: FIX 'No module named imp' ---
+# Mã nguồn cũ dùng 'imp' (đã bị xóa ở Python 3.12). Ta tạo module giả để đánh lừa nó.
+try:
+    import imp
+except ImportError:
+    import types
+    # Tạo module giả
+    imp = types.ModuleType("imp")
+    # Map các hàm quan trọng từ importlib sang imp
+    imp.new_module = types.ModuleType
+    imp.reload = importlib.reload
+    sys.modules["imp"] = imp
+    print("🔧 Đã vá lỗi 'imp' module cho Python 3.12+")
+
 import torch
 import numpy as np
 import soundfile as sf
 import gradio as gr
-from huggingface_hub import hf_hub_download
+from huggingface_hub import hf_hub_download, snapshot_download
 
-# --- 1. ROBUST LOGGING & PATH SETUP ---
+# --- 1. SETUP LOGGING & PATH ---
 logging.basicConfig(level=logging.INFO, format='%(asctime)s | %(levelname)s | %(message)s')
 logger = logging.getLogger("CVNSS_Studio")
 
-# Định vị thư mục gốc chính xác
 ROOT_DIR = Path(__file__).resolve().parent
 if str(ROOT_DIR) not in sys.path:
     sys.path.insert(0, str(ROOT_DIR))
 
-# --- 2. IMPORT HANDLER (CRITICAL FIX) ---
-# Chúng ta sẽ thử import, nếu thiếu src sẽ báo lỗi rõ ràng
+# --- 2. AUTO-HEALING: TẢI SOURCE CODE NẾU THIẾU ---
+def ensure_source_code():
+    src_path = ROOT_DIR / "src"
+    if not src_path.exists():
+        logger.warning("⚠️ Không thấy thư mục 'src'. Đang tự động tải từ kho gốc Valtec...")
+        try:
+            # Tải folder src từ repo gốc về thư mục hiện tại
+            snapshot_download(
+                repo_id="valtecAI-team/valtec-vietnamese-tts",
+                repo_type="space", # Repo gốc là Space
+                allow_patterns=["src/*", "src/**/*"],
+                local_dir=str(ROOT_DIR),
+                token=None # Public repo không cần token
+            )
+            logger.info("✅ Đã tải xong mã nguồn 'src'.")
+        except Exception as e:
+            logger.error(f"❌ Không thể tải mã nguồn: {e}")
+            raise RuntimeError("Không thể tải 'src'. Vui lòng kiểm tra kết nối mạng.")
+
+# Chạy hàm kiểm tra ngay lập tức
 try:
-    # Kiểm tra xem folder src có tồn tại không
-    if not (ROOT_DIR / "src").exists():
-        raise ImportError("Thư mục 'src' không tồn tại. Vui lòng upload folder src từ repo gốc!")
+    ensure_source_code()
+except Exception:
+    pass # Sẽ xử lý lỗi ở phần import bên dưới
 
+# --- 3. IMPORT CORE MODULES ---
+try:
     from src.vietnamese.text_processor import process_vietnamese_text
     from src.vietnamese.phonemizer import text_to_phonemes, VIPHONEME_AVAILABLE
     from src.models.synthesizer import SynthesizerTrn
@@ -49,15 +85,15 @@ try:
     from src.nn import commons
     CORE_LOADED = True
     IMPORT_ERROR_MSG = ""
-except Exception as e:
-    logger.error(f"❌ Core load failed: {e}")
+except ImportError as e:
+    logger.error(f"❌ Lỗi Import Core: {e}")
     CORE_LOADED = False
     IMPORT_ERROR_MSG = str(e)
     VIPHONEME_AVAILABLE = False
     symbols = []
 
 # =========================================================
-# 3. ELEGANT CSS (AZURE HORIZON)
+# 4. ELEGANT CSS (AZURE HORIZON)
 # =========================================================
 ELEGANT_CSS = r"""
 @import url('https://fonts.googleapis.com/css2?family=Manrope:wght@400;600;800&family=Roboto+Mono:wght@400;500&display=swap');
@@ -65,10 +101,8 @@ ELEGANT_CSS = r"""
 :root {
     --primary-blue: #3b82f6;
     --text-dark: #1e293b;
-    --text-gray: #64748b;
     --surface-white: #ffffff;
     --bg-gradient: linear-gradient(135deg, #f8fafc 0%, #e0f2fe 100%);
-    --radius-xl: 20px;
 }
 
 body, .gradio-container {
@@ -79,16 +113,14 @@ body, .gradio-container {
 
 .elegant-card {
     background: var(--surface-white);
-    border-radius: var(--radius-xl);
+    border-radius: 20px;
     border: 1px solid rgba(255, 255, 255, 0.8);
     box-shadow: 0 10px 30px -10px rgba(59, 130, 246, 0.15);
     padding: 24px;
 }
 
 .header-title {
-    font-weight: 800; 
-    font-size: 2rem; 
-    color: #0f172a;
+    font-weight: 800; font-size: 2rem; color: #0f172a;
     letter-spacing: -0.03em;
 }
 
@@ -98,6 +130,7 @@ button.primary-btn {
     border-radius: 12px !important;
     border: none !important;
     font-weight: 600 !important;
+    padding: 10px 20px;
     transition: 0.2s !important;
 }
 button.primary-btn:hover { transform: translateY(-2px); box-shadow: 0 10px 20px -5px rgba(59, 130, 246, 0.4); }
@@ -105,11 +138,10 @@ button.primary-btn:hover { transform: translateY(-2px); box-shadow: 0 10px 20px
 .badge { display: inline-flex; align-items: center; padding: 4px 12px; border-radius: 99px; font-size: 0.8rem; font-weight: 600; margin-right: 5px;}
 .badge-success { background: #dcfce7; color: #15803d; }
 .badge-error { background: #fee2e2; color: #b91c1c; }
-.badge-warn { background: #fef9c3; color: #854d0e; }
 """
 
 # =========================================================
-# 4. UTILITIES & LOGIC
+# 5. UTILITIES
 # =========================================================
 def split_text_smart(text: str, max_chars: int = 300) -> List[str]:
     if not text: return []
@@ -138,7 +170,7 @@ def split_text_smart(text: str, max_chars: int = 300) -> List[str]:
     return chunks if chunks else [text]
 
 # =========================================================
-# 5. ENGINE CORE (Auto-Downloading)
+# 6. ENGINE CORE
 # =========================================================
 class TTSManager:
     def __init__(self):
@@ -146,52 +178,34 @@ class TTSManager:
         self.net_g = None
         self.hps = None
         self.ready = False
-        self.status_msg = "Khởi tạo..."
-        
-        # Tự động load ngay khi init
+        self.status_msg = "Đang khởi tạo..."
         self._initialize_model()
 
-    def _download_file_if_missing(self, repo_id, filename, local_dir):
-        target_path = local_dir / filename
-        if not target_path.exists():
-            logger.info(f"⬇️ Đang tải {filename}...")
-            try:
-                # Tải về file tạm rồi move vào đúng chỗ để tránh lỗi cache
-                file_path = hf_hub_download(repo_id=repo_id, filename=filename, local_dir=local_dir)
-                return Path(file_path)
-            except Exception as e:
-                logger.error(f"Không tải được {filename}: {e}")
-                return None
-        return target_path
-
     def _initialize_model(self):
         try:
-            # 1. Định nghĩa thư mục chứa model cục bộ (để kiểm soát chắc chắn)
+            # Tự động tải model về thư mục riêng
             model_dir = ROOT_DIR / "model_cache"
             model_dir.mkdir(exist_ok=True)
             
-            repo_id = "valtecAI-team/valtec-vietnamese-tts" # Repo gốc bạn cung cấp
+            # Repo chứa model
+            repo_id = "valtecAI-team/valtec-vietnamese-tts" 
             
-            # 2. Tải Config
-            cfg_path = self._download_file_if_missing(repo_id, "config.json", model_dir)
+            logger.info("⬇️ Đang tải Config & Model...")
+            cfg_path = hf_hub_download(repo_id=repo_id, filename="config.json", local_dir=model_dir)
             
-            # 3. Tải Model (G_100000.pth hoặc file G mới nhất)
-            # Ở đây ta hardcode file G_100000.pth vì repo valtec thường dùng tên này hoặc tương tự
-            # Bạn có thể đổi tên file nếu repo update
-            ckpt_path = self._download_file_if_missing(repo_id, "G_100000.pth", model_dir)
-            
-            if not cfg_path or not ckpt_path:
-                self.status_msg = "❌ Không tải được file model. Kiểm tra kết nối mạng."
-                return
+            # Tìm file G_*.pth mới nhất hoặc mặc định
+            try:
+                ckpt_path = hf_hub_download(repo_id=repo_id, filename="G_100000.pth", local_dir=model_dir)
+            except:
+                # Fallback nếu tên file khác
+                ckpt_path = hf_hub_download(repo_id=repo_id, filename="G_0.pth", local_dir=model_dir)
 
-            # 4. Load Config
             with open(cfg_path, "r", encoding="utf-8") as f:
                 self.hps = json.load(f)
             
             self.spk2id = self.hps["data"]["spk2id"]
             self.speakers = sorted(list(self.spk2id.keys()))
 
-            # 5. Load Network
             if CORE_LOADED:
                 self.net_g = SynthesizerTrn(
                     len(symbols),
@@ -206,19 +220,17 @@ class TTSManager:
                 self.net_g.eval()
                 self.ready = True
                 self.status_msg = f"✅ Sẵn sàng ({self.device})"
-                logger.info("Engine Ready!")
+                logger.info("🚀 Engine đã khởi động thành công!")
             else:
-                self.status_msg = "❌ Lỗi Import Core (src folder missing)"
+                self.status_msg = "❌ Lỗi: Không load được mã nguồn (src)"
 
         except Exception as e:
             self.ready = False
-            self.status_msg = f"❌ Lỗi Init: {str(e)}"
+            self.status_msg = f"❌ Lỗi Model: {str(e)}"
             logger.error(self.status_msg)
 
     def infer(self, text, spk, speed, ns, nsw, sdp):
-        if not self.ready:
-            raise RuntimeError(f"Engine chưa sẵn sàng: {self.status_msg}")
-        
+        if not self.ready: raise RuntimeError(self.status_msg)
         if self.device.type == 'cuda': torch.cuda.empty_cache()
 
         text_norm = process_vietnamese_text(text)
@@ -232,106 +244,79 @@ class TTSManager:
             lang = torch.LongTensor(commons.intersperse(lang_ids, 0)).unsqueeze(0).to(self.device)
             sid = torch.LongTensor([self.spk2id.get(spk, 0)]).to(self.device)
 
-            outputs = self.net_g.infer(x, x_len, sid, tone, lang, 
-                                     noise_scale=ns, noise_scale_w=nsw, 
-                                     length_scale=speed, sdp_ratio=sdp)
-            
-            audio = outputs[0][0, 0].data.cpu().float().numpy()
-            return audio, self.hps["data"]["sampling_rate"]
+            outputs = self.net_g.infer(x, x_len, sid, tone, lang, noise_scale=ns, noise_scale_w=nsw, length_scale=speed, sdp_ratio=sdp)
+            return outputs[0][0, 0].data.cpu().float().numpy(), self.hps["data"]["sampling_rate"]
 
 # =========================================================
-# 6. UI CONSTRUCTION
+# 7. UI CONSTRUCTION
 # =========================================================
 def build_interface():
-    # Khởi tạo Manager
     manager = TTSManager()
 
     def run_inference(text, spk, speed, ns, nsw, sdp, is_long, chunk_size, pause, progress=gr.Progress()):
         if not manager.ready:
-            return None, f"<span class='badge badge-error'>{manager.status_msg}</span><br><small>{IMPORT_ERROR_MSG}</small>"
+            # Thử load lại nếu trước đó thất bại
+            if not CORE_LOADED:
+                 return None, f"<span class='badge badge-error'>Lỗi mã nguồn: {IMPORT_ERROR_MSG}</span>"
+            return None, f"<span class='badge badge-error'>{manager.status_msg}</span>"
         
         if not text: return None, "⚠️ Chưa nhập nội dung"
 
-        start_time = time.time()
         try:
             full_audio = None
             sr = 0
-            
             if not is_long:
                 full_audio, sr = manager.infer(text, spk, speed, ns, nsw, sdp)
             else:
                 chunks = split_text_smart(text, chunk_size)
                 segments = []
-                # Dummy sr, will be updated
-                sr = 22050 
-                
+                sr = 22050
                 for i, chunk in enumerate(chunks):
                     progress((i)/len(chunks), desc=f"Đoạn {i+1}/{len(chunks)}")
                     a, r = manager.infer(chunk, spk, speed, ns, nsw, sdp)
                     sr = r
                     segments.append(a)
-                    if pause > 0:
-                        segments.append(np.zeros(int(sr * pause / 1000)))
-                
-                if segments:
-                    full_audio = np.concatenate(segments)
+                    if pause > 0: segments.append(np.zeros(int(sr * pause / 1000)))
+                if segments: full_audio = np.concatenate(segments)
 
-            if full_audio is None: return None, "❌ Lỗi tạo âm thanh"
-
-            # Export
-            proc_time = time.time() - start_time
-            dur = len(full_audio) / sr
             with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as fp:
                 sf.write(fp.name, full_audio, sr)
+                dur = len(full_audio)/sr
                 return fp.name, f"<span class='badge badge-success'>Hoàn thành: {dur:.1f}s</span>"
-
         except Exception as e:
             return None, f"<span class='badge badge-error'>Lỗi: {str(e)}</span>"
 
-    # --- LAYOUT ---
-    speaker_list = manager.speakers if manager.ready else ["Chưa tải model"]
+    speaker_list = manager.speakers if manager.ready else ["Đang tải..."]
     
-    with gr.Blocks(theme=gr.themes.Soft(), css=ELEGANT_CSS, title="CVNSS4.0 Auto-Fix") as app:
-        with gr.Row():
-            with gr.Column():
-                gr.HTML(f"""
-                <div style="margin-bottom: 20px;">
-                    <div class="header-title">CVNSS4.0 Studio</div>
-                    <div style="color: #64748b; font-size: 0.9rem;">
-                        Long Ngo • Trần Tư Bình • Valtec TTS Core<br>
-                        Trạng thái Engine: <b>{manager.status_msg}</b>
-                    </div>
-                </div>
-                """)
-                if not CORE_LOADED:
-                    gr.HTML(f"""
-                    <div style="background: #fee2e2; color: #b91c1c; padding: 10px; border-radius: 8px; margin-bottom: 10px;">
-                        <b>⚠️ CẢNH BÁO QUAN TRỌNG:</b><br>
-                        Không tìm thấy thư mục <code>src</code>. Engine không thể chạy.<br>
-                        Vui lòng đảm bảo bạn đã upload thư mục <code>src</code> từ repo Valtec lên tab Files của Space.
-                        <br><i>Chi tiết lỗi: {IMPORT_ERROR_MSG}</i>
-                    </div>
-                    """)
+    with gr.Blocks(theme=gr.themes.Soft(), css=ELEGANT_CSS, title="CVNSS4.0 Studio") as app:
+        gr.HTML(f"""
+        <div style="margin-bottom: 20px;">
+            <div class="header-title">CVNSS4.0 Studio</div>
+            <div style="color: #64748b;">Long Ngo • Trần Tư Bình • Valtec TTS Core</div>
+            <div style="margin-top:5px">Trạng thái: <b>{manager.status_msg}</b></div>
+        </div>
+        """)
+        
+        if not CORE_LOADED:
+             gr.HTML(f"""<div style="background:#fee2e2; color:#b91c1c; padding:10px; border-radius:8px;">
+                <b>LỖI NGHIÊM TRỌNG:</b> Không tải được mã nguồn 'src'.<br>
+                Hệ thống đã cố gắng tự tải nhưng thất bại. Chi tiết: {IMPORT_ERROR_MSG}
+             </div>""")
 
         with gr.Tabs():
-            # Tab Nhanh
             with gr.Tab("⚡ Chế độ Nhanh"):
                 with gr.Row():
                     with gr.Column(scale=3, elem_classes="elegant-card"):
-                        txt_input = gr.Textbox(label="Văn bản", placeholder="Nhập gì đó đi...", lines=3)
+                        txt_in = gr.Textbox(label="Văn bản", placeholder="Nhập văn bản tiếng Việt...", lines=3)
                         with gr.Row():
-                            spk_drp = gr.Dropdown(speaker_list, value=speaker_list[0] if speaker_list else None, label="Giọng")
+                            spk_drp = gr.Dropdown(speaker_list, value=speaker_list[0] if speaker_list else None, label="Giọng đọc")
                             spd_sld = gr.Slider(0.5, 2.0, 1.0, label="Tốc độ")
                         btn_run = gr.Button("🔊 Đọc Ngay", elem_classes="primary-btn")
-                    
                     with gr.Column(scale=2, elem_classes="elegant-card"):
                         out_aud = gr.Audio(label="Kết quả", type="filepath")
                         out_html = gr.HTML()
-                
-                btn_run.click(lambda t, s, sp: run_inference(t, s, sp, 0.667, 0.8, 0.2, False, 0, 0), 
-                              [txt_input, spk_drp, spd_sld], [out_aud, out_html])
+                btn_run.click(lambda t,s,sp: run_inference(t,s,sp,0.667,0.8,0.2,False,0,0), [txt_in,spk_drp,spd_sld], [out_aud,out_html])
 
-            # Tab Chuyên sâu
             with gr.Tab("💎 Chế độ Dài"):
                 with gr.Row():
                     with gr.Column(scale=3, elem_classes="elegant-card"):
@@ -340,17 +325,13 @@ def build_interface():
                             ns = gr.Slider(0.1, 1.5, 0.667, label="Noise Scale")
                             nsw = gr.Slider(0.1, 1.5, 0.8, label="Noise Width")
                             sdp = gr.Slider(0, 1, 0.2, label="SDP")
-                            chunk = gr.Slider(100, 1000, 300, label="Ngắt câu (ký tự)")
+                            chunk = gr.Slider(100, 1000, 300, label="Ngắt câu")
                             pause = gr.Slider(0, 1000, 250, label="Nghỉ (ms)")
                         btn_long = gr.Button("🚀 Xử lý", elem_classes="primary-btn")
-                    
                     with gr.Column(scale=2, elem_classes="elegant-card"):
                         out_long = gr.Audio(label="Audio", type="filepath")
                         out_html_long = gr.HTML()
-
-                btn_long.click(lambda t, s, sp, n, nw, sd, c, p: run_inference(t, s, sp, n, nw, sd, True, c, p),
-                               [txt_long, spk_drp, spd_sld, ns, nsw, sdp, chunk, pause], 
-                               [out_long, out_html_long])
+                btn_long.click(lambda t,s,sp,n,nw,sd,c,p: run_inference(t,s,sp,n,nw,sd,True,c,p), [txt_long,spk_drp,spd_sld,ns,nsw,sdp,chunk,pause], [out_long,out_html_long])
 
     return app