Update app.py

app.py

@@ -495,12 +495,11 @@ class TextToSQLSystem:
         return formatted.strip()
 
     # 在 class TextToSQLSystem 內
-    
+
     def _validate_and_fix_sql(self, question: str, raw_response: str) -> Tuple[Optional[str], str]:
         """
-        (V27 / 統一實體識別 + 模組化意圖引擎 + 實驗組偵測 最終版)
-        一個多層次的SQL生成引擎。它優先使用基於規則的動態模板生成器，
-        如果無法匹配，則回退到解析和修正AI模型的輸出。
+        (V28 / 實驗組別名映射版)
+        增加了實驗組別名到資料庫真實值的映射，以處理 "A組" -> "TA" 這類轉換。
         """
         q_lower = question.lower()
     
@@ -508,7 +507,7 @@ class TextToSQLSystem:
         #  第零層：統一實體識別引擎 (Unified Entity Recognition Engine)
         # ==============================================================================
         entity_match_data = None
-        # 包含了繁簡體兼容的關鍵字
+        # (您的 entity_patterns 邏輯保持不變，此處省略以保持簡潔)
         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'},
@@ -554,7 +553,8 @@ class TextToSQLSystem:
     
         # --- 運行一系列獨立的意圖偵測器 ---
         
-        # 偵測器 2.1: 核心動作意圖 (計數 vs. 列表)
+        # 偵測器 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")
@@ -566,6 +566,7 @@ 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,7 +578,8 @@ 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:
             if "TSR53SampleDescription" not in " ".join(sql_components['joins']):
                  sql_components['joins'].append("JOIN TSR53SampleDescription AS sd ON jt.JobNo = sd.JobNo")
@@ -589,7 +591,8 @@ class TextToSQLSystem:
             if intents.get('action') == 'list':
                 sql_components['select'].append("sd.BuyerName")
     
-        # 偵測器 2.4: 評級意圖 (Pass/Fail)
+        # 偵測器 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")
@@ -601,16 +604,32 @@ class TextToSQLSystem:
             sql_components['where'].append("sd.OverallRating = 'Pass'")
             sql_components['log_parts'].append("Pass")
     
-        # 偵測器 2.5: 實驗組 (LabGroup) 意圖
+        # *** 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_match = re.search(r'([A-Z]{1,2})組', question, re.IGNORECASE)
         if lab_group_match:
-            lab_group = lab_group_match.group(1).upper()
+            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 = '{lab_group}'")
-            sql_components['log_parts'].append(f"{lab_group}組")
+            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':
+            if intents.get('action') == 'count' and "測試項目" in question:
                 sql_components['select'] = ["COUNT(jip.ItemCode) AS item_count"] # 覆蓋掉原有的 SELECT
                 sql_components['log_parts'][0] = "測試項目總數" # 更新日誌
     
@@ -618,7 +637,8 @@ class TextToSQLSystem:
         if 'action' in intents:
             # 動態決定主表和必要的預設條件
             sql_components['from'] = "FROM JobTimeline AS jt"
-            sql_components['where'].insert(0, "jt.ReportAuthorization IS NOT NULL")
+            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):
+                 sql_components['where'].insert(0, "jt.ReportAuthorization IS NOT NULL")
             
             # 組合所有SQL組件
             select_clause = "SELECT " + ", ".join(sorted(list(set(sql_components['select']))))
@@ -638,6 +658,7 @@ 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")