Update app.py

app.py

@@ -249,7 +249,7 @@ def classify_email(json_input):
 
 
 # ===================== 核心处理函数（模拟邮件分析） =====================
-def classify_email(sender: str, subject: str, body: str) -> tuple[str, str, str]:
+def classify_email(sender: str, subject: str, body: str) -> tuple[str, str]:
     email = {
         "sender": sender,
         "subject": subject,
@@ -264,7 +264,7 @@ def classify_email(sender: str, subject: str, body: str) -> tuple[str, str, str]
         "draft_response": None,
         "messages": []
     })
-    return result.get("is_spam"), result.get("spam_reason"), result.get("email_category")
+    return result.get("is_spam"), result.get("spam_reason")
 
 
 # ===================== Gradio 表单界面构建 =====================
@@ -308,14 +308,13 @@ with gr.Blocks(title="邮件分析表单", theme=gr.themes.Soft()) as demo:
     gr.Markdown("### 📝 分析结果")
     with gr.Row():
         is_spam_output = gr.Textbox(label="是否垃圾邮件", interactive=False)
-        category_output = gr.Textbox(label="邮件分类", interactive=False)
         reason_output = gr.Textbox(label="判定原因", lines=3, interactive=False)
 
     # 绑定按钮与处理函数
     submit_btn.click(
         fn=classify_email,
         inputs=[sender_input, subject_input, content_input],  # 3个输入框
-        outputs=[is_spam_output, reason_output, category_output]  # 3个输出框
+        outputs=[is_spam_output, reason_output]  # 2个输出框
     )
 
     # 重置按钮（可选）
@@ -323,7 +322,7 @@ with gr.Blocks(title="邮件分析表单", theme=gr.themes.Soft()) as demo:
     reset_btn.click(
         fn=lambda: ("", "", "", "", "", ""),  # 重置所有输入输出
         inputs=[],
-        outputs=[sender_input, subject_input, content_input, is_spam_output, reason_output, category_output]
+        outputs=[sender_input, subject_input, content_input, is_spam_output, reason_output]
     )
 
 # ===================== 启动界面 =====================