Update app.py

app.py

@@ -494,12 +494,13 @@ class TextToSQLSystem:
 
         return formatted.strip()
 
-# 在 class TextToSQLSystem 內
-
+    # 在 class TextToSQLSystem 內
+    
     def _validate_and_fix_sql(self, question: str, raw_response: str) -> Tuple[Optional[str], str]:
         """
-        (V26 / 統一實體識別 + 模組化意圖引擎版)
-        在模組化意圖偵測前，先運行一個帶優先級的統一實體識別引擎。
+        (V27 / 統一實體識別 + 模組化意圖引擎 + 實驗組偵測 最終版)
+        一個多層次的SQL生成引擎。它優先使用基於規則的動態模板生成器，
+        如果無法匹配，則回退到解析和修正AI模型的輸出。
         """
         q_lower = question.lower()
     
@@ -507,61 +508,51 @@ class TextToSQLSystem:
         #  第零層：統一實體識別引擎 (Unified Entity Recognition Engine)
         # ==============================================================================
         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:
             match = re.search(p['pattern'], question, re.IGNORECASE)
             if match:
-                # 根據正則表達式捕獲組的數量來決定取哪個��
                 entity_value = match.group(2) if len(match.groups()) > 1 else match.group(1)
                 entity_match_data = {
                     "type": p['type'], 
-                    "name": entity_value.strip().upper(), # 標準化：移除前後空格並轉大寫
+                    "name": entity_value.strip().upper(),
                     "column": p['column']
                 }
-                # 找到第一個匹配項（最高優先級）後就立刻停止
                 break
                 
         # ==============================================================================
-        #  第一層：模組化意圖偵測 (Modular Intent Detection)
+        #  第一層：模組化意圖偵測與動態SQL組合
         # ==============================================================================
         
-        # --- 1. 初始化 SQL 組件 ---
         intents = {}
         sql_components = {
             'select': [],
-            'from': "FROM JobTimeline AS jt",
+            'from': "",
             'joins': [],
-            'where': ["jt.ReportAuthorization IS NOT NULL"],
+            'where': [],
             'group_by': [],
             'order_by': [],
             'log_parts': []
         }
     
-        # --- 2. 運行一系列獨立的意圖偵測器 ---
+        # --- 運行一系列獨立的意圖偵測器 ---
         
         # 偵測器 2.1: 核心動作意圖 (計數 vs. 列表)
         if any(kw in q_lower for kw in ['幾份', '多少', '數量', '總數', 'how many', 'count']):
@@ -586,20 +577,15 @@ class TextToSQLSystem:
             sql_components['where'].append(f"strftime('%m', jt.ReportAuthorization) = '{month}'")
             sql_components['log_parts'].append(f"{month}月")
     
-        # 偵測器 2.3: 實體意圖 (現在使用預處理的結果)
+        # 偵測器 2.3: 實體意圖 (使用預處理的結果)
         if entity_match_data:
-            # 確保 JOIN 只添加一次
             if "TSR53SampleDescription" not in " ".join(sql_components['joins']):
                  sql_components['joins'].append("JOIN TSR53SampleDescription AS sd ON jt.JobNo = sd.JobNo")
-            
             entity_name, column_name = entity_match_data["name"], entity_match_data["column"]
             match_operator = "=" if column_name.endswith("ID") else "LIKE"
-            # 對於 LIKE 查詢，在值兩邊加上 %
             entity_value = f"'%{entity_name}%'" if match_operator == "LIKE" else f"'{entity_name}'"
             sql_components['where'].append(f"{column_name} {match_operator} {entity_value}")
             sql_components['log_parts'].append(entity_match_data["type"] + ":" + entity_name)
-            
-            # 如果是列表查詢，我們可以顯示更多實體相關的資訊
             if intents.get('action') == 'list':
                 sql_components['select'].append("sd.BuyerName")
     
@@ -614,17 +600,34 @@ class TextToSQLSystem:
                  sql_components['joins'].append("JOIN TSR53SampleDescription AS sd ON jt.JobNo = sd.JobNo")
             sql_components['where'].append("sd.OverallRating = 'Pass'")
             sql_components['log_parts'].append("Pass")
-            
-        # --- 3. 判斷是否觸發了模板，並組合 SQL ---
-        
+    
+        # 偵測器 2.5: 實驗組 (LabGroup) 意圖
+        lab_group_match = re.search(r'([A-Z]{1,2})組', question, re.IGNORECASE)
+        if lab_group_match:
+            lab_group = lab_group_match.group(1).upper()
+            # 測試項目數量應從 JobItemsInProgress 計算
+            sql_components['joins'].append("JOIN JobItemsInProgress AS jip ON jt.JobNo = jip.JobNo")
+            sql_components['where'].append(f"jip.LabGroup = '{lab_group}'")
+            sql_components['log_parts'].append(f"{lab_group}組")
+            # 如果是計數，目標應該是測試項目而不是報告
+            if intents.get('action') == 'count':
+                sql_components['select'] = ["COUNT(jip.ItemCode) AS item_count"] # 覆蓋掉原有的 SELECT
+                sql_components['log_parts'][0] = "測試項目總數" # 更新日誌
+    
+        # --- 3. 判斷是否觸發了模板，並動態組合 SQL ---
         if 'action' in intents:
-            final_select = "SELECT " + ", ".join(sorted(list(set(sql_components['select']))))
+            # 動態決定主表和必要的預設條件
+            sql_components['from'] = "FROM JobTimeline AS jt"
+            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'])
-            where_clause = "WHERE " + " AND ".join(sql_components['where'])
+            where_clause = "WHERE " + " AND ".join(sql_components['where']) if sql_components['where'] else ""
             orderby_clause = "ORDER BY " + ", ".join(sql_components['order_by']) if sql_components['order_by'] else ""
             
-            template_sql = f"{final_select} {from_clause} {joins_clause} {where_clause} {orderby_clause};"
+            template_sql = f"{select_clause} {from_clause} {joins_clause} {where_clause} {orderby_clause};"
             
             query_log = " ".join(sql_components['log_parts'])
             self._log(f"🔄 偵測到組合意圖【{query_log}】，啟用動態模板。", "INFO")