Update app.py

app.py

@@ -1,19 +1,35 @@
+import os
 from flask import Flask, request, abort
 from linebot import LineBotApi, WebhookHandler
 from linebot.exceptions import InvalidSignatureError
 from linebot.models import MessageEvent, TextMessage, TextSendMessage
-import os
 
-# 建立 Flask 應用
+from google import generativeai as genai
+import markdown
+from bs4 import BeautifulSoup
+
+# 初始化 Flask
 app = Flask(__name__)
 
-# 從 Huggingface 的 Secrets 中讀取 LINE 憑證
+# 設定 LINE Bot 憑證（從 Huggingface Secrets 讀取）
 line_bot_api = LineBotApi(os.environ.get("LINE_CHANNEL_ACCESS_TOKEN"))
 handler = WebhookHandler(os.environ.get("LINE_CHANNEL_SECRET"))
 
+# 初始化 Gemini Flash（使用免費模型）
+genai.configure(api_key=os.environ.get("GOOGLE_API_KEY"))
+model = genai.GenerativeModel("gemini-1.5-flash")
+
+def query(prompt: str) -> str:
+    try:
+        response = model.generate_content(prompt)
+        html = markdown.markdown(response.text)
+        return BeautifulSoup(html, "html.parser").get_text()
+    except Exception as e:
+        return f"⚠️ Gemini 發生錯誤：{e}"
+
+# LINE webhook 入口
 @app.route("/", methods=["POST"])
 def callback():
-    # 接收 X-Line-Signature
     signature = request.headers.get("X-Line-Signature")
     body = request.get_data(as_text=True)
 
@@ -21,16 +37,11 @@ def callback():
         handler.handle(body, signature)
     except InvalidSignatureError:
         abort(400)
-
     return "OK"
 
-# 收到訊息時回覆
+# 處理文字訊息
 @handler.add(MessageEvent, message=TextMessage)
 def handle_message(event):
-    user_text = event.message.text
-    reply_text = f"你說的是：{user_text}"
-    line_bot_api.reply_message(event.reply_token, TextSendMessage(text=reply_text))
-
-# 本地測試用（部署在 Huggingface 時不需要）
-if __name__ == "__main__":
-    app.run()
+    user_msg = event.message.text.strip()
+    bot_reply = query(user_msg)
+    line_bot_api.reply_message(event.reply_token, TextSendMessage(text=bot_reply))