import time

import streamlit as st
import numpy as np
import torch
from espnet2.bin.tts_inference import Text2Speech
from scipy.io.wavfile import write
from PIL import Image


fs, lang = 44100, "Japanese"
model= "./100epoch.pth"
x = "これはテストメッセージです"

text2speech = Text2Speech.from_pretrained(
    model_file=model,
    device="cpu",
    speed_control_alpha=1.0,
    noise_scale=0.333,
    noise_scale_dur=0.333,
)
pause = np.zeros(30000, dtype=np.float32)

st.title("おしゃべりAI晋さんメーカー")
image = Image.open('abe.jpg')
st.image(image)
text = st.text_area(label='ここにテキストを入力 (Input Text)↓', height=100, max_chars=128)


if st.button("生成（Generate）"):
    with torch.no_grad():
        wav = text2speech(text)["wav"]

    wav_list = []
    wav_list.append(np.concatenate([wav.view(-1).cpu().numpy(), pause]))
    final_wav = np.concatenate(wav_list)
    st.audio(final_wav, sample_rate=fs)