Update app.py

app.py

@@ -4,7 +4,7 @@ from argparse import ArgumentParser
 from loguru import logger
 from tools.llama.generate import launch_thread_safe_queue
 from tools.vqgan.inference import load_model as load_decoder_model
-from tools.webui import build_app
+import gradio as gr  # 导入 Gradio
 
 def parse_args():
     parser = ArgumentParser()
@@ -72,7 +72,20 @@ def inference(
     # 模拟推理过程
     result = f"Processed text: {text}"
     return result
-    
+
+def inference_function(text):
+    return f"Processed: {text}"
+
+def build_app(args):
+    with gr.Blocks() as app:
+        gr.Markdown(f"# Fish Speech Inference - Theme: {args.theme}")
+        text_input = gr.Textbox(label="Input Text")
+        output = gr.Textbox(label="Output Text")
+        submit_button = gr.Button("Submit")
+
+        submit_button.click(fn=inference_function, inputs=text_input, outputs=output)
+    return app
+
 def main():
     args = parse_args()
 
@@ -110,8 +123,8 @@ def main():
 
     logger.info("Warming up done, launching the web UI...")
 
-    # Launch the Gradio app
-    app = build_app()
+    # Launch the Gradio app, passing args to build_app
+    app = build_app(args)
     app.launch(show_api=True)