Update command_handler.py

command_handler.py

@@ -1,4 +1,4 @@
-# command_handler.py (Corrected Version)
+# command_handler.py (Corrected and Modified Version)
 import pandas as pd
 from linebot.v3.messaging import TextMessage, ImageMessage
 
@@ -12,19 +12,19 @@ from config import CURRENT_YEAR, MCP_SERVER_URL
 
 def get_help_message() -> TextMessage:
     text = (
-        "📖 指令列表 (輸入 #數字 即可)\n\n"
+        "📖 指令列表 (輸入數字即可)\n\n"
         "【地震資訊】\n"
-        "• #1 - 最新一筆顯著地震 (含圖)\n"
-        "• #2 - 全球近24小時顯著地震（USGS)\n"
-        "• #3 - 今年台灣顯著地震列表（USGS)\n"
-        "• #4 - CWA 地震目錄查詢 (外部連結)\n"
-        "• #5 - CWA 最新地震預警\n"
-        "• #6 - CWA 最近7天顯著有感地震\n\n"
+        "• 1 - 最新一筆顯著地震 (含圖)\n"
+        "• 2 - 全球近24小時顯著地震（USGS)\n"
+        "• 3 - 今年台灣顯著地震列表（USGS)\n"
+        "• 4 - CWA 地震目錄查詢 (外部連結)\n"
+        "• 5 - CWA 最新地震預警\n"
+        "• 6 - CWA 最近7天顯著有感地震\n\n"
         "【AI 與工具】\n"
-        "• #7 <問題> - 與 AI 助理對話\n\n"
+        "• 7 <問題> - 與 AI 助理對話\n\n"
         "【基本指令】\n"
-        "• #8 - 關於此機器人\n"
-        "• #9 - 顯示此說明"
+        "• 8 - 關於此機器人\n"
+        "• 9 - 顯示此說明"
     )
     return TextMessage(text=text)
 
@@ -32,7 +32,7 @@ def get_info_message() -> TextMessage:
     text = (
         "🤖 關於我\n\n"
         "我是一個多功能助理機器人，提供地震查詢與 AI 對話功能。\n\n"
-        "• 版本: 5.0 (Gemini Edition) 搭配搜尋地震目錄的MCP功能（可用資料 : 1973-01-01 至 2025-07-06）\n"
+        "• 版本: 5.0 (Gemini Edition) 搭配搜尋地震目錄的MCP功能（可用資料 : 1973-01-01 至 2025-07-06）\n"
         "• 資料來源: CWA, USGS, Google Gemini\n"
         "• 開發者: dayichen"
     )
@@ -90,9 +90,9 @@ def process_message(user_message_raw: str, request_base_url: str) -> list:
     user_message = (user_message_raw or "").strip()
     
     cmd_map = {
-        '#1': '/latest', '#2': '/global', '#3': '/taiwan',
-        '#4': '/map', '#5': '/alert', '#6': '/significant',
-        '#7': '/ai', '#8': '/info', '#9': '/help',
+        '1': '/latest', '2': '/global', '3': '/taiwan',
+        '4': '/map', '5': '/alert', '6': '/significant',
+        '7': '/ai', '8': '/info', '9': '/help',
         '地震': '/global', 'quake': '/global', '幫助': '/help',
         '台灣地震': '/taiwan', '臺灣地震': '/taiwan',
         '台灣地震畫圖': '/map', '臺灣地震畫圖': '/map',
@@ -102,14 +102,18 @@ def process_message(user_message_raw: str, request_base_url: str) -> list:
     command = ""
     arg = ""
     
-    if user_message.lower() in cmd_map:
-        command = cmd_map[user_message.lower()]
-    elif user_message.startswith(('#', '/')):
-        parts = user_message.split(' ', 1)
-        key = parts[0]
-        arg = parts[1].strip() if len(parts) > 1 else ""
-        if key in cmd_map:
-            command = cmd_map[key]
+    parts = user_message.split(' ', 1)
+    cmd_key = parts[0].lower()
+
+    if cmd_key in cmd_map:
+        command = cmd_map[cmd_key]
+        if len(parts) > 1:
+            arg = parts[1].strip()
+    # Allow users to still use /command format
+    elif user_message.startswith('/') and cmd_key in cmd_map.values():
+        command = cmd_key
+        if len(parts) > 1:
+            arg = parts[1].strip()
 
     if command:
         if command == '/help': return [get_help_message()]
@@ -123,7 +127,7 @@ def process_message(user_message_raw: str, request_base_url: str) -> list:
         if command == '/significant': return [TextMessage(text=fetch_significant_earthquakes(limit=5))]
         if command == '/ai':
             prompt = arg
-            if not prompt: return [TextMessage(text="請輸入問題，例如：#7 台灣最高的山是哪座？")]
+            if not prompt: return [TextMessage(text="請輸入問題，例如：7 台灣最高的山是哪座？")]
             return [TextMessage(text=generate_ai_text(prompt))]
 
     return [TextMessage(text=generate_ai_text(user_message))]