Update app.py

app.py

@@ -1,96 +1,104 @@
-from fastapi import FastAPI, HTTPException, Query
-from starlette.responses import JSONResponse
-from obspy.clients.fdsn import Client
-from obspy.core.utcdatetime import UTCDateTime
-import logging
-from datetime import date
+import os
+from flask import Flask, request, abort
+from linebot.v3 import WebhookHandler
+from linebot.v3.exceptions import InvalidSignatureError
+from linebot.v3.messaging import (
+    Configuration, ApiClient, MessagingApi,
+    ReplyMessageRequest, TextMessage
+)
+from linebot.v3.webhooks import MessageEvent, TextMessageContent
 
-# Configure logging
-logging.basicConfig(level=logging.INFO)
+import requests
+import pandas as pd
+from datetime import datetime, timedelta
 
-# Initialize the FastAPI app
-app = FastAPI(
-    title="Dynamic Earthquake Catalog API",
-    description="A robust API to fetch earthquake data from the IRIS FDSN web service using URL parameters.",
-    version="2.1.0",
-)
+# --- 環境變數：我們將在 Hugging Face Spaces 的 Secrets 中設定 ---
+CHANNEL_ACCESS_TOKEN = os.getenv('CHANNEL_ACCESS_TOKEN')
+CHANNEL_SECRET = os.getenv('CHANNEL_SECRET')
 
-# Initialize the IRIS FDSN client
-try:
-    client = Client("IRIS")
-    logging.info("Successfully initialized IRIS FDSN client.")
-except Exception as e:
-    client = None
-    logging.error(f"Could not initialize IRIS FDSN client on startup: {e}")
+# --- Flask & LINE Bot 初始化 ---
+app = Flask(__name__)
 
-@app.get("/")
-def greet_json():
-    """
-    A simple 'Hello World' endpoint.
-    """
-    return {"Hello": "World!", "message": "Go to /docs to try the dynamic earthquake endpoint.  cwadayi \n https://cwadayi-python-app.hf.space/earthquakes?start_date=2024-07-01&end_date=2024-07-07&min_magnitude=5.0"}
+configuration = Configuration(access_token=CHANNEL_ACCESS_TOKEN)
+handler = WebhookHandler(CHANNEL_SECRET)
 
-@app.get("/earthquakes")
-async def get_earthquake_catalog(
-    start_date: date = Query(default="2024-01-01", description="Start date in YYYY-MM-DD format"),
-    end_date: date = Query(default="2024-01-07", description="End date in YYYY-MM-DD format"),
-    min_magnitude: float = Query(default=4.5, description="Minimum earthquake magnitude (e.g., 5.5)", gt=0, le=10)
-):
-    """
-    Fetches earthquake data based on date and magnitude parameters provided in the URL.
-    """
-    if not client:
-        raise HTTPException(status_code=503, detail="The IRIS FDSN client is not available.")
-    
-    if start_date > end_date:
-        raise HTTPException(status_code=400, detail="The start_date cannot be after the end_date.")
+# --- 地震查詢核心邏輯 (與之前相同) ---
+USGS_API_BASE_URL = "https://earthquake.usgs.gov/fdsnws/event/1/query"
 
+def fetch_earthquake_data_for_line():
+    """查詢過去 24 小時內規模 5.0 以上的地震，並回傳格式化的文字。"""
+    now = datetime.now()
+    yesterday = now - timedelta(days=1)
+    
+    params = {
+        "format": "geojson",
+        "starttime": yesterday.strftime('%Y-%m-%d'),
+        "endtime": now.strftime('%Y-%m-%d'),
+        "minmagnitude": 5.0,
+        "limit": 10,
+        "orderby": "time-desc"
+    }
+    
     try:
