Update app.py

app.py

@@ -498,8 +498,11 @@ class TextToSQLSystem:
 
     def _validate_and_fix_sql(self, question: str, raw_response: str) -> Tuple[Optional[str], str]:
         """
-        (V28 / 實驗組別名映射版)
-        增加了實驗組別名到資料庫真實值的映射，以處理 "A組" -> "TA" 這類轉換。
+        (V29 / 穩健正則 + 智能計數 最終版)
+        一個多層次的SQL生成引擎。它優先使用基於規則的動態模板生成器，
+        如果無法匹配，則回退到解析和修正AI模型的輸出。
+        - 使用更簡潔、穩健的正則表達式來捕獲實體名稱。
+        - 根據問題是關於「報告」還是「測試項目」來智能地決定計數目標。
         """
         q_lower = question.lower()
     
@@ -507,21 +510,21 @@ class TextToSQLSystem:
         #  第零層：統一實體識別引擎 (Unified Entity Recognition Engine)
         # ==============================================================================
         entity_match_data = None
-        # (您的 entity_patterns 邏輯保持不變，此處省略以保持簡潔)
+        # 包含了繁簡體兼容和更穩健的模式
         entity_patterns = [
-            # 模式1: 匹配 "类型 + ID" (e.g., "买家ID C0761N") - 最高优先级
+            # 模式1: 匹配 "类型 + ID" - (保持不變)
             {'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': '代理商'},
+            # 模式2: 匹配 "類型 + 名稱" - (簡化了模式，使其更穩健)
+            {'pattern': r"(買家|买家|buyer|客戶)\s+([a-zA-Z0-9&.-]+)", 'column': 'sd.BuyerName', 'type': '買家'},
+            {'pattern': r"(申請方|申请方|申請廠商|申请厂商|applicant)\s+([a-zA-Z0-9&.-]+)", 'column': 'sd.ApplicantName', 'type': '申請方'},
+            {'pattern': r"(付款方|付款厂商|invoiceto)\s+([a-zA-Z0-9&.-]+)", 'column': 'sd.InvoiceToName', 'type': '付款方'},
+            {'pattern': r"(代理商|agent)\s+([a-zA-Z0-9&.-]+)", 'column': 'sd.AgentName', 'type': '代理商'},
             
-            # 模式3: 单独匹配一个 ID (e.g., "c0761n") - 较低优先级
+            # 模式3: 单独匹配一个 ID - (保持不變)
             {'pattern': r"\b([A-Z]\d{4}[A-Z])\b", 'column': 'sd.ApplicantID', 'type': 'ID'}
         ]
         
@@ -542,23 +545,22 @@ class TextToSQLSystem:
         
         intents = {}
         sql_components = {
-            'select': [],
-            'from': "",
-            'joins': [],
-            'where': [],
-            'group_by': [],
-            'order_by': [],
-            'log_parts': []
+            'select': [], 'from': "", 'joins': [], 'where': [],
+            'group_by': [], 'order_by': [], 'log_parts': []
         }
     
         # --- 運行一系列獨立的意圖偵測器 ---
         
         # 偵測器 2.1: 核心動作意圖
-        # (此處邏輯不變)
         if any(kw in q_lower for kw in ['幾份', '多少', '數量', '總數', 'how many', 'count']):
             intents['action'] = 'count'
-            sql_components['select'].append("COUNT(DISTINCT jt.JobNo) AS report_count")
-            sql_components['log_parts'].append("報告總數")
+            # 智能決定計數目標
+            if "測試項目" in question or "test item" in q_lower:
+                sql_components['select'].append("COUNT(jip.ItemCode) AS item_count")
+                sql_components['log_parts'].append("測試項目總數")
+            else: # 預設是計數報告
+                sql_components['select'].append("COUNT(DISTINCT jt.JobNo) AS report_count")
+                sql_components['log_parts'].append("報告總數")
         elif any(kw in q_lower for kw in ['報告號碼', '報告清��', '列出報告', 'report number', 'list of reports']):
             intents['action'] = 'list'
             sql_components['select'].append("jt.JobNo, jt.ReportAuthorization")
@@ -566,7 +568,6 @@ class TextToSQLSystem:
             sql_components['log_parts'].append("報告列表")
     
         # 偵測器 2.2: 時間意圖
