Restructure

app.py

@@ -1,17 +1,24 @@
+import os
 import sqlite3
 import requests
-import os
-from flask import Flask, request, abort
+import tempfile
+import logging
+from io import BytesIO
+
+from flask import Flask, request, abort, send_from_directory
+from PIL import Image
+
 from linebot.v3.webhook import WebhookParser, WebhookHandler
-from linebot.v3.webhooks import MessageEvent, TextMessageContent
-from linebot.v3.messaging import MessagingApi, Configuration, ApiClient
+from linebot.v3.webhooks import MessageEvent, TextMessageContent, ImageMessageContent
+from linebot.v3.messaging import MessagingApi, Configuration, ApiClient, MessagingApiBlob
 from linebot.v3.messaging.models import (
     TextMessage, ReplyMessageRequest,
     FlexMessage, FlexBubble, FlexBox, FlexText, FlexButton, URIAction,
-    QuickReply, QuickReplyItem, LocationAction
+    QuickReply, QuickReplyItem, LocationAction, ImageMessage
 )
 from linebot.v3.exceptions import InvalidSignatureError
 
+import google.generativeai as genai
 import io
 import json
 from dotenv import load_dotenv
@@ -27,8 +34,15 @@ load_dotenv()
 app = Flask(__name__)
 
 # 從環境變數讀取 LINE Bot 設定
-LINE_CHANNEL_SECRET = os.getenv("LINE_CHANNEL_SECRET", "YOUR_FALLBACK_SECRET") # 建議提供一個預設值或錯誤處理
-LINE_CHANNEL_ACCESS_TOKEN = os.getenv("LINE_CHANNEL_ACCESS_TOKEN", "YOUR_FALLBACK_TOKEN")
+LINE_CHANNEL_SECRET = os.environ.get("LINE_CHANNEL_SECRET")
+LINE_CHANNEL_ACCESS_TOKEN = os.environ.get("LINE_CHANNEL_ACCESS_TOKEN")
+
+if not LINE_CHANNEL_SECRET or not LINE_CHANNEL_ACCESS_TOKEN:
+    raise RuntimeError("Missing essential environment variables")
+
+print(f"LINE_CHANNEL_SECRET: {LINE_CHANNEL_SECRET}")
+print(f"LINE_CHANNEL_ACCESS_TOKEN: {LINE_CHANNEL_ACCESS_TOKEN}")
+#print(f"GOOGLE_MAP_API_KEY: {GOOGLE_MAP_API_KEY}")
 
 # --- 資料庫路徑設定 ---
 BASE_DIR = os.path.dirname(os.path.abspath(__file__))
@@ -93,14 +107,24 @@ if not DOWNLOAD_SUCCESS:
     print("CRITICAL: Database download failed or was skipped. The application might not function as expected if the database is required at startup.")
 # --- END 應用程式啟動時執行資料庫下載 ---
 
-print(f"LINE_CHANNEL_SECRET: {LINE_CHANNEL_SECRET}")
-print(f"LINE_CHANNEL_ACCESS_TOKEN: {LINE_CHANNEL_ACCESS_TOKEN}")
-#print(f"GOOGLE_MAP_API_KEY: {GOOGLE_MAP_API_KEY}")
-
 configuration = Configuration(access_token=LINE_CHANNEL_ACCESS_TOKEN)
 parser = WebhookParser(LINE_CHANNEL_SECRET)
 handler = WebhookHandler(LINE_CHANNEL_SECRET)
 
+genai.configure(api_key=GOOGLE_API_KEY)
+chat = genai.GenerativeModel(model_name="gemini-1.5-flash")
+
+logging.basicConfig(level=logging.INFO)
+app.logger.setLevel(logging.INFO)
+
+@app.route("/images/<filename>")
+def serve_image(filename):
+    return send_from_directory(static_tmp_path, filename)
+
+@app.route("/")
+def home():
+    return {"message": "Line Webhook Server"}
+
 @app.route("/callback", methods=["POST"])
 def callback():
     
@@ -122,14 +146,32 @@ def callback():
 
     with ApiClient(configuration) as api_client:
         messaging_api = MessagingApi(api_client)
+        blob_api = MessagingApiBlob(api_client)
 
         for event in events:
             if event.type == "message":
+                # 文字訊息
                 if event.message.type == "text":
                     user_input = event.message.text.strip()
                     print("📨 收到訊息：", user_input)
 
-                    if user_input == "查詢藥品":
+                    # AI 問答
+                    if user_input.startswith("AI "):
+                        prompt = "你是一個中文的AI助手，請用繁體中文回答。\n" + user_input[3:].strip()
+                        try:
+                            response = chat.generate_content(prompt)
+                            reply_text = response.text
+                        except Exception as e:
+                            reply_text = f"AI 回答失敗：{e}"
+
+                        reply_request = ReplyMessageRequest(
+                            reply_token=event.reply_token,
+                            messages=[TextMessage(text=reply_text)]
+                        )
+                        messaging_api.reply_message(reply_message_request=reply_request)
+
+                    # 查詢藥品
+                    elif user_input == "查詢藥品":
                         reply_text = "請輸入藥品名稱，例如：口服感冒藥"
                         reply_request = ReplyMessageRequest(
                             reply_token=event.reply_token,
@@ -137,6 +179,7 @@ def callback():
                         )
                         messaging_api.reply_message(reply_message_request=reply_request)
 
