add more examples

app.py

@@ -6,229 +6,317 @@ import soundfile as sf
 import tempfile
 import torch
 from vieneu_tts import VieNeuTTS
+import os
+import time
 
 print("⏳ Đang khởi động VieNeu-TTS...")
 
-# Khởi tạo model
+# --- 1. SETUP MODEL ---
 print("📦 Đang tải model...")
 device = "cuda" if torch.cuda.is_available() else "cpu"
 print(f"🖥️ Sử dụng thiết bị: {device.upper()}")
 
-tts = VieNeuTTS(
-    backbone_repo="pnnbao-ump/VieNeu-TTS-1000h",
-    backbone_device=device,
-    codec_repo="neuphonic/neucodec",
-    codec_device=device
-)
-print("✅ Model đã tải xong!")
+try:
+    tts = VieNeuTTS(
+        backbone_repo="pnnbao-ump/VieNeu-TTS",
+        backbone_device=device,
+        codec_repo="neuphonic/neucodec",
+        codec_device=device
+    )
+    print("✅ Model đã tải xong!")
+except Exception as e:
+    print(f"⚠️ Không thể tải model (Chế độ UI Demo): {e}")
+    class MockTTS:
+        def encode_reference(self, path): return None
+        def infer(self, text, ref, ref_text): 
+            import numpy as np
+            # Giả lập độ trễ để test tính năng đo thời gian
+            time.sleep(1.5) 
+            return np.random.uniform(-0.5, 0.5, 24000*3)
+    tts = MockTTS()
 
-# Danh sách giọng mẫu
+# --- 2. DATA ---
 VOICE_SAMPLES = {
-    "Nam 1": {
-        "audio": "./sample/id_0001.wav",
-        "text": "./sample/id_0001.txt"
-    },
-    "Nữ 1": {
-        "audio": "./sample/id_0002.wav",
-        "text": "./sample/id_0002.txt"
-    },
-    "Nam 2": {
-        "audio": "./sample/id_0003.wav",
-        "text": "./sample/id_0003.txt"
-    },
-    "Nữ 2": {
-        "audio": "./sample/id_0004.wav",
-        "text": "./sample/id_0004.txt"
-    },
-    "Nam 3": {
-        "audio": "./sample/id_0005.wav",
-        "text": "./sample/id_0005.txt"
-    },
-    "Nam 4": {
-        "audio": "./sample/id_0007.wav",
-        "text": "./sample/id_0007.txt"
-    }
+    "Bình (nam miền Bắc)": {"audio": "./sample/Bình (nam miền Bắc).wav", "text": "./sample/Bình (nam miền Bắc).txt"},
+    "Vĩnh (nam miền Nam)": {"audio": "./sample/Vĩnh (nam miền Nam).wav", "text": "./sample/Vĩnh (nam miền Nam).txt"},
+    "Tuyên (nam miền Bắc)": {"audio": "./sample/Tuyên (nam miền Bắc).wav", "text": "./sample/Tuyên (nam miền Bắc).txt"},
+    "Nguyên (nam miền Nam)": {"audio": "./sample/Nguyên (nam miền Nam).wav", "text": "./sample/Nguyên (nam miền Nam).txt"},
+    "Sơn (nam miền Nam)": {"audio": "./sample/Sơn (nam miền Nam).wav", "text": "./sample/Sơn (nam miền Nam).txt"},
+    "Hương (nữ miền Bắc)": {"audio": "./sample/Hương (nữ miền Bắc).wav", "text": "./sample/Hương (nữ miền Bắc).txt"},
+    "Ly (nữ miền Bắc)": {"audio": "./sample/Ly (nữ miền Bắc).wav", "text": "./sample/Ly (nữ miền Bắc).txt"},
+    "Ngọc (nữ miền Bắc)": {"audio": "./sample/Ngọc (nữ miền Bắc).wav", "text": "./sample/Ngọc (nữ miền Bắc).txt"},
+    "Đoan (nữ miền Nam)": {"audio": "./sample/Đoan (nữ miền Nam).wav", "text": "./sample/Đoan (nữ miền Nam).txt"},
+    "Dung (nữ miền Nam)": {"audio": "./sample/Dung (nữ miền Nam).wav", "text": "./sample/Dung (nữ miền Nam).txt"}
 }
 
