Update core/callbacks.py

core/callbacks.py

@@ -5,7 +5,7 @@ import traceback
 from datetime import datetime
 import pytz
 import difflib
-import re # ✨ 1. 匯入 re 模組
+import re
 
 from services import cwa_service, news_service, pws_service, usgs_service
 from core.visits import get_current_visit_count
@@ -53,13 +53,20 @@ def execute_user_code(code_string, source_lab):
             notification_text += f"\n錯誤類型: {error_type}"
         send_line_notification_in_background(notification_text)
 
+# --- ✨ 重新排序此列表，使其與 UI 上的編號 (#1, #2...) 對應 ---
+# 這個列表同時用於快捷指令和關鍵字查詢
 LIVE_TOOLS = [
+    # #1
     {"keywords": ["新聞", "今日新聞", "news"], "function": news_service.fetch_today_news, "name": "今日新聞"},
+    # #2
     {"keywords": ["cwa地震", "顯著地震", "有感地震"], "function": cwa_service.fetch_significant_earthquakes, "name": "CWA 顯著有感地震"},
-    # --- ✨ 在這裡新增了 "eew" 和 "現在有地震預警嗎？" ---
+    # #3
     {"keywords": ["地震預警", "cwa alarm", "eew", "現在有地震預警嗎？"], "function": cwa_service.fetch_cwa_alarm_list, "name": "CWA 地震預警"},
+    # #4
     {"keywords": ["全球地震", "usgs", "最近全球有哪些大地震"], "function": usgs_service.fetch_global_last24h_text, "name": "全球顯著地震"},
+    # #5
     {"keywords": ["pws發布", "pws info"], "function": pws_service.fetch_latest_pws_info, "name": "PWS 發布情形"},
+    # #6
     {"keywords": ["pws地震", "pws alert", "pws", "最新的 pws 地震警報"], "function": pws_service.fetch_cwa_pws_earthquake_info, "name": "PWS 地震警報"},
 ]
 
@@ -83,11 +90,11 @@ def find_best_match_from_kb(user_input, knowledge_base, threshold=0.6):
     else:
         return best_answer
 
-# --- ✨ 2. 這是修改後的核心函式 ---
+# --- ✨ 這是修改後的核心函式 ---
 def ai_chatbot_with_kb(message, history):
     """
     處理聊天機器人互動的主函式。
-    優先檢查進階查詢格式，其次檢查即時工具，最後查詢靜態知識庫。
+    優先處理快捷指令，其次是進階查詢，再其次是關鍵字工具，最後查詢靜態知識庫。
     """
     # (通知邏輯維持不變)
     tz = pytz.timezone('Asia/Taipei')
@@ -101,10 +108,23 @@ def ai_chatbot_with_kb(message, history):
     )
     send_line_notification_in_background(notification_text)
 
-    user_message = message.strip() # 保留大小寫以符合可能的未來需求，但比對時再轉小寫
+    user_message = message.strip()
 
-    # 步驟 1: 檢查是否為進階地震查詢格式
-    # 使用正規表示式來匹配 "查詢 YYYY-MM-DD 到 YYYY-MM-DD 規模 X.X 以上地震"
+    # 步驟 1: 檢查是否為快捷指令 (例如 #1, #2)
+    if user_message.startswith('#'):
+        try:
+            # 取得 # 後面的數字，並轉換成索引 (數字1對應索引0)
+            tool_index = int(user_message[1:]) - 1
+            if 0 <= tool_index < len(LIVE_TOOLS):
+                tool = LIVE_TOOLS[tool_index]
+                print(f"🚀 觸發快捷指令：{tool['name']}")
+                return tool["function"]()
+            else:
+                return f"⚠️ 無效的快捷指令！請輸入 #1 到 #{len(LIVE_TOOLS)} 之間的數字。"
+        except (ValueError, IndexError):
+            return "⚠️ 快捷指令格式錯誤！請輸入像 `#1` 這樣的格式。"
+
+    # 步驟 2: 檢查是否為進階地震查詢格式
     pattern = r"查詢\s*(\d{4}-\d{2}-\d{2})\s*到\s*(\d{4}-\d{2}-\d{2})\s*規模\s*(\d+(?:\.\d+)?)\s*以上地震"
     match = re.search(pattern, user_message)
     
@@ -112,12 +132,11 @@ def ai_chatbot_with_kb(message, history):
         start_date, end_date, magnitude = match.groups()
         print(f"🔍 觸發進階地震查詢：{start_date} 到 {end_date}, M≥{magnitude}")
         try:
-            # 呼叫 usgs_service 中的新函式
             return usgs_service.fetch_usgs_earthquakes_by_date(start_date, end_date, float(magnitude))
         except Exception as e:
             return f"❌ 執行進階查詢時發生錯誤：{e}"
 
-    # 步驟 2: 如果不是進階查詢，則檢查是否觸發任何即時工具 (關鍵字比對)
+    # 步驟 3: 如果都不是，則檢查是否觸發即時工具 (關鍵字比對)
     user_message_lower = user_message.lower()
     for tool in LIVE_TOOLS:
         for keyword in tool["keywords"]:
@@ -128,5 +147,5 @@ def ai_chatbot_with_kb(message, history):
                 except Exception as e:
                     return f"❌ 執行「{tool['name']}」工具時發生錯誤：{e}"
 
-    # 步驟 3: 如果都沒有觸發，則查詢靜態知識庫
+    # 步驟 4: 如果都沒有觸發，則查詢靜態知識庫
     return find_best_match_from_kb(user_message_lower, KNOWLEDGE_BASE)