from vietTTS.hifigan.mel2wave import mel2wave
from vietTTS.nat.text2mel import text2mel
from vietTTS.synthesizer import nat_normalize_text
import numpy as np
import gradio as gr
import os


def text_to_speech(text,stop_duration_text):
    print("starting")
    # prevent too long text
    if len(text) > 500:
        text = text[:500]
    stop_duration_float = float(stop_duration_text)
    text = nat_normalize_text(text)
    mel = text2mel(
        text,
        "lexicon.txt",
        stop_duration_float,
        "acoustic_latest_ckpt.pickle",
        "duration_latest_ckpt.pickle",
    )
    print("mel")
    print(mel)
    wave = mel2wave(mel, "config.json", "hk_hifi.pickle")
    print("wave")
    print(wave)
    return (wave * (2**15)).astype(np.int16)


def speak(text,stop_duration_text):
    y = text_to_speech(text,stop_duration_text)
    return 16_000, y


title = "SLT TTS"
description = "SLT Vietnamese Text to speech demo."

gr.Interface(
    fn=speak, 
    inputs=["text","text"],
    outputs="audio",
    title = title,
    examples = [
    "Thành phố muốn thí điểm thu thuế bất động sản thứ 2, tự quyết nhiều quyết định đầu tư để thu hút nguồn vốn tư nhân",
    "Thứ năm, 22/12/2022, 10:59 những mét hàng rào đầu tiên quanh công viên lớn nhất thủ đô được tháo dỡ, chuyển công viên sang hình thức mở, không thu vé.",
    "Cô dì chú bác nhà em không học hành vì vẫn làm bất động sản. Có phải làm ngành này thì chỉ cần biết sơ sơ, không cần học hành?... Đó là vấn đề được học sinh trường Trung học phổ thông  Gia Định, Thành phố Hồ Chí Minh đặt ra tại buổi khai mạc chương trình tư vấn tuyển sinh do Tạp chí Giáo dục Thành phố Hồ Chí Minh tổ chức sáng mồng 9 tháng 1.",
    "Trong chuyến công du nước ngoài thứ hai từ khi Nga phát động chiến dịch quân sự tại Ukraine gần 12 tháng trước, Tổng thống Volodymyr Zelensky dự kiến phát biểu tại Nghị viện châu Âu ngày 9/2",
    "Tuy nhiên, kế hoạch này đang đứng trước nguy cơ đổ vỡ sau khi các quan chức Nghị viện châu âu làm rò rỉ kế hoạch bí mật tiếp đón Tổng thống Ukraine. Hồi đầu tuần, thông tin về chuyến thăm xuất hiện trên nhiều tờ báo châu âu, khởi nguồn từ báo La Stampa của Italy."
    ],
    description=description,
    theme="default",
    allow_screenshot=False,
    allow_flagging="never",
).launch(debug=False)