Update app.py

app.py

@@ -399,7 +399,85 @@ with gr.Blocks() as app_tts:
 with gr.Blocks() as app_multistyle:
     gr.Markdown("# Multiple Speech-Type Generation")
     # ... [Keep original multistyle interface unchanged] ...
+with gr.Blocks() as app_chat:
+    gr.Markdown("# Voice Chat")
+    # ... [Keep original voice chat interface unchanged] ...
+
+with gr.Blocks() as app:
+    gr.Markdown(f"""
+# E2/F5 TTS
+{"Local web UI for [F5 TTS](https://github.com/SWivid/F5-TTS)" if not USING_SPACES else "Online demo for [F5-TTS](https://github.com/SWivid/F5-TTS)"}
+""")
+    
+    with gr.Row():
+        if not USING_SPACES:
+            choose_tts_model = gr.Radio(
+                choices=[DEFAULT_TTS_MODEL, "E2-TTS", "Custom"],
+                label="TTS Model",
+                value=DEFAULT_TTS_MODEL
+            )
+        else:
+            choose_tts_model = gr.Radio(
+                choices=[DEFAULT_TTS_MODEL, "E2-TTS"],
+                label="TTS Model",
+                value=DEFAULT_TTS_MODEL
+            )
+        
+        custom_ckpt_path = gr.Dropdown(
+            choices=[DEFAULT_TTS_MODEL_CFG[0]],
+            value=load_last_used_custom()[0],
+            allow_custom_value=True,
+            label="Model Path",
+            visible=False
+        )
+        custom_vocab_path = gr.Dropdown(
+            choices=[DEFAULT_TTS_MODEL_CFG[1]],
+            value=load_last_used_custom()[1],
+            allow_custom_value=True,
+            label="Vocab Path",
+            visible=False
+        )
+        custom_model_cfg = gr.Dropdown(
+            choices=[DEFAULT_TTS_MODEL_CFG[2]],
+            value=load_last_used_custom()[2],
+            allow_custom_value=True,
+            label="Model Config",
+            visible=False
+        )
+
+    choose_tts_model.change(
+        switch_tts_model,
+        inputs=[choose_tts_model],
+        outputs=[custom_ckpt_path, custom_vocab_path, custom_model_cfg]
+    )
+    
+    gr.TabbedInterface(
+        [app_tts, app_podcast, app_multistyle, app_chat, app_credits],
+        ["Basic TTS", "Podcast", "Multi-Style", "Voice Chat", "Credits"],
+    )
 
+@click.command()
+@click.option("--port", "-p", default=None, type=int)
+@click.option("--host", "-H", default=None)
+@click.option("--share", "-s", default=True, is_flag=True)
+@click.option("--api", "-a", default=True, is_flag=True)
+@click.option("--root_path", "-r", default=None)
+def main(port, host, share, api, root_path):
+    global app
+    print("Launching app...")
+    app.queue(api_open=api).launch(
+        server_name=host,
+        server_port=port,
+        share=share,
+        show_api=api,
+        root_path=root_path
+    )
+
+if __name__ == "__main__":
+    if not USING_SPACES:
+        main()
+    else:
+        app.queue().launch()
 # ========== REST OF ORIGINAL CODE (UNCHANGED BELOW) ========== #
 # [Main app configuration, other tabs (TTS, Multistyle, Chat, Credits)]
 # [Keep all original code after this point exactly as it was]