Update app.py

app.py

@@ -1,15 +1,25 @@
 # app.py
 
-import gradio as gr
-import html
+from fastapi import FastAPI, HTTPException
+from starlette.responses import JSONResponse
 import requests
 import re
 import xml.etree.ElementTree as ET
 from datetime import datetime, timedelta, timezone
+import logging
+import html
+
+# --- 配置日誌 ---
+logging.basicConfig(level=logging.INFO)
 
-# ===============================================================
-# Geocode 與縣市名稱的對照字典
-# ===============================================================
+# --- 初始化 FastAPI 應用 ---
+app = FastAPI(
+    title="台灣地震速報 API",
+    description="一個從台灣中央氣象署(CWA)公開預警XML Feed中，擷取最新地震速報的API。",
+    version="1.0.0",
+)
+
+# --- Geocode 與縣市名稱的對照字典 ---
 GEOCODE_MAP = {
     "10002": "宜蘭縣", "10004": "新竹縣", "10005": "苗栗縣",
     "10007": "彰化縣", "10008": "南投縣", "10009": "雲林縣",
@@ -21,42 +31,29 @@ GEOCODE_MAP = {
     "09020": "金門縣"
 }
 
-# ===============================================================
-# 主要功能：讀取並過濾地震速報
-# ===============================================================
-def get_disaster_warnings() -> str:
+# --- 核心功能函式 ---
+def get_cwa_earthquakes():
     """
-    Fetches and formats Earthquake Early Warnings from the CWA PWS feed.
-
-    This function retrieves the latest disaster warnings, filters them to show
-    only 'Earthquake Early Warning' (地震速報) alerts issued by the CWA, and
-    formats the output in Markdown, including affected counties.
+    抓取並解析 CWA 地震速報 Feed，回傳結構化的 Python 列表。
     """
-    # 使用時間戳避免快取
     timestamp = int(datetime.now().timestamp())
     feed_url = f"https://cbs.tw/files/rssatomfeed.xml?_={timestamp}"
     keywords = ["中央氣象署", "氣象署"]
-    
     headers = {
         'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36',
         'Cache-Control': 'no-cache',
         'Pragma': 'no-cache'
     }
-    
+
     try:
         response = requests.get(feed_url, headers=headers, timeout=15)
         response.raise_for_status()
-        
+
         xml_content = response.content.decode('utf-8')
         xml_content = re.sub(' xmlns="[^"]+"', '', xml_content, count=1)
         root = ET.fromstring(xml_content)
-
-        # 取得台灣時區
-        tw_tz = timezone(timedelta(hours=8))
-        current_time_str = datetime.now(tw_tz).strftime('%Y-%m-%d %H:%M:%S')
         
-        markdown_output = f"## 最新地震速報 (僅顯示來自：中央氣象署)\n---\n*最後更新時間：{current_time_str} (GMT+8)*\n\n"
-        messages_found = 0
+        earthquake_alerts = []
 
         for entry in root.findall('entry'):
             summary_text = ''
@@ -73,58 +70,71 @@ def get_disaster_warnings() -> str:
             is_from_cwa = any(keyword in summary_text for keyword in keywords)
 
             if is_earthquake_alert and is_from_cwa:
-                messages_found += 1
-                
                 sent_tag = entry.find('sent')
-                published_time = sent_tag.text if sent_tag is not None else "無發布時間"
-
-                markdown_output += f"### {html.escape(title)}\n"
-                markdown_output += f"**發布時間:** {html.escape(published_time)}\n\n"
+                published_time = sent_tag.text if sent_tag is not None else "N/A"
 
+                area_desc = ''
+                affected_counties = []
                 area_tag = entry.find('area')
                 if area_tag is not None:
                     area_desc_tag = area_tag.find('areadesc')
                     area_desc = area_desc_tag.text if area_desc_tag is not None else ''
 
-                    affected_locations = []
                     for geocode_tag in area_tag.findall('geocode'):
                         value_tag = geocode_tag.find('value')
                         if value_tag is not None:
                             code = value_tag.text
                             location_name = GEOCODE_MAP.get(code, f"未知代碼({code})")
-                            affected_locations.append(location_name)
-                    
-                    area_details = ""
-                    if affected_locations:
-                        area_details = f" ({'、'.join(affected_locations)})"
-                    
-                    full_area_info = f"{html.escape(area_desc)}{area_details}"
-                    markdown_output += f"**影響範圍:** {full_area_info}\n\n"
+                            affected_counties.append(location_name)
                 
-                markdown_output += f"**內容:**\n{html.escape(summary_text)}\n\n"
-                markdown_output += "---\n"
+                alert = {
+                    "title": html.unescape(title),
+                    "publish_time": published_time,
+                    "summary": html.unescape(summary_text),
+                    "affected_area_desc": html.unescape(area_desc),
+                    "affected_counties": affected_counties
+                }
+                earthquake_alerts.append(alert)
+        
+        return earthquake_alerts
 
-        if messages_found == 0:
-            return markdown_output + "目前沒有最新的地震速報。"
-            
-        return markdown_output
     except requests.exceptions.RequestException as e:
-        return f"網路請求失敗，無法取得資料: {e}"
+        logging.error(f"無法從 CWA 取得資料: {e}")
+        raise HTTPException(status_code=503, detail="無法連接到 CWA 資料來源")
     except ET.ParseError as e:
-        return f"XML 資料解析失敗，請稍後再試: {e}"
+        logging.error(f"XML 資料解析失敗: {e}")
+        raise HTTPException(status_code=500, detail="解析 CWA 資料時發生錯誤")
     except Exception as e:
-        return f"讀取訊息時發生未預期的錯誤: {e}"
+        logging.error(f"發生未預期的錯誤: {e}")
+        raise HTTPException(status_code=500, detail=f"發生內部錯誤: {e}")
 
-# ===============================================================
-# 建立 Gradio 介面
-# ===============================================================
-with gr.Blocks(title="台灣地震速報") as app:
-    gr.Markdown("## 台灣地震速報")
-    gr.Markdown("自動讀取最新的地震速報，並過濾只顯示由「中央氣象署」發布的內容。")
-    output_markdown = gr.Markdown()
-    app.load(get_disaster_warnings, outputs=output_markdown, every=60) # 每60秒自動更新一次
+# --- API 端點 ---
+@app.get("/")
+def root():
+    """
+    根目錄端點，提供 API 簡介和範例連結。
+    """
+    return {
+        "message": "歡迎使用台灣地震速報 API",
+        "documentation": "/docs",
+        "earthquake_feed_endpoint": "/cwa-earthquakes"
+    }
 
+@app.get("/cwa-earthquakes")
+async def get_earthquakes_endpoint():
+    """
+    提供最新的地震速報。
 
-# 啟動 Gradio 網頁介面
-if __name__ == "__main__":
-    app.launch()
+    此端點會從 CWA 的公開預警 Feed 中擷取資料，
+    篩選出來自「中央氣象署」的「地震速報」，
+    並以 JSON 格式回傳。
+    """
+    alerts = get_cwa_earthquakes()
+    tw_tz = timezone(timedelta(hours=8))
+    last_updated = datetime.now(tw_tz).isoformat()
+    
+    return JSONResponse(content={
+        "last_updated": last_updated,
+        "source": "台灣中央氣象署 (CWA)",
+        "alerts": alerts
+    })