Update app.py

app.py

@@ -5,7 +5,7 @@ import requests
 import tempfile
 from languages import languages
 
-def generate_audio(input_text, voice, output_format, base_url_input, api_key_input, request: gr.Request):
+def generate_audio(input_text, model, voice, output_format, base_url_input, api_key_input, request: gr.Request):
     # 获取 URL 参数
     query_params = request.query_params
     # 优先使用 URL 参数中的 base_url 和 api_key
@@ -13,7 +13,7 @@ def generate_audio(input_text, voice, output_format, base_url_input, api_key_inp
     api_key = query_params.get("api_key", api_key_input)
     # 准备请求负载
     payload = {
-        "model": "tts-1",
+        "model": model,  # 使用选择的模型
         "input": input_text,
         "voice": voice.lower(),
         "format": output_format
@@ -38,8 +38,8 @@ def generate_audio(input_text, voice, output_format, base_url_input, api_key_inp
     # 返回文件路径
     return temp_file.name
 
-def on_generate_click(input_text_value, voice_value, output_format_value, base_url_value, api_key_value, request: gr.Request):
-    audio_path = generate_audio(input_text_value, voice_value, output_format_value, base_url_value, api_key_value, request)
+def on_generate_click(input_text_value, model_value, voice_value, output_format_value, base_url_value, api_key_value, request: gr.Request):
+    audio_path = generate_audio(input_text_value, model_value, voice_value, output_format_value, base_url_value, api_key_value, request)
     return audio_path
 
 def update_settings_from_url(request: gr.Request):
@@ -61,6 +61,7 @@ def update_language(language_choice):
         gr.update(value=f"# {labels['title']}"),  # title_markdown
         gr.update(value=labels['intro']),  # intro_text
         gr.update(label=labels['input_text_label'], placeholder=labels['input_text_placeholder']),  # input_text
+        gr.update(label=labels['model_label']),  # model
         gr.update(label=labels['voice_label']),  # voice
         gr.update(label=labels['output_format_label']),  # output_format
         gr.update(value=labels['generate_button_label']),  # generate_button
@@ -94,6 +95,8 @@ with gr.Blocks(css=None, theme="Zarkel/IBM_Carbon_Theme", title="TTS Web") as de
             intro_text = gr.Markdown(languages["中文"]["intro"])  # 初始加载中文的介绍
             # 输入文本
             input_text = gr.Textbox(label="输入文本", placeholder="请输入要合成的文本")
+            # 选择模型
+            model = gr.Radio(choices=["tts-1", "tts-1-hd"], label="模型", value="tts-1")
             # 选择声音
             voice = gr.Radio(choices=["Alloy", "Echo", "Fable", "Onyx", "Nova", "Shimmer"], label="声音", value="Alloy")
             # 选择输出格式
@@ -128,6 +131,7 @@ with gr.Blocks(css=None, theme="Zarkel/IBM_Carbon_Theme", title="TTS Web") as de
             title_markdown,
             intro_text,  # 添加 intro_text 的更新
             input_text,
+            model,        # 更新模型组件
             voice,
             output_format,
             generate_button,
@@ -159,8 +163,8 @@ with gr.Blocks(css=None, theme="Zarkel/IBM_Carbon_Theme", title="TTS Web") as de
     # 按钮点击事件
     generate_button.click(
         on_generate_click,
-        inputs=[input_text, voice, output_format, base_url_input, api_key_input],
+        inputs=[input_text, model, voice, output_format, base_url_input, api_key_input, gr.Request()],
         outputs=[audio_output]
     )
 
-demo.launch(server_name="0.0.0.0", server_port=7860)
+demo.launch(server_name="0.0.0.0", server_port=7860)