app.py

import os
import tempfile

import gradio as gr
from huggingface_hub import login
from cached_path import cached_path
from vinorm import TTSnorm

from f5_tts.model import DiT
from f5_tts.infer.utils_infer import (
    preprocess_ref_audio_text,
    load_vocoder,
    load_model,
    infer_process,
    save_spectrogram,
)

# —————————————————————————————
# 0. CƠ SỞ DỮ LIỆU TRONG BỘ NHỚ
# —————————————————————————————
# Admin mặc định
user_db = {
    "lytran": {
        "password": "16051998",
        "approved": True,
        "is_admin": True
    }
}
# Các tài khoản chờ duyệt
pending_db = {}  # key=username, value=password

# —————————————————————————————
# 1. ĐĂNG NHẬP HUGGINGFACE HUB (nếu có TOKEN)
# —————————————————————————————
hf_token = os.getenv("HUGGINGFACEHUB_API_TOKEN")
if hf_token:
    login(token=hf_token)

# —————————————————————————————
# 2. Hàm tiền xử lý văn bản
# —————————————————————————————
def post_process(text: str) -> str:
    text = f" {text} "
    for old, new in [(" . . ", " . "), (" .. ", " . "),
                     (" , , ", " , "), (" ,, ", " , ")]:
        text = text.replace(old, new)
    text = text.replace('"', "")
    return " ".join(text.split())

# —————————————————————————————
# 3. Load model và vocoder
# —————————————————————————————
vocoder = load_vocoder()
model = load_model(
    DiT,
    dict(dim=1024, depth=22, heads=16, ff_mult=2, text_dim=512, conv_layers=4),
    ckpt_path=str(cached_path("hf://LTTEAM/100h-VN/model_100h.pt")),
    vocab_file=str(cached_path("hf://LTTEAM/100h-VN/vocab.txt")),
)

# —————————————————————————————
# 4. Hàm chuyển TTS
# —————————————————————————————
def infer_tts(ref_audio_path: str, gen_text: str, speed: float = 1.0):
    if not ref_audio_path:
        raise gr.Error("Vui lòng tải lên hoặc chọn tệp âm thanh mẫu.")
    if not gen_text.strip():
        raise gr.Error("Vui lòng nhập nội dung văn bản cần chuyển thành giọng nói.")
    if len(gen_text.split()) > 1000:
        raise gr.Error("Vui lòng nhập văn bản có ít hơn 1000 từ.")
    try:
        ref_audio, ref_text = preprocess_ref_audio_text(ref_audio_path, "")
        wave, sr, spec = infer_process(
            ref_audio,
            ref_text.lower(),
            post_process(TTSnorm(gen_text)).lower(),
            model, vocoder,
            speed=speed
        )
        # Lưu tạm spectrogram
        with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp:
            save_spectrogram(spec, tmp.name)
            spec_path = tmp.name
        return (sr, wave), spec_path
    except Exception as e:
        raise gr.Error(f"Lỗi khi chuyển văn bản sang giọng nói: {e}")

# —————————————————————————————
# 5. Hàm đăng ký (đưa vào pending_db)
# —————————————————————————————
def register_user(username: str, password: str):
    username = username.strip()
    if not username or not password:
        return "Vui lòng nhập đủ tên và mật khẩu."
    if username in user_db:
        return "Tên này đã tồn tại."
    if username in pending_db:
        return "Tên này đang chờ admin duyệt."
    pending_db[username] = password
    return "Đăng ký thành công! Vui lòng chờ admin duyệt."

# —————————————————————————————
# 6. Hàm đăng nhập (kiểm tra pending vs user_db)
# —————————————————————————————
def login_user(username: str, password: str):
    username = username.strip()
    if username in pending_db:
        return False, "Tài khoản đang chờ admin duyệt."
    u = user_db.get(username)
    if not u or u["password"] != password:
        return False, "Tên đăng nhập hoặc mật khẩu không đúng."
    if not u["approved"]:
        return False, "Tài khoản chưa được duyệt."
    return True, ""

# —————————————————————————————
# 7. Hàm duyệt/từ chối (chỉ admin)
# —————————————————————————————
def approve_user(username: str):
    if username not in pending_db:
        return gr.update(choices=list(pending_db.keys()), value=None), "Không tìm thấy user."
    # chuyển từ pending_db sang user_db
    user_db[username] = {
        "password": pending_db.pop(username),
        "approved": True,
        "is_admin": False
    }
    return gr.update(choices=list(pending_db.keys()), value=None), f"Đã duyệt {username}."

def reject_user(username: str):
    if username not in pending_db:
        return gr.update(choices=list(pending_db.keys()), value=None), "Không tìm thấy user."
    pending_db.pop(username)
    return gr.update(choices=list(pending_db.keys()), value=None), f"Đã từ chối {username}."