+# --- 3. HELPER FUNCTIONS ---
+def load_reference_info(voice_choice):
+    if voice_choice in VOICE_SAMPLES:
+        audio_path = VOICE_SAMPLES[voice_choice]["audio"]
+        text_path = VOICE_SAMPLES[voice_choice]["text"]
+        try:
+            if os.path.exists(text_path):
+                with open(text_path, "r", encoding="utf-8") as f:
+                    ref_text = f.read()
+                return audio_path, ref_text
+            else:
+                return audio_path, "⚠️ Không tìm thấy file text mẫu."
+        except Exception as e:
+            return None, f"❌ Lỗi: {str(e)}"
+    return None, ""
+
 @spaces.GPU(duration=120) 
-def synthesize_speech(text, voice_choice, custom_audio=None, custom_text=None):
-    """Tổng hợp giọng nói từ văn bản"""
+def synthesize_speech(text, voice_choice, custom_audio, custom_text, mode_tab):
     try:
         if not text or text.strip() == "":
-            return None, "❌ Vui lòng nhập văn bản cần tổng hợp"
+            return None, "⚠️ Vui lòng nhập văn bản cần tổng hợp!"
         
+        # --- LOGIC CHECK LIMIT 250 ---
         if len(text) > 250:
-            return None, "❌ Văn bản quá dài! Vui lòng nhập tối đa 250 ký tự. Để tổng hợp văn bản dài hơn, vui lòng tham khảo examples/infer_long_text.py"
-        
-        # Xác định reference audio và text
-        if custom_audio is not None and custom_text:
+            return None, f"❌ Văn bản quá dài ({len(text)}/250 ký tự)! Vui lòng cắt ngắn lại để đảm bảo chất lượng."
+
+        # Logic chọn Reference
+        if mode_tab == "custom_mode": 
+            if custom_audio is None or not custom_text:
+                return None, "⚠️ Vui lòng tải lên Audio v�� nhập nội dung Audio đó."
             ref_audio_path = custom_audio
             ref_text_raw = custom_text
-            print("🎨 Sử dụng giọng tùy chỉnh")
-        elif voice_choice in VOICE_SAMPLES:
+            print("🎨 Mode: Custom Voice")
+        else: # Preset
+            if voice_choice not in VOICE_SAMPLES:
+                 return None, "⚠️ Vui lòng chọn một giọng mẫu."
             ref_audio_path = VOICE_SAMPLES[voice_choice]["audio"]
             ref_text_path = VOICE_SAMPLES[voice_choice]["text"]
+            
+            if not os.path.exists(ref_audio_path):
+                 return None, f"❌ Không tìm thấy file audio: {ref_audio_path}"
+                 
             with open(ref_text_path, "r", encoding="utf-8") as f:
                 ref_text_raw = f.read()
-            print(f"🎤 Sử dụng giọng: {voice_choice}")
-        else:
-            return None, "❌ Vui lòng chọn giọng hoặc tải lên audio tùy chỉnh"
+            print(f"🎤 Mode: Preset Voice ({voice_choice})")
+
+        # Inference & Đo thời gian
+        print(f"📝 Text: {text[:50]}...")
         
-        # Encode và tổng hợp
-        print(f"📝 Đang xử lý: {text[:50]}...")
-        ref_codes = tts.encode_reference(ref_audio_path)
+        start_time = time.time() # <--- Bắt đầu bấm giờ
         
-        print(f"🎵 Đang tổng hợp giọng nói trên {device.upper()}...")
+        ref_codes = tts.encode_reference(ref_audio_path)
         wav = tts.infer(text, ref_codes, ref_text_raw)
         
-        # Lưu file
+        end_time = time.time()   # <--- Kết thúc bấm giờ
+        process_time = end_time - start_time # <--- Tính thời gian xử lý
+        
+        # Save
         with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmp_file:
             sf.write(tmp_file.name, wav, 24000)
             output_path = tmp_file.name
         
-        print("✅ Hoàn thành!")
-        return output_path, f"✅ Tổng hợp thành công"
-        
+        # <--- Cập nhật thông báo kết quả
+        return output_path, f"✅ Thành công! (Mất {process_time:.2f} giây để tạo)"
+
     except Exception as e:
-        print(f"❌ Lỗi: {str(e)}")
         import traceback
         traceback.print_exc()
