Update app.py

app.py

@@ -508,20 +508,28 @@ class TextToSQLSystem:
         # ==============================================================================
         entity_match_data = None
         entity_patterns = [
-            # 模式1: 匹配 "类型 + ID" (e.g., "买家ID C0761N") - 最高优先级
-            {'pattern': r"(買家|buyer)\s*(?:id|代號|代碼)\s*'\"?\b([A-Z]\d{4}[A-Z])\b'\"?", 'column': 'sd.BuyerID', 'type': '買家ID'},
-            {'pattern': r"(申請方|申請廠商|applicant)\s*(?:id|代號|代碼)\s*'\"?\b([A-Z]\d{4}[A-Z])\b'\"?", 'column': 'sd.ApplicantID', 'type': '申請方ID'},
-            {'pattern': r"(付款方|付款廠商|invoiceto)\s*(?:id|代號|代碼)\s*'\"?\b([A-Z]\d{4}[A-Z])\b'\"?", 'column': 'sd.InvoiceToID', 'type': '付款方ID'},
-            {'pattern': r"(代理商|agent)\s*(?:id|代號|代碼)\s*'\"?\b([A-Z]\d{4}[A-Z])\b'\"?", 'column': 'sd.AgentID', 'type': '代理商ID'},
-            
-            # 模式2: 匹配 "類型 + 名稱" (e.g., "買家 Gap")
-            {'pattern': r"(買家|buyer|客戶)\s*'\"?([a-zA-Z0-9&.\s-]+?)(?:\s*的|\s+|$|有)", 'column': 'sd.BuyerName', 'type': '買家'},
-            {'pattern': r"(申請方|申請廠商|applicant)\s*'\"?([a-zA-Z0-9&.\s-]+?)(?:\s*的|\s+|$|有)", 'column': 'sd.ApplicantName', 'type': '申請方'},
-            {'pattern': r"(付款方|付款廠商|invoiceto)\s*'\"?([a-zA-Z0-9&.\s-]+?)(?:\s*的|\s+|$|有)", 'column': 'sd.InvoiceToName', 'type': '付款方'},
-            {'pattern': r"(代理商|agent)\s*'\"?([a-zA-Z0-9&.\s-]+?)(?:\s*的|\s+|$|有)", 'column': 'sd.AgentName', 'type': '代理商'},
-            
-            # 模式3: 单独匹配一个 ID (e.g., "c0761n") - 较低优先级
-            {'pattern': r"\b([A-Z]\d{4}[A-Z])\b", 'column': 'sd.ApplicantID', 'type': 'ID'}
+        # 模式1: 匹配 "类型 + ID" (e.g., "买家ID C0761N") - 最高优先级
+        # 增加了繁體的 "買家", "代號", "代碼"
+        {'pattern': r"(買家|买家|buyer)\s*(?:id|代號|代碼|代号|代码)\s*'\"?\b([A-Z]\d{4}[A-Z])\b'\"?", 'column': 'sd.BuyerID', 'type': '買家ID'},
+        # 增加了繁體的 "申請方", "申請廠商"
+        {'pattern': r"(申請方|申请方|申請廠商|申请厂商|applicant)\s*(?:id|代號|代碼|代号|代码)\s*'\"?\b([A-Z]\d{4}[A-Z])\b'\"?", 'column': 'sd.ApplicantID', 'type': '申請方ID'},
+        # 增加了繁體的 "付款方", "付款廠商"
+        {'pattern': r"(付款方|付款厂商|invoiceto)\s*(?:id|代號|代碼|代号|代码)\s*'\"?\b([A-Z]\d{4}[A-Z])\b'\"?", 'column': 'sd.InvoiceToID', 'type': '付款方ID'},
+        # 增加了繁體的 "代理商"
+        {'pattern': r"(代理商|agent)\s*(?:id|代號|代碼|代号|代码)\s*'\"?\b([A-Z]\d{4}[A-Z])\b'\"?", 'column': 'sd.AgentID', 'type': '代理商ID'},
+        
+        # 模式2: 匹配 "類型 + 名稱" (e.g., "買家 Gap")
+        # 增加了繁體的 "買家"
+        {'pattern': r"(買家|买家|buyer|客戶)\s*'\"?([a-zA-Z0-9&.\s-]+?)(?:\s*的|\s+|$|有)", 'column': 'sd.BuyerName', 'type': '買家'},
+        # 增加了繁體的 "申請方", "申請廠商"
+        {'pattern': r"(申請方|申请方|申請廠商|申请厂商|applicant)\s*'\"?([a-zA-Z0-9&.\s-]+?)(?:\s*的|\s+|$|有)", 'column': 'sd.ApplicantName', 'type': '申請方'},
+        # 增加了繁體的 "付款方", "付款廠商"
+        {'pattern': r"(付款方|付款厂商|invoiceto)\s*'\"?([a-zA-Z0-9&.\s-]+?)(?:\s*的|\s+|$|有)", 'column': 'sd.InvoiceToName', 'type': '付款方'},
+        # 增加了繁體的 "代理商"
+        {'pattern': r"(代理商|agent)\s*'\"?([a-zA-Z0-9&.\s-]+?)(?:\s*的|\s+|$|有)", 'column': 'sd.AgentName', 'type': '代理商'},
+        
+        # 模式3: 单独匹配一个 ID (e.g., "c0761n") - 较低优先级
+        {'pattern': r"\b([A-Z]\d{4}[A-Z])\b", 'column': 'sd.ApplicantID', 'type': 'ID'}
         ]
         
         for p in entity_patterns: