cwadayi commited on
Commit
f1e04a3
·
verified ·
1 Parent(s): 4804f0f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +126 -126
app.py CHANGED
@@ -1,127 +1,127 @@
1
- # app.py
2
- # 首先匯入 config,以設定環境變數
3
- import config
4
-
5
- from flask import Flask, request, abort, send_from_directory
6
- from linebot.v3 import WebhookHandler
7
- from linebot.v3.exceptions import InvalidSignatureError
8
- from linebot.v3.messaging import (
9
- Configuration, ApiClient, MessagingApi,
10
- ReplyMessageRequest
11
- )
12
- from linebot.v3.webhooks import MessageEvent, TextMessageContent
13
-
14
- # 匯入指令處理器
15
- from command_handler import process_message
16
-
17
- # ------------------------------------------------------------------------------
18
- # Flask & LINE Bot 設定
19
- # ------------------------------------------------------------------------------
20
- app = Flask(__name__)
21
- line_config = Configuration(access_token=config.CHANNEL_ACCESS_TOKEN)
22
- handler = WebhookHandler(config.CHANNEL_SECRET)
23
-
24
- # ------------------------------------------------------------------------------
25
- # Web 伺服器路由
26
- # ------------------------------------------------------------------------------
27
- @app.route("/", methods=["GET"])
28
- def home():
29
- """渲染首頁,包含說明與狀態。"""
30
- base = (config.HF_SPACE_URL or request.url_root).rstrip("/")
31
- webhook_url = f"{base}/callback"
32
- static_hint = f"{base}/static/<filename>"
33
- channel_ok = "✅" if config.CHANNEL_ACCESS_TOKEN and config.CHANNEL_SECRET else "⚠️"
34
- space_ok = "✅" if config.HF_SPACE_URL else "ℹ️"
35
-
36
- return f"""
37
- <!doctype html>
38
- <html lang="zh-Hant"><head>
39
- <meta charset="utf-8"/><meta name="viewport" content="width=device-width,initial-scale=1"/>
40
- <title>地震預警 dayichen – LINE Bot Server</title>
41
- <style>
42
- :root{{--bg:#0f1115;--card:#151821;--text:#e6e8ef;--muted:#9aa4b2;--border:rgba(255,255,255,.08)}}
43
- *{{box-sizing:border-box}} body{{margin:0;background:#0f1115;color:#e6e8ef;
44
- font:16px/1.6 -apple-system,BlinkMacSystemFont,"Segoe UI",Roboto,"Noto Sans TC","PingFang TC",sans-serif;
45
- padding:32px 16px;display:flex;justify-content:center}}
46
- .wrap{{width:100%;max-width:980px}} .hero{{background:linear-gradient(135deg,#1f2937,#0f172a);
47
- border:1px solid var(--border);border-radius:16px;padding:28px;margin-bottom:20px;box-shadow:0 8px 30px rgba(0,0,0,.25)}}
48
- .title{{margin:0 0 6px;font-size:28px;font-weight:800}} .subtitle{{margin:0;color:#9aa4b2}}
49
- .grid{{display:grid;gap:16px;grid-template-columns:repeat(auto-fit,minmax(260px,1fr));margin-top:18px}}
50
- .card{{background:#151821;border:1px solid var(--border);border-radius:14px;padding:16px 18px}}
51
- h3{{margin:0 0 8px;font-size:18px}} .kbd{{padding:2px 6px;border:1px solid var(--border);border-radius:6px;background:#0b0e14}}
52
- .mono,code{{font-family:ui-monospace,SFMono-Regular,Menlo,Consolas,monospace;color:#e6e8ef}}
53
- a{{color:#60a5fa;text-decoration:none}} a:hover{{text-decoration:underline}}
54
- .badge{{display:inline-block;padding:2px 8px;border-radius:999px;background:#1f2937;border:1px solid var(--border);font-size:12px;color:#9aa4b2}}
55
- .foot{{color:#9aa4b2;font-size:13px;margin-top:18px;text-align:center}}
56
- </style></head>
57
- <body><div class="wrap"><section class="hero">
58
- <div class="badge">狀態:<span style="color:{'#86efac' if channel_ok=='✅' else '#fbbf24'}">{channel_ok}</span> LINE 金鑰 · HF Space:{space_ok}</div>
59
- <h1 class="title">地震預警 dayichen – LINE Bot</h1>
60
- <p class="subtitle">指令:/help、地震/quake、臺灣地震/台灣地震、臺灣地震畫圖/台灣地震畫圖、地震預警、AI(ai + 問題)。</p>
61
- <div class="grid">
62
- <div class="card"><h3>🚀 快速開始</h3><ul>
63
- <li><span class="kbd">/help</span>:顯示所有指令</li>
64
- <li><span class="kbd">地震</span>/<span class="kbd">quake</span>:全球近 24 小時 M≥5.0</li>
65
- <li><span class="kbd">臺灣地震</span>/<span class="kbd">台灣地震</span>:今年台灣區域清單(含日期時間)</li>
66
- <li><span class="kbd">臺灣地震畫圖</span>/<span class="kbd">台灣地震畫圖</span>:回傳地圖圖片</li>
67
- <li><span class="kbd">地震預警</span>:CWA 地震預警(最新 5 筆)</li>
68
- <li><span class="kbd">ai 你的問題</span>:AI 對話(模型:<span class="mono">{config.LLM_MODEL}</span>)</li>
69
- </ul></div>
70
- <div class="card"><h3>🛠️ Webhook / 靜態檔</h3><ul>
71
- <li>Webhook:<span class="mono"><a href="{webhook_url}">{webhook_url}</a></span></li>
72
- <li>靜態圖片:<span class="mono">{static_hint}</span></li>
73
- <li>健康檢查:<span class="mono"><a href="{base}/healthz">{base}/healthz}</a></span></li>
74
- </ul></div>
75
- <div class="card"><h3>ℹ️ 備註</h3><ul>
76
- <li>AI 快取位置:<span class="mono">{config.TRANSFORMERS_CACHE}</span></li>
77
- <li>若 AI 未安裝依賴,機器人會提示安裝,不會影響其他功能。</li>
78
- </ul></div>
79
- </div>
80
- <p class="foot">© {config.CURRENT_YEAR} dayichen · server: {base}</p>
81
- </section></div></body></html>"""
82
-
83
- @app.route("/healthz")
84
- def healthz():
85
- """健康檢查端點。"""
86
- return "ok"
87
-
88
- @app.route("/static/<path:filename>")
89
- def serve_static(filename):
90
- """提��靜態檔案(例如,生成的地圖)。"""
91
- return send_from_directory(config.STATIC_DIR, filename)
92
-
93
- # ------------------------------------------------------------------------------
94
- # LINE Webhook 處理器
95
- # ------------------------------------------------------------------------------
96
- @app.route("/callback", methods=["POST"])
97
- def callback():
98
- """處理來自 LINE 平台的傳入 Webhooks。"""
99
- signature = request.headers.get("X-Line-Signature")
100
- body = request.get_data(as_text=True)
101
- try:
102
- handler.handle(body, signature)
103
- except InvalidSignatureError:
104
- abort(400)
105
- return "OK"
106
-
107
- @handler.add(MessageEvent, message=TextMessageContent)
108
- def handle_message(event):
109
- """
110
- 處理來自使用者的文字訊息並回覆。
111
- 所有邏輯都委派給 command_handler。
112
- """
113
- # 決定用於生成圖片連結的基礎 URL
114
- base_url = request.url_root.rstrip("/")
115
-
116
- # 從處理器獲取回覆訊息
117
- reply_messages = process_message(event.message.text, base_url)
118
-
119
- # 發送回覆
120
- with ApiClient(line_config) as api_client:
121
- line_bot_api = MessagingApi(api_client)
122
- line_bot_api.reply_message_with_http_info(
123
- ReplyMessageRequest(
124
- reply_token=event.reply_token,
125
- messages=reply_messages
126
- )
127
  )
 