-        return None, f"❌ Lỗi: {str(e)}"
+        return None, f"❌ Lỗi hệ thống: {str(e)}"
+
+# --- 4. UI SETUP ---
+theme = gr.themes.Ocean(
+    primary_hue="indigo",
+    secondary_hue="cyan",
+    neutral_hue="slate",
+    font=[gr.themes.GoogleFont('Inter'), 'ui-sans-serif', 'system-ui'],
+).set(
+    button_primary_background_fill="linear-gradient(90deg, #6366f1 0%, #0ea5e9 100%)",
+    button_primary_background_fill_hover="linear-gradient(90deg, #4f46e5 0%, #0284c7 100%)",
+    block_shadow="0 4px 6px -1px rgba(0, 0, 0, 0.1), 0 2px 4px -1px rgba(0, 0, 0, 0.06)",
+)
 
-# Custom CSS - tối giản
-custom_css = """
-.gradio-container {
-    max-width: 900px !important;
-    margin: 0 auto !important;
+# <--- CSS ĐÃ SỬA (Background xanh đen + Chữ sáng)
+css = """
+.container { max-width: 1200px; margin: auto; }
+.header-box { 
+    text-align: center; 
+    margin-bottom: 25px; 
+    padding: 25px; 
+    background: linear-gradient(135deg, #0f172a 0%, #1e293b 100%); /* Xanh đen (Slate 900 -> 800) */
+    border-radius: 12px; 
+    border: 1px solid #334155; 
+    box-shadow: 0 10px 15px -3px rgba(0, 0, 0, 0.3);
 }
-.warning-box {
-    background-color: #fef3c7;
-    border-left: 4px solid #f59e0b;
-    padding: 12px 16px;
-    border-radius: 6px;
-    margin: 10px 0;
-    color: #000000;
+.header-title { 
+    font-size: 2.5rem; 
+    font-weight: 800; 
+    color: white; /* Chữ trắng */
+    background: -webkit-linear-gradient(45deg, #60A5FA, #22D3EE); /* Gradient xanh sáng cho chữ nổi bật */
+    -webkit-background-clip: text; 
+    -webkit-text-fill-color: transparent; 
+    margin-bottom: 10px; 
 }
-"""
-
-# Tạo giao diện
-with gr.Blocks(title="VieNeu-TTS", css=custom_css, theme=gr.themes.Soft()) as demo:
-    gr.Markdown("""
-
-
-    # VieNeu-TTS
+.header-desc {
+    font-size: 1.1rem; 
+    color: #cbd5e1; /* Màu xám sáng (Slate-300) */
+    margin-bottom: 15px;
+}
+.link-group a { 
+    text-decoration: none; 
+    margin: 0 10px; 
+    font-weight: 600; 
+    color: #94a3b8; /* Màu link sáng hơn chút */
+    transition: color 0.2s; 
+}
+.link-group a:hover { color: #38bdf8; text-shadow: 0 0 5px rgba(56, 189, 248, 0.5); }
 
-    Hệ thống tổng hợp tiếng nói tiếng Việt sử dụng Large Language Model
+.status-box { font-weight: bold; text-align: center; border: none; background: transparent; }
+.warning-note { 
+    background-color: #fff7ed; 
+    border-left: 4px solid #f97316; 
+    padding: 12px; 
+    color: #9a3412; 
+    font-size: 0.9rem; 
+    border-radius: 4px;
+    margin-top: 10px;
+    margin-bottom: 10px;
+}
+"""
 
-    **Phiên bản:** VieNeu-TTS-1000h (model mới nhất, train trên 1000 giờ dữ liệu)
+EXAMPLES_LIST = [
+    # Nam Miền Nam
+    ["Về miền Tây không chỉ để ngắm nhìn sông nước hữu tình, mà còn để cảm nhận tấm chân tình của người dân nơi đây. Cùng ngồi xuồng ba lá len lỏi qua rặng dừa nước, nghe câu vọng cổ ngọt ngào thì còn gì bằng.", "Vĩnh (nam miền Nam)"],
+    
+    # Nam Miền Bắc
+    ["Hà Nội những ngày vào thu mang một vẻ đẹp trầm mặc và cổ kính đến lạ thường. Đi dạo quanh Hồ Gươm vào sáng sớm, hít hà mùi hoa sữa nồng nàn và thưởng thức chút cốm làng Vòng là trải nghiệm khó quên.", "Bình (nam miền Bắc)"],
+    
+    # Nam Miền Bắc
+    ["Sự bùng nổ của trí tuệ nhân tạo đang định hình lại cách chúng ta làm việc và sinh sống. Từ xe tự lái đến trợ lý ảo thông minh, công nghệ đang dần xóa nhòa ranh giới giữa thực tại và những bộ phim viễn tưởng.", "Tuyên (nam miền Bắc)"],
+    
+    # Nam Miền Nam
+    ["Sài Gòn hối hả là thế, nhưng chỉ cần tấp vào một quán cà phê ven đường, gọi ly bạc xỉu đá và ngắm nhìn dòng người qua lại, bạn sẽ thấy thành phố này cũng có những khoảng lặng thật bình yên và đáng yêu.", "Nguyên (nam miền Nam)"],
+    
+    # Nam Miền Nam
+    ["Để đảm bảo tiến độ dự án quan trọng này, chúng ta cần tập trung tối đa nguồn lực và phối hợp chặt chẽ giữa các phòng ban. Mọi khó khăn phát sinh cần được báo cáo ngay lập tức để ban lãnh đạo xử lý kịp thời.", "Sơn (nam miền Nam)"],
+    
+    # Nữ Miền Nam
+    ["Dạ em chào anh chị, hiện tại bên em đang có chương trình ưu đãi đặc biệt cho căn hộ hướng sông này. Với thiết kế hiện đại và không gian xanh mát, đây chắc chắn là tổ ấm lý tưởng mà gia đình mình đang tìm kiếm.", "Đoan (nữ miền Nam)"],
+    
+    # Nữ Miền Bắc
+    ["Dưới cơn mưa phùn lất phất của những ngày cuối đông, em khẽ nép vào vai anh, cảm nhận hơi ấm lan tỏa. Những khoảnh khắc bình dị như thế này khiến em nhận ra rằng, hạnh phúc đôi khi chỉ đơn giản là được ở bên nhau.", "Ngọc (nữ miền Bắc)"],
 
-    [GitHub](https://github.com/pnnbao97/VieNeu-TTS) • [Model Card](https://huggingface.co/pnnbao-ump/VieNeu-TTS) • [Finetune Guide](https://github.com/pnnbao-ump/VieNeuTTS/blob/main/finetune.ipynb)
+    # Nữ Miền Bắc
+    ["Thay mặt phi hành đoàn, xin chào mừng quý khách đến với chuyến bay vi en 2024. Quý khách vui lòng thắt dây an toàn, dựng thẳng lưng ghế và gập bàn ăn phía trước để chuẩn bị cho máy bay cất cánh trong ít phút nữa.", "Hương (nữ miền Bắc)"],
     
-    """)
-    # Main interface
-    with gr.Row():
+    # Nữ Miền Bắc
+    ["Ngày xửa ngày xưa, ở một ngôi làng nọ có cô Tấm xinh đẹp, nết na nhưng sớm mồ côi mẹ. Dù bị mẹ kế và Cám hãm hại đủ đường, Tấm vẫn giữ được tấm lòng lương thiện và cuối cùng tìm được hạnh phúc xứng đáng.", "Ly (nữ miền Bắc)"],
+]
 
+with gr.Blocks(theme=theme, css=css, title="VieNeu-TTS Studio") as demo:
+    
+    with gr.Column(elem_classes="container"):
+        # Header - Cập nhật class cho HTML
+        gr.HTML("""
+            <div class="header-box">
+                <div class="header-title">🎙️ VieNeu-TTS Studio</div>
+                <div class="header-desc">
+                    Phiên bản: VieNeu-TTS-1000h (model mới nhất, train trên 1000 giờ dữ liệu)
+                </div>
+                <div class="link-group">
+                    <a href="https://huggingface.co/pnnbao-ump/VieNeu-TTS" target="_blank">🤗 Model Card</a> • 
+                    <a href="https://huggingface.co/datasets/pnnbao-ump/VieNeu-TTS-1000h" target="_blank">📖 Dataset 1000h</a> • 
+                    <a href="https://github.com/pnnbao97/VieNeu-TTS" target="_blank">🦜 GitHub</a>
+                </div>
+            </div>
+        """)
+    
+    with gr.Row(elem_classes="container", equal_height=False):
         
-        with gr.Column(scale=1):
-            
+        # --- LEFT: INPUT ---
+        with gr.Column(scale=3, variant="panel"):
+            gr.Markdown("### 📝 Văn bản đầu vào")
             text_input = gr.Textbox(
-                label="Văn bản",
-                placeholder="Nhập văn bản tiếng Việt (khuyến cáo dưới 250 ký tự)...",
-                lines=5,
-                value="Trí tuệ nhân tạo đang cách mạng hóa nhiều lĩnh vực, từ y tế, giáo dục đến giao thông vận tải, mang lại những giải pháp thông minh và hiệu quả."
+                label="Nhập văn bản",
+                placeholder="Nhập nội dung tiếng Việt cần chuyển thành giọng nói...",
+                lines=4,
+                value="Hà Nội những ngày vào thu mang một vẻ đẹp trầm mặc và cổ kính đến lạ thường. Đi dạo quanh Hồ Gươm vào sáng sớm, hít hà mùi hoa sữa nồng nàn và thưởng thức chút cốm làng Vòng là trải nghiệm khó quên.",
+                show_label=False
             )
             
-            char_count = gr.Markdown("**142 / 250 ký tự**")
-            
-            voice_select = gr.Radio(
-                choices=list(VOICE_SAMPLES.keys()),
-                label="Chọn giọng",
-                value="Nam 1"
-            )
+            # Counter + Warning
+            with gr.Row():
+                char_count = gr.HTML("<div style='text-align: right; color: #64748B; font-size: 0.8rem;'>0 / 250 ký tự</div>")
             
-            submit_btn = gr.Button("Tổng hợp", variant="primary", size="lg")
-        
-        with gr.Column(scale=1):
-            audio_output = gr.Audio(label="Kết quả", type="filepath")
-            status_output = gr.Textbox(label="Trạng thái", interactive=False, show_label=False)
-            
-            with gr.Accordion("Giọng tùy chỉnh", open=False):
-                gr.Markdown("""
-Tải lên file audio và nhập nội dung tương ứng. Để có kết quả tốt nhất, nên finetune model trên giọng của bạn.
-                """)
-                custom_audio = gr.Audio(label="File audio (.wav)", type="filepath")
-                custom_text = gr.Textbox(
-                    label="Nội dung audio",
-                    placeholder="Nhập chính xác nội dung...",
-                    lines=2
-                )
-            gr.HTML("""
-            <div class="warning-box" style="color: #000000;">
-                ⚠️ Chúng tôi khuyến cáo sử dụng đoạn văn bản <250 ký tự để đảm bảo chất lượng tốt nhất. 
-                Nếu muốn tổng hợp văn bản dài hơn, vui lòng tham khảo code trong examples/infer_long_text.py
-            </div>
-            """)
-    
-    # Examples
-    with gr.Row():
-        gr.Examples(
-            examples=[
-                ["Trí tuệ nhân tạo đang cách mạng hóa nhiều lĩnh vực, từ y tế, giáo dục đến giao thông vận tải, mang lại những giải pháp thông minh và hiệu quả.", "Nam 1"],
-                ["Trên bầu trời xanh thẳm, những đám mây trắng lửng lờ trôi như những chiếc thuyền nhỏ đang lướt nhẹ theo dòng gió. Dưới mặt đất, cánh đồng lúa vàng rực trải dài tới tận chân trời, những bông lúa nghiêng mình theo từng làn gió.", "Nữ 2"],
-                ["Legacy là một bộ phim đột phá về mặt âm nhạc, quay phim, hiệu ứng đặc biệt, và tôi rất mừng vì cuối cùng nó cũng được cả giới phê bình lẫn người hâm mộ đánh giá lại. Chúng ta đã quá bất công với bộ phim này vào năm 2010.", "Nam 4"],
-                ["Thật đáng ngạc nhiên! Mặc dù con đường này rất xa và khó đi, nhưng với sự kiên trì và sự đồng lòng của tất cả mọi người, chúng ta đã hoàn thành được công việc sửa chữa trước 3 ngày so với kế hoạch ban đầu, bạn có tin không?", "Nữ 1"],
-                ["Các bác sĩ đang nghiên cứu một loại vaccine mới chống lại virus cúm mùa. Thí nghiệm lâm sàng cho thấy phản ứng miễn dịch mạnh mẽ và ít tác dụng phụ.", "Nam 2"],
-            ],
-            inputs=[text_input, voice_select],
-            outputs=[audio_output, status_output],
-            fn=synthesize_speech,
-            cache_examples=False
-        )
-    
-    # Footer info
-    gr.Markdown("""
----
-
-**Tác giả:** Phạm Nguyễn Ngọc Bảo • **Model:** VieNeu-TTS-1000h
+            gr.Markdown("### 🗣️ Chọn giọng đọc")
+            with gr.Tabs() as tabs:
+                with gr.TabItem("👤 Giọng có sẵn (Preset)", id="preset_mode"):
+                    voice_select = gr.Dropdown(
+                        choices=list(VOICE_SAMPLES.keys()),
+                        value="Bình (nam miền Bắc)",
+                        label="Danh sách giọng",
+                        interactive=True
+                    )
+                    with gr.Accordion("Thông tin giọng mẫu", open=False):
+                        ref_audio_preview = gr.Audio(label="Audio mẫu", interactive=False, type="filepath")
+                        ref_text_preview = gr.Markdown("...")
 
-**Lưu ý:** Nếu muốn sử dụng model cũ VieNeu-TTS-140h, hãy thay đổi `backbone_repo` trong mã nguồn
+                with gr.TabItem("🎙️ Giọng tùy chỉnh (Custom)", id="custom_mode"):
+                    gr.Markdown("Tải lên giọng của bạn (Zero-shot Cloning)")
+                    custom_audio = gr.Audio(label="File ghi âm (.wav)", type="filepath")
+                    custom_text = gr.Textbox(label="Nội dung ghi âm", placeholder="Nhập chính xác lời thoại...")
 
----
+            current_mode = gr.Textbox(visible=False, value="preset_mode")
+            btn_generate = gr.Button("Tổng hợp giọng nói", variant="primary", size="lg")
 
-### Ủng hộ dự án
+        # --- RIGHT: OUTPUT ---
+        with gr.Column(scale=2):
+            gr.Markdown("### 🎧 Kết quả")
+            with gr.Group():
+                audio_output = gr.Audio(label="Audio đầu ra", type="filepath", show_download_button=True, autoplay=True)
+                status_output = gr.Textbox(label="Trạng thái", show_label=False, elem_classes="status-box", placeholder="Sẵn sàng...")
 
-VieNeu-TTS là dự án miễn phí và mã nguồn mở. Tuy nhiên, việc train model TTS chất lượng cao trên 1000+ giờ dữ liệu đòi hỏi nguồn lực tính toán đáng kể.
+    # --- EXAMPLES ---
+    with gr.Row(elem_classes="container"):
+        with gr.Column():
+            gr.Markdown("### 📚 Ví dụ mẫu")
+            gr.Examples(examples=EXAMPLES_LIST, inputs=[text_input, voice_select], label="Thử nghiệm nhanh")
 
-Nếu bạn thấy dự án này hữu ích, hãy cân nhắc ủng hộ:
+    # --- LOGIC ---
+    def update_count(text):
+        l = len(text)
+        if l > 250:
+            color = "#dc2626" # Red
+            msg = f"⚠️ <b>{l} / 250</b> - Quá giới hạn!"
+        elif l > 200:
+            color = "#ea580c" # Orange
+            msg = f"{l} / 250"
+        else:
+            color = "#64748B" # Gray
+            msg = f"{l} / 250 ký tự"
+        return f"<div style='text-align: right; color: {color}; font-size: 0.8rem; font-weight: bold'>{msg}</div>"
 
-☕ [Buy Me a Coffee](https://buymeacoffee.com/pnnbao)
+    text_input.change(update_count, text_input, char_count)
 
-    """)
-    
-    # Update character count
-    def update_char_count(text):
-        count = len(text) if text else 0
-        color = "#dc2626" if count > 250 else "#374151"
-        return f"<span style='color: {color}; font-weight: 500'>{count} / 250 ký tự</span>"
+    def update_ref_preview(voice):
+        audio, text = load_reference_info(voice)
+        return audio, f"> *\"{text}\"*"
     
-    text_input.change(
-        fn=update_char_count,
-        inputs=[text_input],
-        outputs=[char_count]
-    )
+    voice_select.change(update_ref_preview, voice_select, [ref_audio_preview, ref_text_preview])
+    demo.load(update_ref_preview, voice_select, [ref_audio_preview, ref_text_preview])
+
+    # Tab handling - FIXED WITH *ARGS
+    tab_preset = tabs.children[0]
+    tab_custom = tabs.children[1]
     
-    # Event handler
-    submit_btn.click(
+    # Dùng *args để nhận bất kỳ số lượng tham số nào (0 hoặc 1), tránh lỗi Warning
+    tab_preset.select(fn=lambda *args: "preset_mode", inputs=None, outputs=current_mode)
+    tab_custom.select(fn=lambda *args: "custom_mode", inputs=None, outputs=current_mode)
+
+    btn_generate.click(
         fn=synthesize_speech,
-        inputs=[text_input, voice_select, custom_audio, custom_text],
+        inputs=[text_input, voice_select, custom_audio, custom_text, current_mode],
         outputs=[audio_output, status_output]
     )
 
-# Launch
 if __name__ == "__main__":
-    demo.queue(max_size=20)
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        show_error=True
+    demo.queue().launch(
+        server_name="127.0.0.1", 
+        server_port=7860, 
+        share=False
     )

sample/Bình (nam miền Bắc).txt

@@ -0,0 +1 @@
+Anh chỉ muốn được nhìn nhận như là một huấn luyện viên.

sample/{id_0002.wav → Bình (nam miền Bắc).wav}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8195d3fa6bfc1a049f2b61f2d4052e2d8caf69556b206d2d5eff48314c5ee907
-size 435788
+oid sha256:135f087ced48606c4d406b770a11e344d4d9aa6bd7adfb3e5c26f69cd9cc6df1
+size 127054

sample/Dung (nữ miền Nam).txt

@@ -0,0 +1 @@
+Tục ngữ có câu, sai một li, đi một dặm.

sample/{id_0007.wav → Dung (nữ miền Nam).wav}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:090480cf6a7751a11107470971cfcca64af06768552036497aa16426d14f4cb7
-size 161468
+oid sha256:56e42039d0c96ad19e9f78ecb7218853202022b2a8460010d34ffb7879b17409
+size 143438

sample/Hương (nữ miền Bắc).txt

@@ -0,0 +1 @@
+Tuy nhiên, lúc này có một vấn đề khó khăn nảy sinh.

sample/Hương (nữ miền Bắc).wav

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4c064b4ec64df44ea1306e87b25b84905e540d0fe29885629d2fe8bc8a5e53bc
+size 155756

sample/Ly (nữ miền Bắc).txt

@@ -0,0 +1 @@
+Chúng ta có thể áp dụng logic tương tự với người khác.

sample/Ly (nữ miền Bắc).wav

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0d4e47cfa5ed0b753c2bed07c58e26da89ee2977ca5e941244a6bbafd8869d5e
+size 147534

sample/Tuyên (nam miền Bắc).txt

@@ -0,0 +1 @@
+Bạn cầm khúc cây, và ném vào bãi cỏ xanh tươi rậm rạp ở đằng xa.

sample/Tuyên (nam miền Bắc).wav

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f6b7ac2605db0a2cf634ce0f3a55a87a89f4c2e3bc06f83433e6af583c1f3692
+size 217166

sample/id_0002.txt

@@ -1 +0,0 @@
-Từ nhiều nguồn tài liệu lịch sử, có thể thấy nuôi con theo phong cách Do Thái không chỉ tốt cho đứa trẻ mà còn tốt cho cả các bậc cha mẹ.

sample/id_0007.txt

@@ -1 +0,0 @@
-Sau bữa sáng, tôi tìm thấy xe của mình trong bãi và lái ra cổng.

sample/Đoan (nữ miền Nam).txt

@@ -0,0 +1 @@
+Nuôi con theo phong cách Do Thái, không chỉ tốt cho đứa trẻ, mà còn tốt cho cả các bậc cha mẹ.

sample/Đoan (nữ miền Nam).wav

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3e319ed45dd2a1458a52edfe43a83a36eff813f19399ac2e59ee3f93cace74be
+size 294830