app.py

import gradio as gr
import os
from constants import VOICE_METHODS, BARK_VOICES, EDGE_VOICES
import platform
from models.model import *
from tts.conversion import COQUI_LANGUAGES
import pytube
import os
import traceback
from pydub import AudioSegment
# from audio_enhance.functions import audio_enhance

import ast
import argparse
import glob
import pickle

import numpy as np
import pandas as pd
block_css = """
#notice_markdown {
    font-size: 104%
}
#notice_markdown th {
    display: none;
}
#notice_markdown td {
    padding-top: 6px;
    padding-bottom: 6px;
}
#leaderboard_markdown {
    font-size: 104%
}
#leaderboard_markdown td {
    padding-top: 6px;
    padding-bottom: 6px;
}
#leaderboard_dataframe td {
    line-height: 0.1em;
}
footer {
    display:none !important
}
.image-container {
    display: flex;
    align-items: center;
    padding: 1px;
}
.image-container img {
    margin: 0 30px;
    height: 20px;
    max-height: 100%;
    width: auto;
    max-width: 20%;
}
"""
def model_hyperlink(model_name, link):
    return f'<a target="_blank" href="{link}" style="color: var(--link-text-color); text-decoration: underline;text-decoration-style: dotted;">{model_name}</a>'
def load_leaderboard_table_csv(filename, add_hyperlink=True):
    lines = open(filename).readlines()
    heads = [v.strip() for v in lines[0].split(",")]
    rows = []
    for i in range(1, len(lines)):
        row = [v.strip() for v in lines[i].split(",")]
        for j in range(len(heads)):
            item = {}
            for h, v in zip(heads, row):
                if h != "AI歌手名/AI Singer" and h != "歌手代表作或介绍链接" and h != "模型zip链接/Link to Model Zip File" and h != "模型贡献者/Model Contributor":
                    item[h] = int(v)
                else:
                    item[h] = v
            if add_hyperlink:
                item["AI歌手名/AI Singer"] = model_hyperlink(item["AI歌手名/AI Singer"], item["歌手代表作或介绍链接"])
        rows.append(item)
    return rows

def get_arena_table(model_table_df):
    # sort by rating
    model_table_df = model_table_df.sort_values(by=["训练素材时长/Duration of Training Dataset(min)"], ascending=False)
    values = []
    for i in range(len(model_table_df)):
        row = []
        model_key = model_table_df.index[i]
        model_name = model_table_df["AI歌手名/AI Singer"].values[model_key]
        # rank
        row.append(i + 1)
        # model display name
        row.append(model_name)

        row.append(
            model_table_df["模型zip链接/Link to Model Zip File"].values[model_key]
        )

        row.append(
            model_table_df["训练素材时长/Duration of Training Dataset(min)"].values[model_key]
        )

        row.append(
            model_table_df["训练epoch数/Epoch"].values[model_key]
        )

        row.append(
            model_table_df["模型贡献者/Model Contributor"].values[model_key]
        )

        values.append(row)
    return values



title_markdown = ("""
<h2 align="center"> 🌊💕🎶 滔滔AI，AI歌手模型开源社区 </h2>
<h3 align="center"> 🌟 完全开源、完全免费、共建共享！全网AI歌手任您选择！ </h3>
""")

pic_markdown = ("""
<h3 align="center">  </h3>
<h1 align="center"><a href="https://www.talktalkai.com/"><img src="https://download.openxlab.org.cn/models/Kevin676/rvc-models/weight/talktalkai-cover.png", alt="talktalkai-cover" border="0" style="margin: 0 auto; height: 300px;" /></a> </h1>
""")


data = load_leaderboard_table_csv("./OCRBench.csv")
model_table_df = pd.DataFrame(data)
text_size = gr.themes.sizes.text_lg


