Update main.py

main.py

@@ -1,16 +1,17 @@
-# 東吳大學資料系2025年LINEBOT
-
-from flask import Flask, request, abort, send_from_directory
+# 東吳大學資料系 2025 年 LINEBOT
 
+import os
+import tempfile
+import logging
+import requests
 import markdown
+
+from flask import Flask, request, abort, send_from_directory
 from bs4 import BeautifulSoup
+import google.generativeai as genai
 
-from linebot.v3 import (
-    WebhookHandler
-)
-from linebot.v3.exceptions import (
-    InvalidSignatureError
-)
+from linebot.v3 import WebhookHandler
+from linebot.v3.exceptions import InvalidSignatureError
 from linebot.v3.messaging import (
     Configuration,
     ApiClient,
@@ -26,77 +27,69 @@ from linebot.v3.webhooks import (
     ImageMessageContent
 )
 
-import os
-import requests
-import logging
-import tempfile
-import google.generativeai as genai
-
-# HF_TOKEN = os.environ.get('HF_TOKEN')
-# headers = {"Authorization": f"Bearer {HF_TOKEN}"}
-
-GOOGLE_API_KEY = os.environ.get('GOOGLE_API_KEY')
+# === 初始化 Google Gemini ===
+GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
 genai.configure(api_key=GOOGLE_API_KEY)
-model = genai.GenerativeModel('gemini-2.0-flash')
+model = genai.GenerativeModel("gemini-2.0-flash")
 
+# === 初始設定 ===
 static_tmp_path = "/tmp"
 os.makedirs(static_tmp_path, exist_ok=True)
+base_url = os.getenv("SPACE_HOST")  # e.g., "your-space-name.hf.space"
+
+# === Flask 應用初始化 ===
+app = Flask(__name__)
+logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
+app.logger.setLevel(logging.INFO)
 
-# API_URL = "https://api-inference.huggingface.co/models/mistralai/Mixtral-8x7B-Instruct-v0.1"
+channel_secret = os.environ.get("YOUR_CHANNEL_SECRET")
+channel_access_token = os.environ.get("YOUR_CHANNEL_ACCESS_TOKEN")
+
+configuration = Configuration(access_token=channel_access_token)
+handler = WebhookHandler(channel_secret)
+
+# === AI Query 包裝 ===
 def query(payload):
-    # response = requests.post(API_URL, headers=headers, json=payload)
     response = model.generate_content(payload)
     return response.text
 
-app = Flask(__name__)
-
-### 將/tmp資料夾中的圖片建立成URL
-
+# === 靜態圖檔路由 ===
 @app.route("/images/<filename>")
 def serve_image(filename):
-    return send_from_directory("/tmp", filename)
-
-base_url = os.getenv("SPACE_HOST")
-###
-
-logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
-app.logger.setLevel(logging.INFO)
+    return send_from_directory(static_tmp_path, filename)
 
-channel_secret = os.environ.get('YOUR_CHANNEL_SECRET')
-channel_access_token = os.environ.get('YOUR_CHANNEL_ACCESS_TOKEN')
-
-configuration = Configuration(access_token=channel_access_token)
-handler = WebhookHandler(channel_secret)
+@app.route("/static/<path:path>")
+def send_static_content(path):
+    return send_from_directory("static", path)
 
+# === LINE Webhook 接收端點 ===
 @app.route("/")
 def home():
     return {"message": "Line Webhook Server"}
 
-@app.route("/", methods=['POST'])
+@app.route("/", methods=["POST"])
 def callback():
-    # get X-Line-Signature header value
-    signature = request.headers['X-Line-Signature']
-
-    # get request body as text
+    signature = request.headers.get("X-Line-Signature")
     body = request.get_data(as_text=True)
-    app.logger.info("Request body: " + body)
+    app.logger.info(f"Request body: {body}")
 
-    # handle webhook body
     try:
         handler.handle(body, signature)
     except InvalidSignatureError:
-        app.logger.info("Invalid signature. Please check your channel access token/channel secret.")
+        app.logger.warning("Invalid signature. Please check channel credentials.")
         abort(400)
 
-    return 'OK'
+    return "OK"
 
+# === 處理文字訊息 ===
 @handler.add(MessageEvent, message=TextMessageContent)
-def handle_message(event):
+def handle_text_message(event):
     with ApiClient(configuration) as api_client:
         line_bot_api = MessagingApi(api_client)
         response = query(event.message.text)
         html_msg = markdown.markdown(response)
-        soup = BeautifulSoup(html_msg, 'html.parser')
+        soup = BeautifulSoup(html_msg, "html.parser")
+
         line_bot_api.reply_message_with_http_info(
             ReplyMessageRequest(
                 reply_token=event.reply_token,
@@ -104,19 +97,21 @@ def handle_message(event):
             )
         )
 
+# === 處理圖片訊息 ===
 @handler.add(MessageEvent, message=ImageMessageContent)
-def handle_content_message(event):
-    ext = 'jpg'
+def handle_image_message(event):
+    ext = "jpg"
     with ApiClient(configuration) as api_client:
         blob_api = MessagingApiBlob(api_client)
         content = blob_api.get_message_content(message_id=event.message.id)
 
-    with tempfile.NamedTemporaryFile(dir=static_tmp_path, suffix='.' + ext, delete=False) as tf:
+    with tempfile.NamedTemporaryFile(dir=static_tmp_path, suffix=f".{ext}", delete=False) as tf:
         tf.write(content)
         filename = os.path.basename(tf.name)
 
     image_url = f"https://{base_url}/images/{filename}"
-    app.logger.info(image_url)
+    app.logger.info(f"Image URL: {image_url}")
+
     with ApiClient(configuration) as api_client:
         line_bot_api = MessagingApi(api_client)
         line_bot_api.reply_message(
@@ -130,8 +125,3 @@ def handle_content_message(event):
                 ]
             )
         )
-
-
-@app.route('/static/<path:path>')
-def send_static_content(path):
-    return send_from_directory('static', path)