Update command_handler.py

command_handler.py

@@ -3,7 +3,7 @@ import pandas as pd
 from linebot.v3.messaging import TextMessage, ImageMessage
 
 # 匯入服務函式
-from cwa_service import fetch_cwa_alarm_list, fetch_significant_earthquakes, fetch_local_earthquakes, fetch_latest_significant_earthquake
+from cwa_service import fetch_cwa_alarm_list, fetch_significant_earthquakes, fetch_latest_significant_earthquake
 from usgs_service import fetch_global_last24h_text, fetch_taiwan_df_this_year
 from plotting_service import create_and_save_map
 from ai_service import generate_ai_text
@@ -12,20 +12,19 @@ from config import CURRENT_YEAR, HF_SPACE_URL
 def get_help_message() -> TextMessage:
     """回傳包含所有可用指令的說明訊息。"""
     text = (
-        "📖 指令列表\n\n"
+        "📖 指令列表 (輸入 #數字 即可)\n\n"
         "【地震資訊】\n"
-        "• /latest - 查詢最新一筆顯著地震(含圖)\n"
-        "• /global - 查詢全球近24小時顯著地震\n"
-        "• /taiwan - 顯示今年台灣顯著地震列表\n"
-        "• /map - 取得台灣顯著地震地圖連結\n"
-        "• /alert - 獲取 CWA 最新地震預警\n"
-        "• /significant - CWA 最新顯著有感地震\n"
-        "• /local <縣市> - 小區域地震 (可選填縣市)\n\n"
+        "• #1 - 最新顯著地震 (含圖)\n"
+        "• #2 - 全球近24小時顯著地震\n"
+        "• #3 - 今年台灣顯著地震列表\n"
+        "• #4 - 台灣顯著地震地圖連結\n"
+        "• #5 - CWA 最新地震預警\n"
+        "• #6 - CWA 最新顯著有感地震\n\n"
         "【AI 與工具】\n"
-        "• /ai <問題> - 與 AI 助理對話\n\n"
+        "• #7 <問題> - 與 AI 助理對話\n\n"
         "【基本指令】\n"
-        "• /info - 關於此機器人\n"
-        "• /help - 顯示此說明"
+        "• #8 - 關於此機器人\n"
+        "• #9 - 顯示此說明"
     )
     return TextMessage(text=text)
 
@@ -34,7 +33,7 @@ def get_info_message() -> TextMessage:
     text = (
         "🤖 關於我\n\n"
         "我是一個多功能助理機器人，提供地震查詢與 AI 對話功能。\n\n"
-        "• 版本: 2.5\n"
+        "• 版本: 3.0\n"
         "• 資料來源: CWA, USGS, Hugging Face\n"
         "• 開發者: dayichen"
     )
@@ -109,25 +108,46 @@ def process_message(user_message_raw: str, request_base_url: str) -> list:
     """處理使用者的文字訊息並回傳一個包含回覆訊息的列表。"""
     user_message = (user_message_raw or "").strip()
     
-    # 將舊指令對應到新指令，提供向下相容
-    if user_message.lower() in ["地震", "quake"]:
-        user_message = "/global"
-    elif user_message.lower() in ["台灣地震", "臺灣地震"]:
-        user_message = "/taiwan"
-    elif user_message.lower() in ["台灣地震畫圖", "臺灣地震畫圖"]:
-        user_message = "/map"
-    elif user_message.lower() == "地震預警":
-        user_message = "/alert"
+    # --- 指令對應 ---
+    cmd_map = {
+        '#1': '/latest', '#2': '/global', '#3': '/taiwan',
+        '#4': '/map', '#5': '/alert', '#6': '/significant',
+        '#7': '/ai', '#8': '/info', '#9': '/help',
+        '地震': '/global', 'quake': '/global',
+        '台灣地震': '/taiwan', '臺灣地震': '/taiwan',
+        '台灣地震畫圖': '/map', '臺灣地震畫圖': '/map',
+        '地震預警': '/alert',
+        '/help': '/help'
+    }
 
-    if user_message.startswith('/'):
+    command = ""
+    arg = ""
+    
+    # 檢查是否為數字指令 (e.g., #7 你好)
+    if user_message.startswith('#') and len(user_message) > 1:
+        parts = user_message.split(' ', 1)
+        key = parts[0]
+        if key in cmd_map:
+            command = cmd_map[key]
+            arg = parts[1] if len(parts) > 1 else ""
+    
+    # 檢查是否為斜線指令 (e.g., /ai 你好)
+    elif user_message.startswith('/'):
         parts = user_message.split(' ', 1)
         command = parts[0].lower()
         arg = parts[1] if len(parts) > 1 else ""
+        
+    # 檢查是否為中文關鍵字指令
+    else:
+        for key, value in cmd_map.items():
+            if user_message.lower() == key:
+                command = value
+                break
 
-        if command == '/help':
-            return [get_help_message()]
-        elif command == '/info':
-            return [get_info_message()]
+    # --- 指令處理 ---
+    if command:
+        if command == '/help' or command == '/info':
+            return [get_info_message()] if command == '/info' else [get_help_message()]
         elif command == '/latest':
             return get_latest_earthquake_reply()
         elif command == '/global':
@@ -140,12 +160,11 @@ def process_message(user_message_raw: str, request_base_url: str) -> list:
             return [TextMessage(text=fetch_cwa_alarm_list(limit=5))]
         elif command == '/significant':
             return [TextMessage(text=fetch_significant_earthquakes(limit=5))]
-        elif command == '/local':
-            return [TextMessage(text=fetch_local_earthquakes(area_name=arg))]
         elif command == '/ai':
-            if not arg:
-                return [TextMessage(text="請輸入問題，例如：/ai 台灣最高的山是哪座？")]
-            return [TextMessage(text=generate_ai_text(arg))]
+            prompt = arg or user_message[2:].lstrip() # 處理 '#7 ...' 和 '/ai ...'
+            if not prompt:
+                return [TextMessage(text="請輸入問題，例如：#7 台灣最高的山是哪座？")]
+            return [TextMessage(text=generate_ai_text(prompt))]
 
-    # 若非指令，則預設交給 AI 處理 (作為備援)
+    # 若非任何指令，則預設交給 AI 處理 (作為備援)
     return [TextMessage(text=generate_ai_text(user_message))]