Update app.py

app.py

@@ -2,29 +2,32 @@ import gradio as gr
 from src.models.deepseek import DeepSeek
 from src.models.flan import FlanT5
 
-# 初始化模型实例
 deepseek = DeepSeek()
 flan = FlanT5()
 
 def generate_all(prompt, temperature):
-    return (
-        deepseek.generate(prompt, temperature),
-        flan.generate(prompt, temperature)
-    )
+    try:
+        return (
+            deepseek.generate(prompt, temperature),
+            flan.generate(prompt, temperature)
+        )
+    except Exception as e:
+        return f"DeepSeek Error: {e}", f"Flan Error: {e}"
 
 def render_arena_tab():
     with gr.Tab("Arena"):
-        gr.Markdown("### 中文语言模型对比实验 - Arena 模式")
+        gr.Markdown("## ✨ Prompt-to-Prompt 对比实验：DeepSeek vs Flan-T5")
         with gr.Row():
             input_box = gr.Textbox(
-                label="请输入中文指令",
-                placeholder="例如：请写一篇关于人工智能发展的议论文",
+                label="请输入 Prompt",
+                placeholder="例如：写一首关于科技未来的诗",
                 lines=4
             )
         with gr.Row():
             temp_slider = gr.Slider(0.1, 1.5, value=0.7, step=0.1, label="Temperature")
         with gr.Row():
-            submit_btn = gr.Button("提交")
+            submit_btn = gr.Button("生成对比结果")
+            clear_btn = gr.Button("清空")
         with gr.Row():
             output_deepseek = gr.Textbox(label="🌊 DeepSeek 输出", lines=10)
             output_flan = gr.Textbox(label="🧠 Flan-T5 输出", lines=10)
@@ -33,20 +36,28 @@ def render_arena_tab():
             inputs=[input_box, temp_slider],
             outputs=[output_deepseek, output_flan]
         )
+        clear_btn.click(
+            lambda: ("", "", ""),
+            None,
+            [input_box, output_deepseek, output_flan]
+        )
 
 def render_report_tab():
     with gr.Tab("Report"):
-        gr.Markdown(open("report.md", encoding="utf-8").read())
-
-def render_benchmark_tab():
-    with gr.Tab("LLM Benchmark"):
-        gr.Markdown("此处可展示 GRACE 框架维度对比图")
+        try:
+            with open("report.md", "r", encoding="utf-8") as f:
+                content = f.read()
+        except FileNotFoundError:
+            content = "# ❗ 未找到 report.md 文件，请补充"
+        gr.Markdown(content)
 
 def main():
-    with gr.Blocks(title="中文语言模型对比实验") as demo:
+    with gr.Blocks(title="语言生成模型对比实验") as demo:
+        gr.Markdown("## 🤖 语言生成模型对比实验平台\n\n对比模型：`DeepSeek LLM` 与 `Flan-T5 Base`")
+
         render_arena_tab()
-        render_benchmark_tab()
         render_report_tab()
+
     demo.launch(share=True, debug=True)
 
 if __name__ == "__main__":