1
+ # app.py
2
+ # 首先匯入 config,以設定環境變數
3
+ import config
4
+
5
+ from flask import Flask, request, abort, send_from_directory
6
+ from linebot.v3 import WebhookHandler
7
+ from linebot.v3.exceptions import InvalidSignatureError
8
+ from linebot.v3.messaging import (
9
+ Configuration, ApiClient, MessagingApi,
10
+ ReplyMessageRequest
11
+ )
12
+ from linebot.v3.webhooks import MessageEvent, TextMessageContent
13
+
14
+ # 匯入指令處理器
15
+ from command_handler import process_message
16
+
17
+ # ------------------------------------------------------------------------------
18
+ # Flask & LINE Bot 設定
19
+ # ------------------------------------------------------------------------------
20
+ app = Flask(__name__)
21
+ line_config = Configuration(access_token=config.CHANNEL_ACCESS_TOKEN)
22
+ handler = WebhookHandler(config.CHANNEL_SECRET)
23
+
24
+ # ------------------------------------------------------------------------------
25
+ # Web 伺服器路由
26
+ # ------------------------------------------------------------------------------
27
+ @app.route("/", methods=["GET"])
28
+ def home():
29
+ """渲染首頁,包含說明與狀態。"""
30
+ base = (config.HF_SPACE_URL or request.url_root).rstrip("/")
31
+ webhook_url = f"{base}/callback"
32
+ static_hint = f"{base}/static/<filename>"
33
+ channel_ok = "✅" if config.CHANNEL_ACCESS_TOKEN and config.CHANNEL_SECRET else "⚠️"
34
+ space_ok = "✅" if config.HF_SPACE_URL else "ℹ️"
35
+
36
+ return f"""
37
+ <!doctype html>
38
+ <html lang="zh-Hant"><head>
39
+ <meta charset="utf-8"/><meta name="viewport" content="width=device-width,initial-scale=1"/>
40
+ <title>地震預警 dayichen – LINE Bot Server</title>
41
+ <style>
42
+ :root{{--bg:#0f1115;--card:#151821;--text:#e6e8ef;--muted:#9aa4b2;--border:rgba(255,255,255,.08)}}
43
+ *{{box-sizing:border-box}} body{{margin:0;background:#0f1115;color:#e6e8ef;
44
+ font:16px/1.6 -apple-system,BlinkMacSystemFont,"Segoe UI",Roboto,"Noto Sans TC","PingFang TC",sans-serif;
45
+ padding:32px 16px;display:flex;justify-content:center}}
46
+ .wrap{{width:100%;max-width:980px}} .hero{{background:linear-gradient(135deg,#1f2937,#0f172a);
47
+ border:1px solid var(--border);border-radius:16px;padding:28px;margin-bottom:20px;box-shadow:0 8px 30px rgba(0,0,0,.25)}}
48
+ .title{{margin:0 0 6px;font-size:28px;font-weight:800}} .subtitle{{margin:0;color:#9aa4b2}}
49
+ .grid{{display:grid;gap:16px;grid-template-columns:repeat(auto-fit,minmax(260px,1fr));margin-top:18px}}
50
+ .card{{background:#151821;border:1px solid var(--border);border-radius:14px;padding:16px 18px}}
51
+ h3{{margin:0 0 8px;font-size:18px}} .kbd{{padding:2px 6px;border:1px solid var(--border);border-radius:6px;background:#0b0e14}}
52
+ .mono,code{{font-family:ui-monospace,SFMono-Regular,Menlo,Consolas,monospace;color:#e6e8ef}}
53
+ a{{color:#60a5fa;text-decoration:none}} a:hover{{text-decoration:underline}}
54
+ .badge{{display:inline-block;padding:2px 8px;border-radius:999px;background:#1f2937;border:1px solid var(--border);font-size:12px;color:#9aa4b2}}
55
+ .foot{{color:#9aa4b2;font-size:13px;margin-top:18px;text-align:center}}
56
+ </style></head>
57
+ <body><div class="wrap"><section class="hero">
58
+ <div class="badge">狀態:<span style="color:{'#86efac' if channel_ok=='✅' else '#fbbf24'}">{channel_ok}</span> LINE 金鑰 · HF Space:{space_ok}</div>
59
+ <h1 class="title">地震預警 dayichen – LINE Bot</h1>
60
+ <p class="subtitle">指令:/help、地震/quake、臺灣地震/台灣地震、臺灣地震畫圖/台灣地震畫圖、地震預警、AI(ai + 問題)。</p>
61
+ <div class="grid">
62
+ <div class="card"><h3>🚀 快速開始</h3><ul>
63
+ <li><span class="kbd">/help</span>:顯示所有指令</li>
64
+ <li><span class="kbd">地震</span>/<span class="kbd">quake</span>:全球近 24 小時 M≥5.0</li>
65
+ <li><span class="kbd">臺灣地震</span>/<span class="kbd">台灣地震</span>:今年台灣區域清單(含日期時間)</li>
66
+ <li><span class="kbd">臺灣地震畫圖</span>/<span class="kbd">台灣地震畫圖</span>:回傳地圖圖片</li>
67
+ <li><span class="kbd">地震預警</span>:CWA 地震預警(最新 5 筆)</li>
68
+ <li><span class="kbd">ai 你的問題</span>:AI 對話(模型:<span class="mono">{config.LLM_MODEL}</span>)</li>
69
+ </ul></div>
70
+ <div class="card"><h3>🛠️ Webhook / 靜態檔</h3><ul>
71
+ <li>Webhook:<span class="mono"><a href="{webhook_url}">{webhook_url}</a></span></li>
72
+ <li>靜態圖片:<span class="mono">{static_hint}</span></li>
73
+ <li>健康檢查:<span class="mono"><a href="{base}/healthz">{base}/healthz}</a></span></li>
74
+ </ul></div>
75
+ <div class="card"><h3>ℹ️ 備註</h3><ul>
76
+ <li>AI 快取位置:<span class="mono">{config.TRANSFORMERS_CACHE}</span></li>
77
+ <li>若 AI 未安裝依賴,機器人會提示安裝,不會影響其他功能。</li>
78
+ </ul></div>
79
+ </div>
80
+ <p class="foot">© {config.CURRENT_YEAR} dayichen · server: {base}</p>
81
+ </section></div></body></html>"""
82
+
83
+ @app.route("/healthz")
84
+ def healthz():
85
+ """健康檢查端點。"""
86
+ return "ok"
87
+
88
+ @app.route("/static/<path:filename>")
89
+ def serve_static(filename):
90
+ """提供靜態檔案(例如,生成的地圖)。"""
91
+ return send_from_directory(config.STATIC_DIR, filename)
92
+
93
+ # ------------------------------------------------------------------------------
94
+ # LINE Webhook 處理器
95
+ # ------------------------------------------------------------------------------
96
+ @app.route("/callback", methods=["POST"])
97
+ def callback():
98
+ """處理來自 LINE 平台的傳入 Webhooks。"""
99
+ signature = request.headers.get("X-Line-Signature")
100
+ body = request.get_data(as_text=True)
101
+ try:
102
+ handler.handle(body, signature)
103
+ except InvalidSignatureError:
104
+ abort(400)
105
+ return "OK"
106
+
107
+ @handler.add(MessageEvent, message=TextMessageContent)
108
+ def handle_message(event):
109
+ """
110
+ 處理來自使用者的文字訊息並回覆。
111
+ 所有邏輯都委派給 command_handler。
112
+ """
113
+ # 決定用於生成圖片連結的基礎 URL
114
+ base_url = request.url_root.rstrip("/")
115
+
116
+ # 從處理器獲取回覆訊息
117
+ reply_messages = process_message(event.message.text, base_url)
118
+
119
+ # 發送回覆
120
+ with ApiClient(line_config) as api_client:
121
+ line_bot_api = MessagingApi(api_client)
122
+ line_bot_api.reply_message_with_http_info(
123
+ ReplyMessageRequest(
124
+ reply_token=event.reply_token,
125
+ messages=reply_messages
126
+ )
127
  )