|
1 |
| -import gradio as gr |
2 |
| -import sys, os |
3 |
| -from tabs.full_inference import full_inference_tab |
4 |
| -from tabs.download_model import download_model_tab |
5 |
| -from tabs.download_music import download_music_tab |
6 |
| -from tabs.settings import select_themes_tab, lang_tab, restart_tab |
7 |
| - |
8 |
| -now_dir = os.getcwd() |
9 |
| -sys.path.append(now_dir) |
10 |
| -DEFAULT_PORT = 7755 |
11 |
| -MAX_PORT_ATTEMPTS = 10 |
12 |
| - |
13 |
| -from assets.i18n.i18n import I18nAuto |
14 |
| - |
15 |
| -i18n = I18nAuto() |
16 |
| - |
17 |
| -import assets.themes.loadThemes as loadThemes |
18 |
| - |
19 |
| -rvc_theme = loadThemes.load_json() or "NoCrypt/miku" |
20 |
| - |
21 |
| -with gr.Blocks( |
22 |
| - theme=rvc_theme, title="Advanced-RVC-Inference" |
23 |
| -) as rvc: |
24 |
| - gr.Markdown("# Advanced-RVC-Inference") |
25 |
| - with gr.Tab(i18n("Full Inference")): |
26 |
| - full_inference_tab() |
27 |
| - with gr.Tab(i18n("Download Music")): |
28 |
| - download_music_tab() |
29 |
| - with gr.Tab(i18n("Download Model")): |
30 |
| - download_model_tab() |
31 |
| - with gr.Tab(i18n("Settings")): |
32 |
| - select_themes_tab() |
33 |
| - lang_tab() |
34 |
| - restart_tab() |
35 |
| - |
36 |
| - |
37 |
| - |
38 |
| -def launch(port): |
39 |
| - rvc.launch( |
40 |
| - share="--share" in sys.argv, |
41 |
| - inbrowser="--open" in sys.argv, |
42 |
| - server_port=port, |
43 |
| - ) |
44 |
| - |
45 |
| - |
46 |
| -def get_port_from_args(): |
47 |
| - if "--port" in sys.argv: |
48 |
| - port_index = sys.argv.index("--port") + 1 |
49 |
| - if port_index < len(sys.argv): |
50 |
| - return int(sys.argv[port_index]) |
51 |
| - return DEFAULT_PORT |
52 |
| - |
53 |
| - |
54 |
| -if __name__ == "__main__": |
55 |
| - port = get_port_from_args() |
56 |
| - for _ in range(MAX_PORT_ATTEMPTS): |
57 |
| - try: |
58 |
| - launch(port) |
59 |
| - break |
60 |
| - except OSError: |
61 |
| - print( |
62 |
| - f"Failed to launch on port {port}, trying again on port {port - 1}..." |
63 |
| - ) |
64 |
| - port -= 1 |
65 |
| - except Exception as error: |
66 |
| - print(f"An error occurred launching Gradio: {error}") |
67 |
| - break |
| 1 | +import gradio as gr |
| 2 | +import sys, os |
| 3 | +from tabs.full_inference import full_inference_tab |
| 4 | +from tabs.download_model import download_model_tab |
| 5 | +from tabs.download_music import download_music_tab |
| 6 | +from tabs.settings import select_themes_tab, lang_tab, restart_tab |
| 7 | + |
| 8 | +now_dir = os.getcwd() |
| 9 | +sys.path.append(now_dir) |
| 10 | +DEFAULT_PORT = 7755 |
| 11 | +MAX_PORT_ATTEMPTS = 10 |
| 12 | + |
| 13 | +from assets.i18n.i18n import I18nAuto |
| 14 | + |
| 15 | +i18n = I18nAuto() |
| 16 | + |
| 17 | +import assets.themes.loadThemes as loadThemes |
| 18 | + |
| 19 | +rvc_theme = loadThemes.load_json() or "NoCrypt/miku" |
| 20 | + |
| 21 | +with gr.Blocks( |
| 22 | + theme=rvc_theme, title="Advanced-RVC-Inference" |
| 23 | +) as rvc: |
| 24 | + gr.Markdown("# Advanced-RVC-Inference") |
| 25 | + with gr.Tab(i18n("Full Inference")): |
| 26 | + full_inference_tab() |
| 27 | + with gr.Tab(i18n("Download Music")): |
| 28 | + download_music_tab() |
| 29 | + with gr.Tab(i18n("Download Model")): |
| 30 | + download_model_tab() |
| 31 | + with gr.Tab(i18n("Settings")): |
| 32 | + select_themes_tab() |
| 33 | + lang_tab() |
| 34 | + restart_tab() |
| 35 | + |
| 36 | + |
| 37 | + |
| 38 | +def launch(port): |
| 39 | + rvc.launch( |
| 40 | + share="--share" in sys.argv, |
| 41 | + inbrowser="--open" in sys.argv, |
| 42 | + server_port=port, |
| 43 | + ) |
| 44 | + |
| 45 | + |
| 46 | +def get_port_from_args(): |
| 47 | + if "--port" in sys.argv: |
| 48 | + port_index = sys.argv.index("--port") + 1 |
| 49 | + if port_index < len(sys.argv): |
| 50 | + return int(sys.argv[port_index]) |
| 51 | + return DEFAULT_PORT |
| 52 | + |
| 53 | + |
| 54 | +if __name__ == "__main__": |
| 55 | + port = get_port_from_args() |
| 56 | + for _ in range(MAX_PORT_ATTEMPTS): |
| 57 | + try: |
| 58 | + launch(port) |
| 59 | + break |
| 60 | + except OSError: |
| 61 | + print( |
| 62 | + f"Failed to launch on port {port}, trying again on port {port - 1}..." |
| 63 | + ) |
| 64 | + port -= 1 |
| 65 | + except Exception as error: |
| 66 | + print(f"An error occurred launching Gradio: {error}") |
| 67 | + break |
0 commit comments