Update command_handler.py

command_handler.py

@@ -1,4 +1,4 @@
-# command_handler.py (Corrected and Modified Version)
+# command_handler.py (已整合 PWS 查詢功能)
 import pandas as pd
 from linebot.v3.messaging import TextMessage, ImageMessage
 
@@ -7,10 +7,11 @@ from cwa_service import fetch_cwa_alarm_list, fetch_significant_earthquakes, fet
 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
-# [修正] 將 HF_SPACE_URL 改為 MCP_SERVER_URL
+from pws_service import fetch_latest_pws_info # 匯入新的 PWS 服務
 from config import CURRENT_YEAR, MCP_SERVER_URL
 
 def get_help_message() -> TextMessage:
+    """產生並回傳包含所有指令的說明文字。"""
     text = (
         "📖 指令列表 (輸入數字即可)\n\n"
         "【地震資訊】\n"
@@ -24,11 +25,14 @@ def get_help_message() -> TextMessage:
         "• 7 <問題> - 與 AI 助理對話\n\n"
         "【基本指令】\n"
         "• 8 - 關於此機器人\n"
-        "• 9 - 顯示此說明"
+        "• 9 - 顯示此說明\n"
+        "【天氣與警報】\n"
+        "• 10 - 最新 PWS 發布情形\n\n"
     )
     return TextMessage(text=text)
 
 def get_info_message() -> TextMessage:
+    """產生並回傳關於此機器人的資訊。"""
     text = (
         "🤖 關於我\n\n"
         "我是一個多功能助理機器人，提供地震查詢與 AI 對話功能。\n\n"
@@ -39,6 +43,7 @@ def get_info_message() -> TextMessage:
     return TextMessage(text=text)
 
 def get_taiwan_earthquake_list() -> TextMessage:
+    """從 USGS 取得今年台灣的地震列表並格式化回覆。"""
     result = fetch_taiwan_df_this_year()
     if isinstance(result, pd.DataFrame):
         count = len(result)
@@ -58,6 +63,7 @@ def get_taiwan_earthquake_list() -> TextMessage:
     return TextMessage(text=reply_text)
 
 def get_latest_earthquake_reply() -> list:
+    """取得 CWA 最新一筆地震，並組合文字與圖片訊息。"""
     try:
         latest_eq = fetch_latest_significant_earthquake()
         if not latest_eq:
@@ -87,12 +93,18 @@ def get_latest_earthquake_reply() -> list:
         return [TextMessage(text=f"❌ 查詢最新地震失敗：{e}")]
 
 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',
+        '10': '/pws', # 新增 PWS 指令
         '地震': '/global', 'quake': '/global', '幫助': '/help',
         '台灣地震': '/taiwan', '臺灣地震': '/taiwan',
         '台灣地震畫圖': '/map', '臺灣地震畫圖': '/map',
@@ -102,6 +114,7 @@ def process_message(user_message_raw: str, request_base_url: str) -> list:
     command = ""
     arg = ""
     
+    # 解析使用者輸入以分離指令和參數
     parts = user_message.split(' ', 1)
     cmd_key = parts[0].lower()
 
@@ -109,25 +122,27 @@ def process_message(user_message_raw: str, request_base_url: str) -> list:
         command = cmd_map[cmd_key]
         if len(parts) > 1:
             arg = parts[1].strip()
-    # Allow users to still use /command format
+    # 允許使用者仍然可以使用 /command 格式
     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()]
         if command == '/info': return [get_info_message()]
         if command == '/latest': return get_latest_earthquake_reply()
         if command == '/global': return [TextMessage(text=fetch_global_last24h_text())]
         if command == '/taiwan': return [get_taiwan_earthquake_list()]
-        # [修正] 將 HF_SPACE_URL 改為 MCP_SERVER_URL
         if command == '/map': return [TextMessage(text=f"🗺️ 外部地震查詢服務\n\n請點擊以下連結：\n{MCP_SERVER_URL}")]
         if command == '/alert': return [TextMessage(text=fetch_cwa_alarm_list(limit=5))]
         if command == '/significant': return [TextMessage(text=fetch_significant_earthquakes(limit=5))]
+        if command == '/pws': return [TextMessage(text=fetch_latest_pws_info())] # 新增 PWS 處理邏輯
         if command == '/ai':
             prompt = arg
             if not prompt: return [TextMessage(text="請輸入問題，例如：7 台灣最高的山是哪座？")]
             return [TextMessage(text=generate_ai_text(prompt))]
 
-    return [TextMessage(text=generate_ai_text(user_message))]
+    # 如果輸入的不是已知指令，則交由 AI 處理
+    return [TextMessage(text=generate_ai_text(user_message))]