import gradio as gr

from inference import load_model, local_run, get_text


pth_path = "model/G_70000.pth"
config_json = "configs/config.json"
character_dict = {
    "十香": 1,
    "折纸": 2,
    "狂三": 3,
    "四糸乃": 4,
    "琴里": 5,
    "夕弦": 6,
    "耶俱矢": 7,
    "美九": 8,
    "凛祢": 9,
    "凛绪": 10,
    "鞠亚": 11,
    "鞠奈": 12,
    "真那": 13,
}

app = gr.Blocks()
with app:
    gr.HTML("""
<div
    style="width: 100%;padding-top:116px;background-image: url('https://huggingface.co/spaces/tumuyan/vits-miki/resolve/main/bg.webp');;background-size:cover">
    <div>
                <div>
                    <h4 class="h-sign" style="font-size: 12px;">
                        这是一个使用<a href="https://github.com/thesupersonic16/DALTools" target="_blank">thesupersonic16/DALTools</a>提供的解包音频作为数据集，
                        使用<a href="https://github.com/jaywalnut310/vits" target="_blank">VITS</a>技术训练的语音合成demo。
                    </h4>
                </div>
            </div>
</div>
    """)
    tmp = gr.Markdown("")
    with gr.Tabs():
        with gr.TabItem("Basic"):
            with gr.Row():
                choice_model = gr.Dropdown(
                    choices=character_dict.keys(), label="模型", value=character_dict.values(), visible=False)
            
        with gr.TabItem("Audios"):

            pass
    gr.HTML("""
<div style="text-align:center"> 
    仅供学习交流，不可用于商业或非法用途
    <br/>
    使用本项目模型直接或间接生成的音频，必须声明由AI技术或VITS技术合成
</div>
    """)
    app.launch()