Cloyir/index-tts2-ForDgxSpark
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(webui.py): add glossary term validation and error handling

Browse Source
This commit is contained in:
Yrom 2025-11-20 14:22:42 +08:00
parent fa7962f1f2
commit a7099c4a6e
No known key found for this signature in database
2 changed files with 31 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tools/i18n/locale/en_US.json
View File

@ -57,5 +57,9 @@
"术语": "Term", "术语": "Term",
"已删除": "Deleted", "已删除": "Deleted",
"英文读法": "English pronunciation", "英文读法": "English pronunciation",
"自定义术语词汇读音": "Customize term pronunciations" "自定义术语词汇读音": "Customize term pronunciations",
"中文读法": "Chinese pronunciation",
"词汇表已更新": "Glossary updated successfully",
"保存词汇表时出错": "Error saving glossary",
"加载词汇表时出错": "Error loading glossary"
} }

25
webui.py
View File

@ -395,6 +395,10 @@ with gr.Blocks(title="IndexTTS Demo") as demo:
# 术语词汇表事件处理函数 # 术语词汇表事件处理函数
def on_add_glossary_term(term, reading_zh, reading_en): def on_add_glossary_term(term, reading_zh, reading_en):
"""添加术语到词汇表并自动保存""" """添加术语到词汇表并自动保存"""
term = term.strip()
reading_zh = reading_zh.strip()
reading_en = reading_en.strip()
if not term: if not term:
gr.Warning(i18n("请输入术语")) gr.Warning(i18n("请输入术语"))
return gr.update() return gr.update()
@ -418,7 +422,13 @@ with gr.Blocks(title="IndexTTS Demo") as demo:
tts.normalizer.term_glossary[term] = reading tts.normalizer.term_glossary[term] = reading
# 自动保存到文件 # 自动保存到文件
try:
tts.normalizer.save_glossary_to_yaml(tts.glossary_path) tts.normalizer.save_glossary_to_yaml(tts.glossary_path)
gr.Info(i18n("词汇表已更新"), duration=1)
except Exception as e:
gr.Error(i18n("保存词汇表时出错"))
print(f"Error details: {e}")
return gr.update()
# 更新Markdown表格 # 更新Markdown表格
return gr.update(value=format_glossary_markdown()) return gr.update(value=format_glossary_markdown())
@ -508,6 +518,15 @@ with gr.Blocks(title="IndexTTS Demo") as demo:
inputs=[], inputs=[],
outputs=[gen_button]) outputs=[gen_button])
def on_demo_load():
"""页面加载时重新加载glossary数据"""
try:
tts.normalizer.load_glossary_from_yaml(tts.glossary_path)
except Exception as e:
gr.Error(i18n("加载词汇表时出错"))
print(f"Failed to reload glossary on page load: {e}")
return gr.update(value=format_glossary_markdown())
# 术语词汇表事件绑定 # 术语词汇表事件绑定
btn_add_term.click( btn_add_term.click(
on_add_glossary_term, on_add_glossary_term,
@ -515,6 +534,12 @@ with gr.Blocks(title="IndexTTS Demo") as demo:
outputs=[glossary_table] outputs=[glossary_table]
) )
# 页面加载时重新加载glossary
demo.load(
on_demo_load,
inputs=[],
outputs=[glossary_table]
)
gen_button.click(gen_single, gen_button.click(gen_single,
inputs=[emo_control_method,prompt_audio, input_text_single, emo_upload, emo_weight, inputs=[emo_control_method,prompt_audio, input_text_single, emo_upload, emo_weight,