Update app.py

app.py

@@ -2,88 +2,103 @@ import gradio as gr
 import re
 
 # ---------------------------
-# 简易规则版 TIRA 分类器（可升级）
+# TIRA 规则分类器
 # ---------------------------
 def classify_tira(cn_text):
     text = cn_text.strip()
 
-    # Retain: 常用礼貌问候、称呼
-    retain_patterns = ["您好", "敬请", "麻烦您", "请查收"]
+    # Retain 保留
+    retain_patterns = ["您好", "敬请", "麻烦您", "请查收", "拜托您"]
     if any(p in text for p in retain_patterns):
-        return "Retain（保留）", "礼貌用语多为直译可保持原意，PDI 变化较小。"
+        return "Retain（保留）", "常规礼貌标记沿用 → 英译时多保持原本关系强度。"
 
-    # Amplify: 请求+模糊词，如“是否方便”“能否请您”
-    amplify_patterns = ["能否", "是否方便", "请您帮忙", "抽空"]
+    # Amplify 放大
+    amplify_patterns = ["能否", "是否方便", "请您帮忙", "抽空", "劳驾", "麻烦帮我"]
     if any(p in text for p in amplify_patterns):
-        return "Amplify（放大）", "英文可能出现更显性礼貌（please / possibly），使 PDI 上移。"
+        return "Amplify（放大）", "英文可能加入 please / possibly → 增加礼貌显性度 → PDI 上移。"
 
-    # Attenuate: 直接请求、没有客套
-    attenuate_patterns = ["请尽快", "需要您", "必须", "尽快处理"]
+    # Attenuate 削弱
+    attenuate_patterns = ["请尽快", "需要您", "必须", "麻烦尽快", "尽快处理"]
     if any(p in text for p in attenuate_patterns):
-        return "Attenuate（削弱）", "英文可能变得更直接，在目标语减弱礼貌，PDI 下移。"
+        return "Attenuate（削弱）", "英文可能变为更直接表达 → 礼貌减弱 → PDI 下移。"
 
-    # Redirect: 情感性、文化特定表达
-    redirect_patterns = ["给您添麻烦", "不好意思打扰", "不胜感激"]
+    # Redirect 转向
+    redirect_patterns = ["给您添麻烦", "不好意思打扰", "不胜感激", "心里很过意不去"]
     if any(p in text for p in redirect_patterns):
-        return "Redirect（转向）", "中文特有的情感敬辞，英文常转为名词化/被动式，方向不定。"
+        return "Redirect（转向）", "中文特有情感敬辞 → 英文常被被动化/名词化 → 方向不定。"
+
+    # 默认 → 按保留处理
+    return "Retain（保留）", "未出现特殊礼貌偏移 → 默认为保留型。"
 
-    return "Retain（保留）", "未检测到明显礼貌偏移特征，默认保持。"
 
 # ---------------------------
-# 建议英文生成（模板版）
+# 建议英文（模板版，可替换成 GPT 输出）
 # ---------------------------
 def suggest_english(cn_text, tira_type):
     if tira_type.startswith("Amplify"):
-        return "Could you please take a moment to look at the attached file?"
+        return "Could you please take a moment to review the attached file?"
     if tira_type.startswith("Attenuate"):
-        return "Please review the attachment when you have time."
+        return "Please review the attachment when possible."
     if tira_type.startswith("Redirect"):
         return "I apologize if this may cause any inconvenience. Please find the attachment below."
     return "Dear Professor, could you please take a look at the attached document?"
 
+
 # ---------------------------
-# 主函数
+# 主预测函数
 # ---------------------------
 def predict_tira(cn_text):
     tira_type, analysis = classify_tira(cn_text)
     en_suggestion = suggest_english(cn_text, tira_type)
 
-    risk = {
-        "Retain": "低风险（PDI 基本保持）",
-        "Amplify": "中风险（可能让关系显得更疏远）",
-        "Attenuate": "中偏高风险（可能显得更直接）",
-        "Redirect": "不稳定（可能造成理解偏差）"
+    risk_map = {
+        "Retain": "低风险（关系保持稳定）",
+        "Amplify": "中风险（英文更显远礼貌）",
+        "Attenuate": "中偏高（可能显得更直接）",
+        "Redirect": "不稳定（误解风险较高）"
     }
-
     key = tira_type.split("（")[0]
-    pdi_risk = risk.get(key, "低风险")
+    pdi_risk = risk_map.get(key, "低风险")
 
     return tira_type, analysis, pdi_risk, en_suggestion
 
 
 # ---------------------------
-# Gradio UI
+# 深蓝学术风 UI
 # ---------------------------
-with gr.Blocks(title="TIRA 预测工具") as demo:
+css = """
+h1 {text-align: center; color: #1B3B6F;}
+.gradio-container {background-color: #F7F9FC;}
+label {font-weight: bold;}
+"""
+
+with gr.Blocks(css=css, title="TIRA 学术邮件翻译分析工具") as demo:
     gr.Markdown("""
-    # **TIRA 学术邮件翻译预测工具**
-    ### Translation-Induced Re-Alignment (保留 / 放大 / 削弱 / 转向)
-    输入中文学术邮件 → 获得 TIRA 类型预测 + 翻译风险 + 建议英文表达。
+# **📘 TIRA Academic Email Translation Tool**
+### *Translation-Induced Re-Alignment (Retain / Amplify / Attenuate / Redirect)*  
+---
+
+输入一段中文��术邮件 → 系统将预测 TIRA 类型、给出解释、评估 PDI 风险，并提供建议英文表达。
     """)
 
-    cn_input = gr.Textbox(label="✉ 输入中文学术邮件内容", lines=5, placeholder="例如：能否请您帮我看一下附件？")
-    
-    btn = gr.Button("🔍 预测 TIRA 类型")
+    with gr.Row():
+        cn_input = gr.Textbox(
+            label="✉ 输入中文邮件内容",
+            lines=6,
+            placeholder="例如：能否请您帮我看一下附件？"
+        )
+
+    btn = gr.Button("🔍 开始预测", variant="primary")
 
     tira_output = gr.Textbox(label="📌 TIRA 类型预测")
-    analysis_output = gr.Textbox(label="📘 解释")
-    risk_output = gr.Textbox(label="⚠ PDI 风险")
-    en_suggestion_output = gr.Textbox(label="✏ 建议英文译文")
+    analysis_output = gr.Textbox(label="📘 解释说明")
+    risk_output = gr.Textbox(label="⚠ PDI 风险评估")
+    en_output = gr.Textbox(label="✏ 建议英文表达")
 
     btn.click(
         fn=predict_tira,
         inputs=cn_input,
-        outputs=[tira_output, analysis_output, risk_output, en_suggestion_output]
+        outputs=[tira_output, analysis_output, risk_output, en_output]
     )
 
 demo.launch()