Hyper-RVC/app.py at main · BF667-IDLE/Hyper-RVC · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
"""
Hyper-RVC WebUI – main entry point.

Replaces the original ``main.py``.  Launch this file to start the Gradio
web interface:

    python app.py
    python app.py --port 8080 --share
    python app.py --open
"""

import gradio as gr
import sys
import os

now_dir = os.getcwd()
sys.path.append(now_dir)

DEFAULT_PORT = 7755
MAX_PORT_ATTEMPTS = 10

from assets.i18n.i18n import I18nAuto

i18n = I18nAuto()

import assets.themes.loadThemes as loadThemes

from tabs.full_inference import full_inference_tab
from tabs.download_model import download_model_tab
from tabs.download_music import download_music_tab
from tabs.settings import select_themes_tab
from tabs.whisper_transcription import whisper_diarization_tab
from tabs.tts_inference import tts_inference_tab
from tabs.realtime_tab import realtime_tab

rvc_theme = loadThemes.load_json() or "ParityError/Interstellar"

with gr.Blocks(
    theme=rvc_theme, title="HyperRVC", css="footer{display:none !important}"
) as app:
    gr.Markdown("# HyperRVC WebUI")
    with gr.Tab(i18n("Inference")):
        with gr.Tab(i18n("Full Inference")):
            full_inference_tab()
        with gr.Tab(i18n("Realtime")):
            realtime_tab()
        with gr.Tab(i18n("TTS")):
            tts_inference_tab()
        with gr.Tab(i18n("Whisper")):
            whisper_diarization_tab()
    with gr.Tab(i18n("Download")):
        with gr.Tab(i18n("Download Model")):
            download_model_tab()
        with gr.Tab(i18n("Download Music")):
            download_music_tab()
    with gr.Tab(i18n("Settings")):
        select_themes_tab()


def launch(port):
    app.launch(
        favicon_path=os.path.join(now_dir, "assets", "logo.ico"),
        share="--share" in sys.argv,
        inbrowser="--open" in sys.argv,
        server_port=port,
    )


def get_port_from_args():
    if "--port" in sys.argv:
        port_index = sys.argv.index("--port") + 1
        if port_index < len(sys.argv):
            return int(sys.argv[port_index])
    return DEFAULT_PORT


if __name__ == "__main__":
    port = get_port_from_args()
    for _ in range(MAX_PORT_ATTEMPTS):
        try:
            launch(port)
            break
        except OSError:
            print(
                f"Failed to launch on port {port}, trying again on port {port - 1}..."
            )
            port -= 1
        except Exception as error:
            print(f"An error occurred launching Gradio: {error}")
            break