update app.py

app.py

@@ -111,12 +111,18 @@ if __name__ == '__main__':
     parser.add_argument("--share", action="store_true", default=False, help="share gradio app")
     args = parser.parse_args()
     device = torch.device(args.device)
-    
+    categories = ["Blue Archive", "Lycoris Recoil"]
+    others = {
+        "Princess Connect! Re:Dive": "https://huggingface.co/spaces/sayashi/vits-models",
+        "Genshin Impact": "https://huggingface.co/spaces/sayashi/vits-models",
+        "Honkai Impact 3rd": "https://huggingface.co/spaces/sayashi/vits-models",
+        "Overwatch 2": "https://huggingface.co/spaces/sayashi/vits-models",
+    }
     models = []
     with open("pretrained_models/info.json", "r", encoding="utf-8") as f:
         models_info = json.load(f)
     for i, info in models_info.items():
-        if not info['enable']:
+        if info['title'].split("-")[0] not in categories or not info['enable']:
             continue
         sid = info['sid']
         name_en = info['name_en']
@@ -138,127 +144,144 @@ if __name__ == '__main__':
         gr.Markdown(
             "# <center> vits-models\n"
             "## <center> Please do not generate content that could infringe upon the rights or cause harm to individuals or organizations.\n"
-            "## <center> ·请不要生成会对个人以及组织造成侵害的内容\n"
+            "## <center> 请不要生成会对个人以及组织造成侵害的内容\n"
             "![visitor badge](https://visitor-badge.glitch.me/badge?page_id=sayashi.vits-models)\n\n"
-            "[Open In Colab]"
-            "(https://colab.research.google.com/drive/10QOk9NPgoKZUXkIhhuVaZ7SYra1MPMKH?usp=share_link)"
-            " without queue and length limitation.(无需等待队列，并且没有长度限制)\n\n"
-            "[Finetune your own model](https://github.com/SayaSS/vits-finetuning)"
+            "[![image](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/drive/10QOk9NPgoKZUXkIhhuVaZ7SYra1MPMKH?usp=share_link)\n\n"
+            "[![Duplicate this Space](https://huggingface.co/datasets/huggingface/badges/raw/main/duplicate-this-space-sm-dark.svg)](https://huggingface.co/spaces/sayashi/vits-models?duplicate=true)"
+            "[![Finetune your own model](https://badgen.net/badge/icon/github?icon=github&label=Finetune%20your%20own%20model)](https://github.com/SayaSS/vits-finetuning)"
         )
 
         with gr.Tabs():
-            with gr.TabItem("EN"):
-                for (sid, name_en, name_zh, title, cover, example, language, net_g_ms, tts_fn, to_symbol_fn) in models:
-                    with gr.TabItem(name_en):
-                        with gr.Row():
-                            gr.Markdown(
-                                '<div align="center">'
-                                f'<a><strong>{title}</strong></a>'
-                                f'<img style="width:auto;height:300px;" src="file/{cover}">' if cover else ""
-                                '</div>'
-                            )
-                        with gr.Row():
-                            with gr.Column():
-                                input_text = gr.Textbox(label="Text (100 words limitation)" if limitation else "Text", lines=5, value=example, elem_id=f"input-text-en-{name_en.replace(' ','')}")
-                                lang = gr.Dropdown(label="Language", choices=["Chinese", "Japanese", "Mix（wrap the Chinese text with [ZH][ZH], wrap the Japanese text with [JA][JA]）"],
-                                            type="index", value=language)
-                                with gr.Accordion(label="Advanced Options", open=False):
-                                    symbol_input = gr.Checkbox(value=False, label="Symbol input")
-                                    symbol_list = gr.Dataset(label="Symbol list", components=[input_text],
-                                                             samples=[[x] for x in hps_ms.symbols])
-                                    symbol_list_json = gr.Json(value=hps_ms.symbols, visible=False)
-                                btn = gr.Button(value="Generate", variant="primary")
+            for category in categories:
+                with gr.TabItem(category):
+                    with gr.TabItem("EN"):
+                        for (sid, name_en, name_zh, title, cover, example, language, net_g_ms, tts_fn, to_symbol_fn) in models:
+                            if title.split("-")[0] != category:
+                                continue
+                            with gr.TabItem(name_en):
                                 with gr.Row():
