Spaces:
Sleeping
Sleeping
Update command_handler.py
Browse files- command_handler.py +40 -54
command_handler.py
CHANGED
|
@@ -3,7 +3,7 @@ import pandas as pd
|
|
| 3 |
from linebot.v3.messaging import TextMessage, ImageMessage
|
| 4 |
|
| 5 |
# 匯入服務函式
|
| 6 |
-
from cwa_service import fetch_cwa_alarm_list, fetch_significant_earthquakes
|
| 7 |
from usgs_service import fetch_global_last24h_text, fetch_taiwan_df_this_year
|
| 8 |
from plotting_service import create_and_save_map
|
| 9 |
from ai_service import generate_ai_text
|
|
@@ -14,11 +14,12 @@ def get_help_message() -> TextMessage:
|
|
| 14 |
text = (
|
| 15 |
"📖 指令列表\n\n"
|
| 16 |
"【地震資訊】\n"
|
|
|
|
| 17 |
"• /global - 查詢全球近24小時顯著地震\n"
|
| 18 |
"• /taiwan - 顯示今年台灣顯著地震列表\n"
|
| 19 |
"• /map - 取得台灣顯著地震地圖連結\n"
|
| 20 |
"• /alert - 獲取 CWA 最新地震預警\n"
|
| 21 |
-
"• /significant - CWA 最新顯著有感地震\n\n"
|
| 22 |
"【AI 與工具】\n"
|
| 23 |
"• /ai <問題> - 與 AI 助理對話\n\n"
|
| 24 |
"【基本指令】\n"
|
|
@@ -27,65 +28,48 @@ def get_help_message() -> TextMessage:
|
|
| 27 |
)
|
| 28 |
return TextMessage(text=text)
|
| 29 |
|
| 30 |
-
|
| 31 |
-
"""回傳機器人資訊。"""
|
| 32 |
-
text = (
|
| 33 |
-
"🤖 關於我\n\n"
|
| 34 |
-
"我是一個多功能助理機器人,提供地震查詢與 AI 對話功能。\n\n"
|
| 35 |
-
"• 版本: 2.4\n"
|
| 36 |
-
"• 資料來源: CWA, USGS, Hugging Face\n"
|
| 37 |
-
"• 開發者: dayichen"
|
| 38 |
-
)
|
| 39 |
-
return TextMessage(text=text)
|
| 40 |
|
| 41 |
-
|
| 42 |
-
|
| 43 |
-
|
| 44 |
-
|
| 45 |
-
|
| 46 |
-
|
| 47 |
-
|
| 48 |
-
t = row["time_utc"].strftime("%Y-%m-%d %H:%M")
|
| 49 |
-
lines.append(
|
| 50 |
-
f"震級: {row['magnitude']:.1f} | 日期時間: {t} (UTC)\n"
|
| 51 |
-
f"地點: {row['place']}\n"
|
| 52 |
-
f"報告連結: {row.get('url', '無')}"
|
| 53 |
-
)
|
| 54 |
-
if count > 15:
|
| 55 |
-
lines.append(f"... (還有 {count - 15} 筆,請用 /map 查看全部)")
|
| 56 |
-
reply_text = "\n\n".join(lines)
|
| 57 |
-
else:
|
| 58 |
-
reply_text = result
|
| 59 |
-
return TextMessage(text=reply_text)
|
| 60 |
|
| 61 |
-
|
| 62 |
-
|
| 63 |
-
|
| 64 |
-
|
| 65 |
-
|
| 66 |
-
|
| 67 |
-
|
| 68 |
-
"
|
| 69 |
-
"
|
| 70 |
-
f"{
|
|
|
|
| 71 |
)
|
| 72 |
-
|
| 73 |
-
|
| 74 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 75 |
|
| 76 |
def process_message(user_message_raw: str, request_base_url: str) -> list:
|
| 77 |
"""處理使用者的文字訊息並回傳一個包含回覆訊息的列表。"""
|
| 78 |
user_message = (user_message_raw or "").strip()
|
| 79 |
|
| 80 |
-
#
|
| 81 |
-
if user_message.lower() in ["地震", "quake"]:
|
| 82 |
-
user_message = "/global"
|
| 83 |
-
elif user_message.lower() in ["台灣地震", "臺灣地震"]:
|
| 84 |
-
user_message = "/taiwan"
|
| 85 |
-
elif user_message.lower() in ["台灣地震畫圖", "臺灣地震畫圖"]:
|
| 86 |
-
user_message = "/map"
|
| 87 |
-
elif user_message.lower() == "地震預警":
|
| 88 |
-
user_message = "/alert"
|
| 89 |
|
| 90 |
if user_message.startswith('/'):
|
| 91 |
parts = user_message.split(' ', 1)
|
|
@@ -96,6 +80,8 @@ def process_message(user_message_raw: str, request_base_url: str) -> list:
|
|
| 96 |
return [get_help_message()]
|
| 97 |
elif command == '/info':
|
| 98 |
return [get_info_message()]
|
|
|
|
|
|
|
| 99 |
elif command == '/global':
|
| 100 |
return [TextMessage(text=fetch_global_last24h_text())]
|
| 101 |
elif command == '/taiwan':
|
|
@@ -106,7 +92,6 @@ def process_message(user_message_raw: str, request_base_url: str) -> list:
|
|
| 106 |
return [TextMessage(text=fetch_cwa_alarm_list(limit=5))]
|
| 107 |
elif command == '/significant':
|
| 108 |
return [TextMessage(text=fetch_significant_earthquakes(limit=5))]
|
| 109 |
-
# [移除] elif command == '/local': ...
|
| 110 |
elif command == '/ai':
|
| 111 |
if not arg:
|
| 112 |
return [TextMessage(text="請輸入問題,例如:/ai 台灣最高的山是哪座?")]
|
|
@@ -114,3 +99,4 @@ def process_message(user_message_raw: str, request_base_url: str) -> list:
|
|
| 114 |
|
| 115 |
# 若非指令,則預設交給 AI 處理 (作為備援)
|
| 116 |
return [TextMessage(text=generate_ai_text(user_message))]
|
|
|
|
|
|
| 3 |
from linebot.v3.messaging import TextMessage, ImageMessage
|
| 4 |
|
| 5 |
# 匯入服務函式
|
| 6 |
+
from cwa_service import fetch_cwa_alarm_list, fetch_significant_earthquakes, fetch_latest_significant_earthquake # [新增]
|
| 7 |
from usgs_service import fetch_global_last24h_text, fetch_taiwan_df_this_year
|
| 8 |
from plotting_service import create_and_save_map
|
| 9 |
from ai_service import generate_ai_text
|
|
|
|
| 14 |
text = (
|
| 15 |
"📖 指令列表\n\n"
|
| 16 |
"【地震資訊】\n"
|
| 17 |
+
"• /latest - 查詢最新一筆顯著地震(含圖)\n" # [新增]
|
| 18 |
"• /global - 查詢全球近24小時顯著地震\n"
|
| 19 |
"• /taiwan - 顯示今年台灣顯著地震列表\n"
|
| 20 |
"• /map - 取得台灣顯著地震地圖連結\n"
|
| 21 |
"• /alert - 獲取 CWA 最新地震預警\n"
|
| 22 |
+
"• /significant - CWA 最新顯著有感地震\n\n"
|
| 23 |
"【AI 與工具】\n"
|
| 24 |
"• /ai <問題> - 與 AI 助理對話\n\n"
|
| 25 |
"【基本指令】\n"
|
|
|
|
| 28 |
)
|
| 29 |
return TextMessage(text=text)
|
| 30 |
|
| 31 |
+
# ... (get_info_message, get_taiwan_earthquake_list, get_taiwan_earthquake_map 函式不變) ...
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 32 |
|
| 33 |
+
# --- [新功能] 處理 /latest 指令的回覆 ---
|
| 34 |
+
def get_latest_earthquake_reply() -> list:
|
| 35 |
+
"""獲取最新地震資訊並組合成 LINE 訊息"""
|
| 36 |
+
try:
|
| 37 |
+
latest_eq = fetch_latest_significant_earthquake()
|
| 38 |
+
if not latest_eq:
|
| 39 |
+
return [TextMessage(text="✅ 近期無顯著有感地震報告。")]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 40 |
|
| 41 |
+
# 組合文字訊息
|
| 42 |
+
mag_str = f"{latest_eq['Magnitude']:.1f}" if latest_eq.get('Magnitude') is not None else "—"
|
| 43 |
+
depth_str = f"{latest_eq['Depth']:.0f}" if latest_eq.get('Depth') is not None else "—"
|
| 44 |
+
|
| 45 |
+
text_message_content = (
|
| 46 |
+
f"🚨 CWA 最新顯著有感地震\n"
|
| 47 |
+
f"----------------------------------\n"
|
| 48 |
+
f"時間: {latest_eq.get('TimeStr', '—')}\n"
|
| 49 |
+
f"地點: {latest_eq.get('Location', '—')}\n"
|
| 50 |
+
f"規模: M{mag_str} | 深度: {depth_str} km\n"
|
| 51 |
+
f"報告: {latest_eq.get('URL', '無')}"
|
| 52 |
)
|
| 53 |
+
reply_messages = [TextMessage(text=text_message_content)]
|
| 54 |
+
|
| 55 |
+
# 如果有圖片網址,則新增圖片訊息
|
| 56 |
+
if latest_eq.get("ImageURL"):
|
| 57 |
+
image_url = latest_eq["ImageURL"]
|
| 58 |
+
reply_messages.append(
|
| 59 |
+
ImageMessage(original_content_url=image_url, preview_image_url=image_url)
|
| 60 |
+
)
|
| 61 |
+
|
| 62 |
+
return reply_messages
|
| 63 |
+
|
| 64 |
+
except Exception as e:
|
| 65 |
+
return [TextMessage(text=f"❌ 查詢最新地震失敗:{e}")]
|
| 66 |
+
|
| 67 |
|
| 68 |
def process_message(user_message_raw: str, request_base_url: str) -> list:
|
| 69 |
"""處理使用者的文字訊息並回傳一個包含回覆訊息的列表。"""
|
| 70 |
user_message = (user_message_raw or "").strip()
|
| 71 |
|
| 72 |
+
# ... (舊指令相容性轉換不變) ...
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 73 |
|
| 74 |
if user_message.startswith('/'):
|
| 75 |
parts = user_message.split(' ', 1)
|
|
|
|
| 80 |
return [get_help_message()]
|
| 81 |
elif command == '/info':
|
| 82 |
return [get_info_message()]
|
| 83 |
+
elif command == '/latest': # [新增]
|
| 84 |
+
return get_latest_earthquake_reply()
|
| 85 |
elif command == '/global':
|
| 86 |
return [TextMessage(text=fetch_global_last24h_text())]
|
| 87 |
elif command == '/taiwan':
|
|
|
|
| 92 |
return [TextMessage(text=fetch_cwa_alarm_list(limit=5))]
|
| 93 |
elif command == '/significant':
|
| 94 |
return [TextMessage(text=fetch_significant_earthquakes(limit=5))]
|
|
|
|
| 95 |
elif command == '/ai':
|
| 96 |
if not arg:
|
| 97 |
return [TextMessage(text="請輸入問題,例如:/ai 台灣最高的山是哪座?")]
|
|
|
|
| 99 |
|
| 100 |
# 若非指令,則預設交給 AI 處理 (作為備援)
|
| 101 |
return [TextMessage(text=generate_ai_text(user_message))]
|
| 102 |
+
|