Update app.py

app.py

@@ -37,24 +37,23 @@ def coref_chat(user_input):
         return "請輸入內容", "等待輸入..."
     
     try:
-        # 1. 偵測語系並統一轉換為「英文」供模型運算
-        # 判斷是否含有中文字
+        # 1. 偵測語系並轉換為英文進行運算 (中翻英橋接)
         has_chinese = any('\u4e00' <= char <= '\u9fff' for char in user_input)
         
         if has_chinese:
-            # 中翻英 (橋接運算)
+            # 強制中翻英
             working_text = GoogleTranslator(source='zh-CN', target='en').translate(user_input)
-            mode_notice = "【模式：中文輸入 ➔ 英文解析】"
+            mode_notice = "【模式：中文 ➔ 英文解析】"
         else:
-            # 已經是英文，直接運算
+            # 本身是英文
             working_text = user_input
-            mode_notice = "【模式：英文原語解析】"
+            mode_notice = "【模式：純英文解析】"
 
-        # 2. 執行指代消解 (Coreference Resolution)
+        # 2. 執行 AI 指代消解
         preds = model.predict(texts=[working_text])
         clusters = preds[0].get_clusters()
         
-        # 3. 整理結果並「翻譯回中文」
+        # 3. 輸出結果整理
         result = f"✨ {mode_notice}\n"
         result += f"📝 英文邏輯空間: {working_text}\n"
         result += "---------------------------------\n"
@@ -62,13 +61,22 @@ def coref_chat(user_input):
         if not clusters:
             result += "🔍 分析結果：指代關係明確，或模型判定關聯度未達門檻。"
         else:
-            result += "🎯【偵測到之實體鏈】:\n"
+            result += "🎯【偵測到之實體鏈 (Entity Chains)】:\n"
             for i, cluster in enumerate(clusters):
-                # 將英文實體鏈翻譯回中文，讓使用者更好懂
-                cluster_str = ' ↔ '.join(cluster)
-                translated_cluster = GoogleTranslator(source='en', target='zh-CN').translate(cluster_str)
-                result += f" 🔗 鏈結 {i+1} (中): {translated_cluster}\n"
-                result += f"    └─ (英): {cluster_str}\n"
+                # 原始英文鏈結
+                cluster_str_en = ' ↔ '.join(cluster)
+                
+                # --- 強制翻譯回中文 ---
+                # 我們把連連看的內容拆開翻譯，再組合回去，這樣翻譯會更準確
+                try:
+                    translated_items = [GoogleTranslator(source='en', target='zh-CN').translate(item) for item in cluster]
+                    cluster_str_zh = ' ↔ '.join(translated_items)
+                except:
+                    # 萬一分開翻失敗，才整串翻
+                    cluster_str_zh = GoogleTranslator(source='en', target='zh-CN').translate(cluster_str_en)
+                
+                result += f" 🔗 鏈結 {i+1} (繁中): {cluster_str_zh}\n"
+                result += f"    └─ (原文): {cluster_str_en}\n"
 
         return user_input, result