# —————————————————————————————
# 8. Xây giao diện Gradio
# —————————————————————————————
with gr.Blocks(theme=gr.themes.Soft()) as demo:
    # === Login + Register column ===
    with gr.Column(visible=True) as login_col:
        gr.Markdown("## Đăng nhập / Đăng ký")
        # -- form login --
        username_input = gr.Textbox(label="Tên đăng nhập", placeholder="Nhập tên đăng nhập...")
        password_input = gr.Textbox(label="Mật khẩu", type="password", placeholder="Nhập mật khẩu...")
        login_btn = gr.Button("Đăng nhập", variant="primary")
        login_msg = gr.Textbox(interactive=False, label="", placeholder="")

        gr.Markdown("---")
        # -- form register --
        reg_username = gr.Textbox(label="Tên đăng ký", placeholder="Chọn tên đăng nhập mới...")
        reg_password = gr.Textbox(label="Mật khẩu", type="password", placeholder="Chọn mật khẩu...")
        register_btn = gr.Button("Đăng ký tài khoản")
        register_msg = gr.Textbox(interactive=False, label="")

        gr.Markdown("---")
        # -- profile card giới thiệu --
        gr.HTML("""
<div style="
    display: flex; align-items: center;
    padding: 20px; border:1px solid #ddd; border-radius:10px;
    background:#fff;
">
  <img src="https://congnghe360.com/wp-content/uploads/2025/05/c11f123eeec30f9d56d2.jpg"
       style="width:80px;height:80px;border-radius:50%;margin-right:20px;"/>
  <div>
    <h4 style="margin:0;">Lý Trần (Admin)</h4>
    <p style="margin:4px 0 0 0;color:#666;">Quản trị viên LT TEAM</p>
  </div>
</div>
        """)

    # === Main UI (sau login) ===
    with gr.Column(visible=False) as main_col:
        # 1) Giao diện TTS
        with gr.Column():
            gr.Markdown("""
# LTTEAM – Chuyển văn bản sang giọng nói tiếng Việt  
Nhập văn bản, chọn mẫu giọng, nhấn “Chuyển văn bản thành giọng nói”.
            """)
            with gr.Row():
                ref_audio = gr.Audio(
                    value="assets/anhkhoiVbee.wav",
                    label="Mẫu giọng (có sẵn)",
                    type="filepath"
                )
                gen_text = gr.Textbox(
                    label="Nhập văn bản",
                    placeholder="Nhập nội dung cần chuyển...",
                    lines=3
                )
            speed = gr.Slider(0.3, 2.0, value=1.0, step=0.1, label="⚡ Tốc độ")
            synth_btn = gr.Button("Chuyển TTS", variant="primary")
            with gr.Row():
                output_audio = gr.Audio(label="Kết quả", type="numpy")
                output_spec  = gr.Image(label="Spectrogram")
            gr.Textbox(
                value="""⚠️ Giới hạn:
1. Chưa xử lý tốt số, ngày tháng…
2. Có thể gián đoạn.
3. Transcription whisper có sai sót.
4. Huấn luyện 150h, chưa hoàn hảo.
5. Văn bản quá dài giảm chất lượng.""",
                label="Lưu ý", lines=6, interactive=False
            )
            synth_btn.click(infer_tts, [ref_audio, gen_text, speed], [output_audio, output_spec])

        # 2) Admin panel: chỉ visible nếu admin
        with gr.Column(visible=False) as admin_col:
            gr.Markdown("## Quản lý người dùng (Admin)")
            pending_dropdown = gr.Dropdown(choices=[], label="Chọn user chờ duyệt")
            btn_approve = gr.Button("Duyệt", variant="primary")
            btn_reject  = gr.Button("Từ chối", variant="secondary")
            admin_msg = gr.Textbox(interactive=False, label="Thông báo")

    # — Callback đăng ký —
    register_btn.click(
        fn=register_user,
        inputs=[reg_username, reg_password],
        outputs=[register_msg]
    )

    # — Callback đăng nhập —
    def on_login(username, password):
        ok, msg = login_user(username, password)
        if not ok:
            return None, None, msg, None, None  # chỉ cập nhật login_msg
        # login thành công
        is_admin = user_db[username]["is_admin"]
        return (
            gr.update(visible=False),              # ẩn login_col
            gr.update(visible=True),               # hiện main_col
            "",                                    # xoá login_msg
            gr.update(visible=is_admin),           # show/hide admin_col
            gr.update(choices=list(pending_db.keys()), value=None)  # cập nhật dropdown
        )

    login_btn.click(
        fn=on_login,
        inputs=[username_input, password_input],
        outputs=[login_col, main_col, login_msg, admin_col, pending_dropdown]
    )

    # — Callback admin duyệt —
    btn_approve.click(
        fn=approve_user,
        inputs=[pending_dropdown],
        outputs=[pending_dropdown, admin_msg]
    )
    btn_reject.click(
        fn=reject_user,
        inputs=[pending_dropdown],
        outputs=[pending_dropdown, admin_msg]
    )

demo.queue().launch(share=True)