-                                    ns = gr.Slider(label="noise_scale", minimum=0.1, maximum=1.0, step=0.1, value=0.6, interactive=True)
-                                    nsw = gr.Slider(label="noise_scale_w", minimum=0.1, maximum=1.0, step=0.1, value=0.668, interactive=True)
-                                    ls = gr.Slider(label="length_scale", minimum=0.1, maximum=2.0, step=0.1, value=1.2 if language=="Chinese" else 1, interactive=True)
-                            with gr.Column():
-                                o1 = gr.Textbox(label="Output Message")
-                                o2 = gr.Audio(label="Output Audio", elem_id=f"tts-audio-en-{name_en.replace(' ','')}")
-                                download = gr.Button("Download Audio")
-                            btn.click(tts_fn, inputs=[input_text, lang,  ns, nsw, ls, symbol_input], outputs=[o1, o2], api_name=f"tts-{name_en}")
-                            download.click(None, [], [], _js=download_audio_js.format(audio_id=f"en-{name_en.replace(' ', '')}"))
-                            lang.change(change_lang, inputs=[lang], outputs=[ns, nsw, ls])
-                            symbol_input.change(
-                                to_symbol_fn,
-                                [symbol_input, input_text, lang],
-                                [input_text]
-                            )
-                            symbol_list.click(None, [symbol_list, symbol_list_json], [input_text],
-                                              _js=f"""
-                            (i,symbols) => {{
-                                let root = document.querySelector("body > gradio-app");
-                                if (root.shadowRoot != null)
-                                    root = root.shadowRoot;
-                                let text_input = root.querySelector("#input-text-en-{name_en.replace(' ', '')}").querySelector("textarea");
-                                let startPos = text_input.selectionStart;
-                                let endPos = text_input.selectionEnd;
-                                let oldTxt = text_input.value;
-                                let result = oldTxt.substring(0, startPos) + symbols[i] + oldTxt.substring(endPos);
-                                text_input.value = result;
-                                let x = window.scrollX, y = window.scrollY;
-                                text_input.focus();
-                                text_input.selectionStart = startPos + symbols[i].length;
-                                text_input.selectionEnd = startPos + symbols[i].length;
-                                text_input.blur();
-                                window.scrollTo(x, y);
-                                return text_input.value;
-                            }}""")
-            with gr.TabItem("中文"):
-                for (sid, name_en, name_zh, title, cover, example, language,  net_g_ms, tts_fn, to_symbol_fn) in models:
-                    with gr.TabItem(name_zh):
-                        with gr.Row():
-                            gr.Markdown(
-                                '<div align="center">'
-                                f'<a><strong>{title}</strong></a>'
-                                f'<img style="width:auto;height:300px;" src="file/{cover}">' if cover else ""
-                                '</div>'
-                            )
-                        with gr.Row():
-                            with gr.Column():
-                                input_text = gr.Textbox(label="文本 (100字上限)" if limitation else "文本", lines=5, value=example, elem_id=f"input-text-zh-{name_zh}")
-                                lang = gr.Dropdown(label="语言", choices=["中文", "日语", "中日混合（中文用[ZH][ZH]包裹起来，日文用[JA][JA]包裹起来）"],
-                                            type="index", value="中文"if language == "Chinese" else "日语")
-                                with gr.Accordion(label="高级选项", open=False):
-                                    symbol_input = gr.Checkbox(value=False, label="符号输入")
-                                    symbol_list = gr.Dataset(label="符号列表", components=[input_text],
-                                                             samples=[[x] for x in hps_ms.symbols])
-                                    symbol_list_json = gr.Json(value=hps_ms.symbols, visible=False)
-                                btn = gr.Button(value="生成", variant="primary")
+                                    gr.Markdown(
+                                        '<div align="center">'
+                                        f'<a><strong>{title}</strong></a>'
+                                        f'<img style="width:auto;height:300px;" src="file/{cover}">' if cover else ""
+                                        '</div>'
+                                    )
                                 with gr.Row():