+                    # 查詢藥局
                     elif "查詢藥局" in user_input:
                         quick_reply = QuickReply(
                             items=[QuickReplyItem(action=LocationAction(label="傳送我的位置"))]
@@ -147,8 +190,8 @@ def callback():
                         )
                         messaging_api.reply_message(reply_message_request=reply_request)
 
+                    # 其他：查詢藥品資料庫，副作用永遠由AI產生
                     else:
-                        # 直接當作藥品名稱查詢
                         medicine_name = user_input.lower()
                         try:
                             conn = sqlite3.connect(DB_PATH)
@@ -159,20 +202,44 @@ def callback():
                                 WHERE LOWER(中文品名) LIKE ?
                                 LIMIT 3
                             """
-                            cursor.execute(query, ('%' + medicine_name + '%',))
+                            like_param = f'%{medicine_name}%'
+                            cursor.execute(query, (like_param,))
                             rows = cursor.fetchall()
                             conn.close()
-
-                            if not rows:
-                                reply_text = f"找不到與「{medicine_name}」相關的藥品。"
+                            
+                            if rows:
+                                zh_name, en_name, indication = rows[0]
+                                # 副作用由 AI 產生
+                                prompt = (
+                                    f"請用簡短條列式，僅列出副作用，針對藥品「{zh_name}」(英文名：{en_name})，"
+                                    "請用繁體中文回答，若無法判斷請推測。"
+                                )
+                                try:
+                                    ai_resp = chat.generate_content(prompt)
+                                    side_effects = ai_resp.text.strip()
+                                except Exception as e:
+                                    side_effects = f"AI 回答失敗：{e}"
+                                reply_text = (
+                                    f"🔹 中文品名：{zh_name}\n"
+                                    f"📌 英文品名：{en_name}\n"
+                                    f"📄 適應症：{indication}\n"
+                                    f"⚠️ 副作用：{side_effects}"
+                                )
                             else:
-                                reply_text = ""
-                                for row in rows:
-                                    reply_text += (
-                                        f"🔹 中文品名：{row[0]}\n"
-                                        f"📌 英文品名：{row[1]}\n"
-                                        f"📄 適應症：{row[2]}\n\n"
-                                    )
+                                # 全部請AI生成
+                                prompt = (
+                                    f"請用以下格式，幫我介紹藥品「{medicine_name}」，若無法查到請盡量推測：\n"
+                                    "🔹 中文品名：\n"
+                                    "📌 英文品名：\n"
+                                    "📄 適應症：\n"
+                                    "⚠️ 副作用："
+                                )
+                                try:
+                                    ai_resp = chat.generate_content(prompt)
+                                    reply_text = ai_resp.text
+                                except Exception as e:
+                                    reply_text = f"AI 回答失敗：{e}"
+
                         except Exception as e:
                             reply_text = f"⚠️ 查詢資料時發生錯誤：{str(e)}"
 
@@ -182,6 +249,7 @@ def callback():
                         )
                         messaging_api.reply_message(reply_message_request=reply_request)
 
+                # 處理位置訊息（查詢附近藥局）
                 elif event.message.type == "location":
                     user_lat = event.message.latitude
                     user_lng = event.message.longitude
@@ -254,8 +322,47 @@ def callback():
                     )
                     messaging_api.reply_message(reply_message_request=reply_request)
 
+                # 圖片訊息：用 Gemini AI 以藥品格式解釋圖片
+                elif event.message.type == "image":
+                    try:
+                        content = blob_api.get_message_content(message_id=event.message.id)
+                        with tempfile.NamedTemporaryFile(dir=static_tmp_path, suffix=".jpg", delete=False) as tf:
+                            tf.write(content)
+                            filename = os.path.basename(tf.name)
+                        image_url = f"https://{base_url}/images/{filename}"
+                        image = Image.open(tf.name)
+
+                        # Gemini 圖片說明（指定格式，四欄都AI產生）
+                        prompt = (
+                            "請根據這張圖片判斷藥品資訊，並用以下格式回答，若無法判斷請盡量推測：\n"
+                            "🔹 中文品名：\n"
+                            "📌 英文品名：\n"
+                            "📄 適應症：\n"
+                            "⚠️ 副作用："
+                        )
+                        response = chat.generate_content([image, prompt])
+                        description = response.text
+
+                        reply_request = ReplyMessageRequest(
+                            reply_token=event.reply_token,
+                            messages=[
+                                ImageMessage(
+                                    original_content_url=image_url,
+                                    preview_image_url=image_url
+                                ),
+                                TextMessage(text=description)
+                            ]
+                        )
+                        messaging_api.reply_message(reply_message_request=reply_request)
+                    except Exception as e:
+                        reply_request = ReplyMessageRequest(
+                            reply_token=event.reply_token,
+                            messages=[TextMessage(text=f"圖片處理失敗：{e}")]
+                        )
+                        messaging_api.reply_message(reply_message_request=reply_request)
+
     return "OK"
 
 if __name__ == "__main__":
     port = int(os.environ.get("PORT", 7860)) # 讀取環境變數 PORT，預設為 7860
-    app.run(host="0.0.0.0", port=port, debug=False)
+    app.run(host="0.0.0.0", port=port, debug=False)