with gr.Blocks(theme=gr.themes.Base(text_size=text_size), css=block_css) as app:
    
    gr.Markdown(title_markdown)

    with gr.Tab("✨模型搜索及上传"):
        gr.HTML("<h3>1. 搜索AI歌手模型</h3>")
        gr.Markdown("##### 点击[此链接](https://docs.google.com/spreadsheets/d/1owfUtQuLW9ReiIwg6U9UkkDmPOTkuNHf0OKQtWu1iaI/edit?gid=1227575351#gid=1227575351)，查看全网所有开源AI歌手模型，超9000个模型任您挑选 🥳")
        search_name = gr.Textbox(placeholder="孙燕姿", label="请填写模型名称进行搜索", show_label=True)
         # Salida
        with gr.Row():
            sarch_output = gr.Markdown(label="搜索结果")
            
        btn_search_model = gr.Button(value="开始搜索吧💖", variant="primary")
        btn_search_model.click(fn=search_model, inputs=[search_name], outputs=[sarch_output])
        
        gr.HTML("<h3>2. 上传AI歌手模型至社区</h3>")
        gr.HTML("<h4>上传完成后您立即可以搜索到您上传的模型</h4>")
        post_name = gr.Textbox(placeholder="滔滔歌姬", label="请填写模型名称", show_label=True)
        post_model_url = gr.Textbox(placeholder="https://huggingface.co/kevinwang676/RVC-models/resolve/main/talktalkgirl.zip", label="模型链接", info="1.推荐使用Hugging Face存放模型 2.复制Hugging Face模型链接后，需要将链接中的blob四个字母替换成resolve，使模型可以通过链接直接下载", show_label=True)
        post_creator = gr.Textbox(placeholder="滔滔AI", label="模型贡献者", info="可填写您的昵称或任何有趣的ID", show_label=True)
        post_version = gr.Dropdown(choices=["RVC v1", "RVC v2"], value="RVC v2", label="RVC模型版本", show_label=True)
        
         # Salida
        with gr.Row():
            post_output = gr.Markdown(label="模型上传状态")
            
        btn_post_model = gr.Button(value="开始上传吧💕", variant="primary")
        btn_post_model.click(fn=post_model, inputs=[post_name, post_model_url, post_version, post_creator], outputs=[post_output])

    with gr.Tab("🍻滔滔AI精选模型"):
        arena_table_vals = get_arena_table(model_table_df)
        md = """
        AI翻唱🎶：您可以在社区中复制您喜欢的AI歌手的“模型zip链接”，之后就可以在“🌟重磅首发 - AI歌手全明星💕”页面中通过粘贴zip链接来使用您喜欢的AI歌手模型啦！\n
        [手机端📱](https://g-app-center-40055665-0593-xqmmjg6.openxlab.space)查看 滔滔AI精选模型
        """
        gr.Markdown(md, elem_id="leaderboard_markdown")
        gr.Dataframe(
            headers=[
                "排序",
                "AI歌手名/AI Singer",
                "模型zip链接/Link to Model Zip File",
                "训练素材时长/Duration of Training Dataset(min)",
                "训练epoch数/Epoch",
                "模型贡献者/Model Contributor",
            ],
            datatype=[
                "str",
                "markdown",
                "str",
                "number",
                "number",
                "str",
            ],
            value=arena_table_vals,
            elem_id="arena_leaderboard_dataframe",
            height=800,
            column_widths=[50, 100, 205, 95, 95, 95],
            wrap=True,
        )

    gr.Markdown(pic_markdown)
    gr.Markdown("###### <center>注意❗：请不要生成会对个人以及组织造成侵害的内容，此程序仅供科研、学习及个人娱乐使用。请自觉合规使用此程序，程序开发者不负有任何责任。</center>")

    gr.HTML('''
        <div class="footer">
                    <p>🌊🏞️🎶 - 江水东流急，滔滔无尽声。 明·顾璘
                    </p>
        </div>
    ''')
   
    
    app.queue(max_size=40, api_open=False)
    app.launch(max_threads=400, show_error=True)