-                                    ns = gr.Slider(label="控制感情变化程度", minimum=0.1, maximum=1.0, step=0.1, value=0.6, interactive=True)
-                                    nsw = gr.Slider(label="控制音素发音长度", minimum=0.1, maximum=1.0, step=0.1, value=0.668, interactive=True)
-                                    ls = gr.Slider(label="控制整体语速", minimum=0.1, maximum=2.0, step=0.1, value=1.2 if language=="Chinese" else 1, interactive=True)
-                            with gr.Column():
-                                o1 = gr.Textbox(label="输出信息")
-                                o2 = gr.Audio(label="输出音频", elem_id=f"tts-audio-zh-{name_zh}")
-                                download = gr.Button("下载音频")
-                            btn.click(tts_fn, inputs=[input_text, lang,  ns, nsw, ls, symbol_input], outputs=[o1, o2])
-                            download.click(None, [], [], _js=download_audio_js.format(audio_id=f"zh-{name_zh}"))
-                            lang.change(change_lang, inputs=[lang], outputs=[ns, nsw, ls])
-                            symbol_input.change(
-                                to_symbol_fn,
-                                [symbol_input, input_text, lang],
-                                [input_text]
-                            )
-                            symbol_list.click(None, [symbol_list, symbol_list_json], [input_text],
-                                              _js=f"""
-                            (i,symbols) => {{
-                                let root = document.querySelector("body > gradio-app");
-                                if (root.shadowRoot != null)
-                                    root = root.shadowRoot;
-                                let text_input = root.querySelector("#input-text-zh-{name_zh}").querySelector("textarea");
-                                let startPos = text_input.selectionStart;
-                                let endPos = text_input.selectionEnd;
-                                let oldTxt = text_input.value;
-                                let result = oldTxt.substring(0, startPos) + symbols[i] + oldTxt.substring(endPos);
-                                text_input.value = result;
-                                let x = window.scrollX, y = window.scrollY;
-                                text_input.focus();
-                                text_input.selectionStart = startPos + symbols[i].length;
-                                text_input.selectionEnd = startPos + symbols[i].length;
-                                text_input.blur();
-                                window.scrollTo(x, y);
-                                return text_input.value;
-                            }}""")
+                                    with gr.Column():
+                                        input_text = gr.Textbox(label="Text (100 words limitation)" if limitation else "Text", lines=5, value=example, elem_id=f"input-text-en-{name_en.replace(' ','')}")
+                                        lang = gr.Dropdown(label="Language", choices=["Chinese", "Japanese", "Mix（wrap the Chinese text with [ZH][ZH], wrap the Japanese text with [JA][JA]）"],
+                                                    type="index", value=language)
+                                        with gr.Accordion(label="Advanced Options", open=False):
+                                            symbol_input = gr.Checkbox(value=False, label="Symbol input")
+                                            symbol_list = gr.Dataset(label="Symbol list", components=[input_text],
+                                                                     samples=[[x] for x in hps_ms.symbols])
+                                            symbol_list_json = gr.Json(value=hps_ms.symbols, visible=False)
+                                        btn = gr.Button(value="Generate", variant="primary")
+                                        with gr.Row():
+                                            ns = gr.Slider(label="noise_scale", minimum=0.1, maximum=1.0, step=0.1, value=0.6, interactive=True)
+                                            nsw = gr.Slider(label="noise_scale_w", minimum=0.1, maximum=1.0, step=0.1, value=0.668, interactive=True)
+                                            ls = gr.Slider(label="length_scale", minimum=0.1, maximum=2.0, step=0.1, value=1.2 if language=="Chinese" else 1, interactive=True)
+                                    with gr.Column():
+                                        o1 = gr.Textbox(label="Output Message")
+                                        o2 = gr.Audio(label="Output Audio", elem_id=f"tts-audio-en-{name_en.replace(' ','')}")
+                                        download = gr.Button("Download Audio")
+                                    btn.click(tts_fn, inputs=[input_text, lang,  ns, nsw, ls, symbol_input], outputs=[o1, o2], api_name=f"tts-{name_en}")
+                                    download.click(None, [], [], _js=download_audio_js.format(audio_id=f"en-{name_en.replace(' ', '')}"))
+                                    lang.change(change_lang, inputs=[lang], outputs=[ns, nsw, ls])
+                                    symbol_input.change(
+                                        to_symbol_fn,
+                                        [symbol_input, input_text, lang],
+                                        [input_text]
+                                    )
+                                    symbol_list.click(None, [symbol_list, symbol_list_json], [input_text],
+                                                      _js=f"""
+                                    (i,symbols) => {{
+                                        let root = document.querySelector("body > gradio-app");
+                                        if (root.shadowRoot != null)
+                                            root = root.shadowRoot;
+                                        let text_input = root.querySelector("#input-text-en-{name_en.replace(' ', '')}").querySelector("textarea");
+                                        let startPos = text_input.selectionStart;
+                                        let endPos = text_input.selectionEnd;
+                                        let oldTxt = text_input.value;
+                                        let result = oldTxt.substring(0, startPos) + symbols[i] + oldTxt.substring(endPos);
+                                        text_input.value = result;
+                                        let x = window.scrollX, y = window.scrollY;
+                                        text_input.focus();
+                                        text_input.selectionStart = startPos + symbols[i].length;
+                                        text_input.selectionEnd = startPos + symbols[i].length;
+                                        text_input.blur();
+                                        window.scrollTo(x, y);
+                                        return text_input.value;
+                                    }}""")
+                    with gr.TabItem("中文"):
+                        for (sid, name_en, name_zh, title, cover, example, language,  net_g_ms, tts_fn, to_symbol_fn) in models:
+                            if title.split("-")[0] != category:
+                                continue
+                            with gr.TabItem(name_zh):
+                                with gr.Row():
+                                    gr.Markdown(
+                                        '<div align="center">'
+                                        f'<a><strong>{title}</strong></a>'
+                                        f'<img style="width:auto;height:300px;" src="file/{cover}">' if cover else ""
+                                        '</div>'
+                                    )
+                                with gr.Row():
+                                    with gr.Column():
+                                        input_text = gr.Textbox(label="文本 (100字上限)" if limitation else "文本", lines=5, value=example, elem_id=f"input-text-zh-{name_zh}")
+                                        lang = gr.Dropdown(label="语言", choices=["中文", "日语", "中日混合（中文用[ZH][ZH]包裹起来，日文用[JA][JA]包裹起来）"],
+                                                    type="index", value="中文"if language == "Chinese" else "日语")
+                                        with gr.Accordion(label="高级选项", open=False):
+                                            symbol_input = gr.Checkbox(value=False, label="符号输入")
+                                            symbol_list = gr.Dataset(label="符号列表", components=[input_text],
+                                                                     samples=[[x] for x in hps_ms.symbols])
+                                            symbol_list_json = gr.Json(value=hps_ms.symbols, visible=False)
+                                        btn = gr.Button(value="生成", variant="primary")
+                                        with gr.Row():
+                                            ns = gr.Slider(label="控制感情变化程度", minimum=0.1, maximum=1.0, step=0.1, value=0.6, interactive=True)
+                                            nsw = gr.Slider(label="控制音素发音长度", minimum=0.1, maximum=1.0, step=0.1, value=0.668, interactive=True)
+                                            ls = gr.Slider(label="控制整体语速", minimum=0.1, maximum=2.0, step=0.1, value=1.2 if language=="Chinese" else 1, interactive=True)
+                                    with gr.Column():
+                                        o1 = gr.Textbox(label="输出信息")
+                                        o2 = gr.Audio(label="输出音频", elem_id=f"tts-audio-zh-{name_zh}")
+                                        download = gr.Button("下载音频")
+                                    btn.click(tts_fn, inputs=[input_text, lang,  ns, nsw, ls, symbol_input], outputs=[o1, o2])
+                                    download.click(None, [], [], _js=download_audio_js.format(audio_id=f"zh-{name_zh}"))
+                                    lang.change(change_lang, inputs=[lang], outputs=[ns, nsw, ls])
+                                    symbol_input.change(
+                                        to_symbol_fn,
+                                        [symbol_input, input_text, lang],
+                                        [input_text]
+                                    )
+                                    symbol_list.click(None, [symbol_list, symbol_list_json], [input_text],
+                                                      _js=f"""
+                                    (i,symbols) => {{
+                                        let root = document.querySelector("body > gradio-app");
+                                        if (root.shadowRoot != null)
+                                            root = root.shadowRoot;
+                                        let text_input = root.querySelector("#input-text-zh-{name_zh}").querySelector("textarea");
+                                        let startPos = text_input.selectionStart;
+                                        let endPos = text_input.selectionEnd;
+                                        let oldTxt = text_input.value;
+                                        let result = oldTxt.substring(0, startPos) + symbols[i] + oldTxt.substring(endPos);
+                                        text_input.value = result;
+                                        let x = window.scrollX, y = window.scrollY;
+                                        text_input.focus();
+                                        text_input.selectionStart = startPos + symbols[i].length;
+                                        text_input.selectionEnd = startPos + symbols[i].length;
+                                        text_input.blur();
+                                        window.scrollTo(x, y);
+                                        return text_input.value;
+                                    }}""")
+            for category, link in others.items():
+                with gr.TabItem(category):
+                    gr.Markdown(
+                        f'''
+                        <center>
+                          <h2>Click to Go</h2>
+                          <a href="{link}">
+                            <img src="https://huggingface.co/datasets/huggingface/badges/raw/main/open-in-hf-spaces-xl-dark.svg"
+                          </a>
+                        </center>
+                        '''
+                    )
     app.queue(concurrency_count=1, api_open=args.api).launch(share=args.share)