-        # (此處邏輯不變)
         year_match = re.search(r'(\d{4})\s*年?', question)
         month_match = re.search(r'(\d{1,2})\s*月', question)
         if year_match:
@@ -577,9 +578,8 @@ class TextToSQLSystem:
             month = month_match.group(1).zfill(2)
             sql_components['where'].append(f"strftime('%m', jt.ReportAuthorization) = '{month}'")
             sql_components['log_parts'].append(f"{month}月")
-    
+            
         # 偵測器 2.3: 實體意圖
-        # (此處邏輯不變)
         if entity_match_data:
             if "TSR53SampleDescription" not in " ".join(sql_components['joins']):
                  sql_components['joins'].append("JOIN TSR53SampleDescription AS sd ON jt.JobNo = sd.JobNo")
@@ -592,7 +592,6 @@ class TextToSQLSystem:
                 sql_components['select'].append("sd.BuyerName")
     
         # 偵測器 2.4: 評級意圖
-        # (此處邏輯不變)
         if 'fail' in q_lower or '失敗' in q_lower:
             if "TSR53SampleDescription" not in " ".join(sql_components['joins']):
                  sql_components['joins'].append("JOIN TSR53SampleDescription AS sd ON jt.JobNo = sd.JobNo")
@@ -604,43 +603,23 @@ class TextToSQLSystem:
             sql_components['where'].append("sd.OverallRating = 'Pass'")
             sql_components['log_parts'].append("Pass")
     
-        # *** V28 修正 ***
         # 偵測器 2.5: 實驗組 (LabGroup) 意圖 (帶有別名映射)
-        lab_group_mapping = {
-            'A': 'TA',
-            'B': 'TB',
-            'C': 'TC',
-            'D': 'TD',
-            'E': 'TE',
-            'Y': 'TY',
-            'GCI': 'GCI',
-            'GCO': 'GCO'
-            # ...可在此處繼續添加其他映射...
-        }
+        lab_group_mapping = {'A': 'TA', 'B': 'TB', 'C': 'TC', 'D': 'TD', 'E': 'TE', 'Y': 'TY'}
         lab_group_match = re.search(r'([A-Z]{1,2})組', question, re.IGNORECASE)
         if lab_group_match:
             user_input_group = lab_group_match.group(1).upper()
-            # 從映射表中查找真實值，如果找不到，則使用用戶的原始輸入作為備用
             db_lab_group = lab_group_mapping.get(user_input_group, user_input_group)
-            
-            # 測試項目數量應從 JobItemsInProgress 計算
             sql_components['joins'].append("JOIN JobItemsInProgress AS jip ON jt.JobNo = jip.JobNo")
             sql_components['where'].append(f"jip.LabGroup = '{db_lab_group}'")
             sql_components['log_parts'].append(f"{user_input_group}組(->{db_lab_group})")
-            
-            # 如果是計數，目標應該是測試項目而不是報告
-            if intents.get('action') == 'count' and "測試項目" in question:
-                sql_components['select'] = ["COUNT(jip.ItemCode) AS item_count"] # 覆蓋掉原有的 SELECT
-                sql_components['log_parts'][0] = "測試項目總數" # 更新日誌
     
         # --- 3. 判斷是否觸發了模板，並動態組合 SQL ---
         if 'action' in intents:
-            # 動態決定主表和必要的預設條件
             sql_components['from'] = "FROM JobTimeline AS jt"
-            if year_match or month_match or entity_match_data or ('fail' in q_lower or '失敗' in q_lower) or ('pass' in q_lower or '通過' in q_lower):
+            # 只要有任何篩選條件，就加上報告已授權的基礎限制
+            if sql_components['where']:
                  sql_components['where'].insert(0, "jt.ReportAuthorization IS NOT NULL")
             
-            # 組合所有SQL組件
             select_clause = "SELECT " + ", ".join(sorted(list(set(sql_components['select']))))
             from_clause = sql_components['from']
             joins_clause = " ".join(sql_components['joins'])
@@ -658,7 +637,6 @@ class TextToSQLSystem:
         # ==============================================================================
         self._log("未觸發任何模板，嘗試解析並修正 AI 輸出...", "INFO")
         
-        # (此處的 Fallback 邏輯保持不變)
         parsed_sql = parse_sql_from_response(raw_response)
         if not parsed_sql:
             self._log(f"❌ 未能從模型回應中解析出任何 SQL。原始回應: {raw_response}", "ERROR")