Update app.py

app.py

@@ -5,6 +5,15 @@ import matplotlib.pyplot as plt
 import gradio as gr
 from transformers import pipeline
 
+# --------------------
+# 礼貌增强模型（T5）
+# --------------------
+polite_rewrite = pipeline(
+    "text2text-generation",
+    model="prithivida/parrot_paraphraser_on_T5"
+)
+
+
 
 # ---------- 1. 加载 Hugging Face 模型 ----------
 
@@ -136,6 +145,10 @@ def analyze_letter(chinese_text: str):
 
     # 2) 中 → 英 翻译
     translated = translator(chinese_text, max_length=512)[0]["translation_text"]
+    
+# 2.1) 礼貌增强版英文改写
+polite_prompt = f"Rewrite the following sentence in polite and respectful English: {translated}"
+polite_version = polite_rewrite(polite_prompt)[0]["generated_text"]
 
     # 3) 英文礼貌度
     pol_out = politeness_cls(translated)[0]
@@ -217,7 +230,16 @@ def analyze_letter(chinese_text: str):
     ax_radar.legend(loc="upper right", bbox_to_anchor=(1.3, 1.1))
     fig_radar.tight_layout()
 
-    return translated, zh_stats, en_stats, level, round(power_distance_score, 3), fig_bar, fig_radar
+    return (
+    translated,        # 原翻译
+    polite_version,    # 新增：更礼貌的翻译
+    zh_stats,
+    en_stats,
+    level,
+    round(power_distance_score, 3),
+    fig_bar,
+    fig_radar
+)
 
 
 # ---------- 6. Gradio 界面 ----------