pretrained_models/info.json

@@ -22,7 +22,7 @@
 		"type": "single"
     },
 	"misora": {
-		"enable": false,
+		"enable": true,
 		"name_en": "Misora",
 		"name_zh": "美空",
         "title": "Princess Connect! Re:Dive-ミソラ",
@@ -33,7 +33,7 @@
 		"type": "single"
     },
 	"kyoka": {
-		"enable": false,
+		"enable": true,
 		"name_en": "Kyoka",
 		"name_zh": "镜华",
         "title": "Princess Connect! Re:Dive-キョウカ",
@@ -44,7 +44,7 @@
 		"type": "multi"
     },
 	"hiyori": {
-		"enable": false,
+		"enable": true,
 		"name_en": "Hiyori",
 		"name_zh": "日和莉",
         "title": "Princess Connect! Re:Dive-ヒヨリ",
@@ -66,7 +66,7 @@
 		"type": "single"
     },
 	"hatsune": {
-		"enable": false,
+		"enable": true,
 		"name_en": "Hatsune",
 		"name_zh": "柏崎初音",
         "title": "Princess Connect! Re:Dive-柏崎初音",
@@ -77,7 +77,7 @@
 		"type": "multi"
     },
 	"eriko": {
-		"enable": false,
+		"enable": true,
 		"name_en": "Eriko",
 		"name_zh": "惠理子",
         "title": "Princess Connect! Re:Dive-倉石恵理子",
@@ -143,7 +143,7 @@
 		"type": "single"
     },
 	"ayaka-jp": {
-		"enable": false,
+		"enable": true,
 		"name_en": "ayaka-jp",
 		"name_zh": "神里绫华-日语",
         "title": "Genshin Impact-神里綾華",
@@ -154,7 +154,7 @@
 		"type": "multi"
     },
 	"nahida-jp": {
-		"enable": false,
+		"enable": true,
 		"name_en": "nahida-jp",
 		"name_zh": "纳西妲-日语",
         "title": "Genshin Impact-ナヒーダ",
@@ -198,7 +198,7 @@
 		"type": "multi"
     },
 	"karin": {
-		"enable": false,
+		"enable": true,
 		"name_en": "Kakudate Karin",
 		"name_zh": "角楯花凛",
         "title": "Blue Archive-角楯カリン",
@@ -209,7 +209,7 @@
 		"type": "multi"
     },
 	"asuna": {
-		"enable": false,
+		"enable": true,
 		"name_en": "Ichinose Asuna",
 		"name_zh": "一之濑明日奈",
         "title": "Blue Archive-一之瀬アスナ",
@@ -253,7 +253,7 @@
 		"type": "multi"
     },
 	"hoshino": {
-		"enable": false,
+		"enable": true,
 		"name_en": "Takanasi Hosino",
 		"name_zh": "小鸟游星野",
         "title": "Blue Archive-小鳥遊 ホシノ",
@@ -308,7 +308,7 @@
 		"type": "multi"
     },
 	"doom": {
-		"enable": false,
+		"enable": true,
 		"name_en": "Doomfist",
 		"name_zh": "末日铁拳",
         "title": "Overwatch 2-Doomfist",
@@ -319,7 +319,7 @@
 		"type": "multi"
     },
 	"echo": {
-		"enable": false,
+		"enable": true,
 		"name_en": "Echo",
 		"name_zh": "回声",
         "title": "Overwatch 2-Echo",
@@ -330,7 +330,7 @@
 		"type": "multi"
     },
 	"zenyatta": {
-		"enable": false,
+		"enable": true,
 		"name_en": "Zenyatta",
 		"name_zh": "禅雅塔",
         "title": "Overwatch 2-Zenyatta",
@@ -341,10 +341,10 @@
 		"type": "multi"
     },
 	"abyssinvoker": {
-		"enable": false,
+		"enable": true,
 		"name_en": "Abyss Invoker",
 		"name_zh": "深渊使徒",
-        "title": "原神-深渊使徒",
+        "title": "Genshin Impact-深渊使徒",
         "cover": "cover.png",
 		"sid": 94,
 		"example": "今天晚上吃啥好呢",
@@ -352,10 +352,10 @@
 		"type": "multi"
     },
 	"keqing": {
-		"enable": false,
+		"enable": true,
 		"name_en": "Keqing",
 		"name_zh": "刻晴",
-        "title": "原神-刻晴",
+        "title": "Genshin Impact-刻晴",
         "cover": "cover.png",
 		"sid": 115,
 		"example": "今天晚上吃啥好呢",
@@ -363,10 +363,10 @@
 		"type": "multi"
     },
 	"eula": {
-		"enable": false,
+		"enable": true,
 		"name_en": "Eula",
 		"name_zh": "优菈",
-        "title": "原神-优菈",
+        "title": "Genshin Impact-优菈",
         "cover": "cover.png",
 		"sid": 124,
 		"example": "今天晚上吃啥好呢",
@@ -374,10 +374,10 @@
 		"type": "multi"
     },
 	"bronya": {
-		"enable": false,
+		"enable": true,
 		"name_en": "Herrscher of Reason",
 		"name_zh": "理之律者",
-        "title": "崩坏3-理之律者",
+        "title": "Honkai Impact 3rd-理之律者",
         "cover": "cover.png",
 		"sid": 193,
 		"example": "今天晚上吃啥好呢",
@@ -385,10 +385,10 @@
 		"type": "multi"
     },
 	"theresa": {
-		"enable": false,
+		"enable": true,
 		"name_en": "Theresa",
 		"name_zh": "德丽莎",
-        "title": "崩坏3-德丽莎",
+        "title": "Honkai Impact 3rd-德丽莎",
         "cover": "cover.png",
 		"sid": 193,
 		"example": "今天晚上吃啥好呢",