-        starttime = UTCDateTime(start_date)
-        endtime = UTCDateTime(end_date)
+        response = requests.get(USGS_API_BASE_URL, params=params, timeout=15)
+        response.raise_for_status()
+        data = response.json()
+        
+        features = data.get('features', [])
+        if not features:
+            return "✅ 過去24小時內，全球無規模 5.0 以上的顯著地震。"
+            
+        reply_text = f"🚨 近 24 小時全球顯著地震 (M≥5.0):\n{'-'*20}\n"
+        
+        for feature in features:
+            prop = feature['properties']
+            mag = prop['mag']
+            place = prop['place']
+            event_time = datetime.fromtimestamp(prop['time'] / 1000).strftime('%H:%M')
+            
+            reply_text += f"震級: {mag:.1f} | 時間: {event_time} (UTC)\n地點: {place}\n\n"
+            
+        return reply_text.strip()
 
-        logging.info(f"Fetching events from {starttime} to {endtime} with min magnitude {min_magnitude}.")
-        catalog = client.get_events(
-            starttime=starttime,
-            endtime=endtime,
-            minmagnitude=min_magnitude,
-        )
-        logging.info(f"Found {len(catalog)} events. Now processing.")
+    except requests.exceptions.RequestException as e:
+        return f"❌ 查詢失敗，無法連接到 USGS 伺服器: {e}"
+    except Exception as e:
+        return f"❌ 處理資料時發生未知錯誤: {e}"
 
-        earthquake_list = []
-        for event in catalog:
-            try:
-                origin = event.preferred_origin()
-                magnitude = event.preferred_magnitude()
+# --- Flask Webhook 路由 ---
+@app.route("/callback", methods=['POST'])
+def callback():
+    signature = request.headers['X-Line-Signature']
+    body = request.get_data(as_text=True)
+    app.logger.info("Request body: " + body)
 
-                if not origin or not magnitude:
-                    continue
+    try:
+        handler.handle(body, signature)
+    except InvalidSignatureError:
+        app.logger.info("Invalid signature. Please check your channel secret.")
+        abort(400)
+    return 'OK'
 
-                # --- Key Change Start ---
-                # Safely get the description using getattr. This avoids errors if the 'descriptions'
-                # attribute doesn't exist, by providing a default empty list [].
-                place = "N/A"
-                descriptions = getattr(event, 'descriptions', [])
-                if descriptions:
-                    place = descriptions[0].text
-                # --- Key Change End ---
+# --- LINE 訊息處理 ---
+@handler.add(MessageEvent, message=TextMessageContent)
+def handle_message(event):
+    user_message = event.message.text
+    reply_text = ""
 
-                earthquake_list.append({
-                    "time": origin.time.isoformat(),
-                    "latitude": origin.latitude,
-                    "longitude": origin.longitude,
-                    "depth_km": origin.depth / 1000.0 if origin.depth is not None else None,
-                    "magnitude": magnitude.mag,
-                    "magnitude_type": magnitude.magnitude_type,
-                    "place": place,
-                })
-            except Exception as e:
-                logging.warning(f"Skipping event due to unexpected processing error: {e}")
-                continue
-        
-        logging.info(f"Successfully processed {len(earthquake_list)} events.")
-        return JSONResponse(content={"earthquakes": earthquake_list})
+    if "地震" in user_message or "quake" in user_message.lower():
+        reply_text = fetch_earthquake_data_for_line()
+    elif "你好" in user_message or "hi" in user_message.lower():
+        reply_text = "👋 你好！我是地震查詢機器人，試著傳送「地震」來查詢最新資訊吧！"
+    else:
+        return
 
-    except Exception as e:
-        logging.error(f"A major error occurred during the request: {e}")
-        raise HTTPException(status_code=500, detail=f"An internal error occurred: {e}")
+    with ApiClient(configuration) as api_client:
+        line_bot_api = MessagingApi(api_client)
+        line_bot_api.reply_message_with_http_info(
+            ReplyMessageRequest(
+                reply_token=event.reply_token,
+                messages=[TextMessage(text=reply_text)]
+            )
+        )
+
+# 注意：我們已經刪除了 if __name__ == "__main__": app.run() 區塊