zhzluke96
update
d2b7e94
import logging
import os
import gradio as gr
from modules import config
from modules.webui import gradio_extensions, webui_config
from modules.webui.changelog_tab import create_changelog_tab
from modules.webui.finetune.ft_tab import create_ft_tabs
from modules.webui.localization_runtime import ENLocalizationVars, ZHLocalizationVars
from modules.webui.readme_tab import create_readme_tab
from modules.webui.speaker_tab import create_speaker_panel
from modules.webui.ssml.podcast_tab import create_ssml_podcast_tab
from modules.webui.ssml.spliter_tab import create_spliter_tab
from modules.webui.ssml.ssml_tab import create_ssml_interface
from modules.webui.system_tab import create_system_tab
from modules.webui.tts_tab import create_tts_interface
logger = logging.getLogger(__name__)
def webui_init():
# fix: If the system proxy is enabled in the Windows system, you need to skip these
os.environ["NO_PROXY"] = "localhost,127.0.0.1,0.0.0.0"
if config.runtime_env_vars.language == "en":
webui_config.localization = ENLocalizationVars()
else:
webui_config.localization = ZHLocalizationVars()
logger.info("WebUI module initialized")
def create_app_footer():
gradio_version = gr.__version__
git_tag = os.environ.get("V_GIT_TAG") or config.versions.git_tag
git_commit = os.environ.get("V_GIT_COMMIT") or config.versions.git_commit
git_branch = os.environ.get("V_GIT_BRANCH") or config.versions.git_branch
python_version = config.versions.python_version
torch_version = config.versions.torch_version
ffmpeg_version = config.versions.ffmpeg_version
config.versions.gradio_version = gradio_version
footer_items = ["🍦 [ChatTTS-Forge](https://github.com/lenML/ChatTTS-Forge)"]
footer_items.append(
f"version: [{git_tag}](https://github.com/lenML/ChatTTS-Forge/commit/{git_commit})"
)
footer_items.append(f"branch: `{git_branch}`")
footer_items.append(f"python: `{python_version}`")
footer_items.append(f"torch: `{torch_version}`")
footer_items.append(f"ffmpeg: `{ffmpeg_version}`")
if config.runtime_env_vars.api and not config.runtime_env_vars.no_docs:
footer_items.append(f"[api](/docs)")
gr.Markdown(
" | ".join(footer_items),
elem_classes=["no-translate"],
)
def create_interface():
gradio_extensions.reload_javascript()
js_func = """
function refresh() {
const url = new URL(window.location);
if (url.searchParams.get('__theme') !== 'dark') {
url.searchParams.set('__theme', 'dark');
window.location.href = url.href;
}
}
"""
head_js = """
<script>
</script>
"""
with gr.Blocks(js=js_func, head=head_js, title="ChatTTS Forge WebUI") as demo:
css = """
<style>
.big-button {
height: 80px;
}
#input_title div.eta-bar {
display: none !important; transform: none !important;
}
footer {
display: none !important;
}
</style>
"""
gr.HTML(css)
with gr.Tabs() as tabs:
with gr.TabItem("TTS"):
create_tts_interface()
with gr.TabItem("SSML", id="ssml"):
with gr.Tabs() as ssml_tabs:
with gr.TabItem("Editor", id="ssml.editor"):
ssml_input = create_ssml_interface()
with gr.TabItem("Spilter"):
create_spliter_tab(
ssml_input=ssml_input, tabs1=tabs, tabs2=ssml_tabs
)
with gr.TabItem("Podcast"):
create_ssml_podcast_tab(
ssml_input=ssml_input, tabs1=tabs, tabs2=ssml_tabs
)
with gr.TabItem("Speaker"):
create_speaker_panel()
with gr.TabItem("Inpainting", visible=webui_config.experimental):
gr.Markdown("🚧 Under construction")
with gr.TabItem("ASR", visible=webui_config.experimental):
gr.Markdown("🚧 Under construction")
with gr.TabItem("Finetune", visible=webui_config.experimental):
create_ft_tabs(demo)
with gr.TabItem("System"):
create_system_tab()
with gr.TabItem("README"):
with gr.Tabs():
with gr.TabItem("readme"):
create_readme_tab()
with gr.TabItem("changelog"):
create_changelog_tab()
create_app_footer()
# Dump the English config for the localization
# ** JUST for developer
# localization.dump_english_config(gradio_hijack.all_components)
return demo