Update main.py

main.py

@@ -1,118 +1,63 @@
 import json, os
 import logging
-from fastapi import FastAPI, Request, Header, BackgroundTasks, HTTPException, status
 from fastapi.middleware.cors import CORSMiddleware
+from fastapi import FastAPI, Request, Header, BackgroundTasks, HTTPException, status
 import google.generativeai as genai
 from linebot import LineBotApi, WebhookHandler
 from linebot.exceptions import InvalidSignatureError
-from linebot.models import MessageEvent, TextMessage, TextSendMessage, AudioMessage
+from linebot.models import MessageEvent, TextMessage, TextSendMessage, ImageSendMessage, AudioMessage
 from dotenv import load_dotenv
-import asyncio
 
-# 從 .env 檔案載入環境變數
+# Load environment variables from .env file
 load_dotenv()
 
-# 設定日誌記錄
-logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s', encoding='utf-8')
-
-# --- 1. 配置設定 ---
-class GeminiModelManager:
-    """
-    負責管理 Gemini 模型相關操作的類別
-    包括初始化模型、建立對話 session 以及傳送訊息
-    """
-    def __init__(self):
-        """
-        初始化 Gemini 模型管理器
-        載入 Google AI API 金鑰，設定文字生成參數，並建立 Gemini 模型
-        """
-        try:
-            genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
-            self.generation_config = genai.types.GenerationConfig(
-                max_output_tokens=2048, temperature=0.2, top_p=0.5, top_k=16
-            )
-            self.model = genai.GenerativeModel(
-                "gemini-2.0-flash-exp",
-                system_instruction="你是咖啡廳員工，請使用親切的招呼語做為開頭，然後以爽朗愉悅的口氣回答問題。",
-            )
-            logging.info("Gemini 模型配置完成")
-        except Exception as e:
-            logging.error(f"Gemini 模型配置錯誤: {e}")
-            raise
-
-    def create_chat_session(self, history=None):
-        """
-        建立一個新的對話 session
-        若沒有提供歷史紀錄，則使用預設的歷史紀錄
-        """
-        if history is None:
-            history = [
-                {
-                    "role": "user",
-                    "parts": ["hi"],
-                },
-                {
-                    "role": "model",
-                    "parts": ["Hi there! How can I help you today?\n"],
-                },
-            ]
-        return self.model.start_chat(history=history)
-        
-    async def send_message(self, chat_session, prompt):
-        """
-        非同步傳送訊息到 Gemini 模型
-        """
-        try:
-            response = await asyncio.to_thread(chat_session.send_message, prompt)
-            return response
-        except Exception as e:
-            logging.error(f"傳送訊息到 Gemini 時發生錯誤: {e}")
-            raise
-
-class LineBotManager:
-    """
-    負責管理 Line Bot API 相關操作的類別
-    包括初始化 Line Bot API 和回覆訊息
-    """
-    def __init__(self):
-        """
-        初始化 Line Bot API 管理器
-        載入 Line Bot API 金鑰和秘密金鑰
-        """
-        try:
-            self.line_bot_api = LineBotApi(os.getenv("CHANNEL_ACCESS_TOKEN"))
-            self.line_handler = WebhookHandler(os.getenv("CHANNEL_SECRET"))
-            logging.info("Line Bot API 配置完成")
-        except Exception as e:
-            logging.error(f"Line Bot API 配置錯誤: {e}")
-            raise
-            
-    async def reply_message(self, event, message):
-        """
-        非同步回覆訊息到 Line
-        """
-        try:
-            await asyncio.to_thread(self.line_bot_api.reply_message, event.reply_token, message)
-            logging.info(f"成功回覆 Line 訊息: {message}")
-        except Exception as e:
-            logging.error(f"回覆 Line 訊息時發生錯誤: {e}")
-            raise
-        
-# 初始化配置
+# Logging configuration
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+
+
+# --- 1. Configuration ---
+def configure_gemini():
+    try:
+        genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
+        generation_config = genai.types.GenerationConfig(
+            max_output_tokens=2048, temperature=0.2, top_p=0.5, top_k=16
+        )
+        model = genai.GenerativeModel(
+            "gemini-2.0-flash-exp",
+            system_instruction="你是咖啡廳員工，請使用親切的招呼語做為開頭，然後以爽朗愉悅的口氣回答問題。",
+        )
+        logging.info("Gemini model configured successfully.")
+        return model, generation_config
+    except Exception as e:
+        logging.error(f"Error configuring Gemini: {e}")
+        raise
+
+
+def configure_line_bot():
+    try:
+      line_bot_api = LineBotApi(os.getenv("CHANNEL_ACCESS_TOKEN"))
+      line_handler = WebhookHandler(os.getenv("CHANNEL_SECRET"))
+      logging.info("Line Bot API configured successfully.")
+      return line_bot_api, line_handler
+    except Exception as e:
+      logging.error(f"Error configuring Line Bot API: {e}")
+      raise
+
+# Initialize configurations
 try:
-    gemini_manager = GeminiModelManager()
-    line_bot_manager = LineBotManager()
+    model, generation_config = configure_gemini()
+    line_bot_api, line_handler = configure_line_bot()
 except Exception as e:
-    logging.critical(f"初始化配置失敗: {e}")
-    exit(1)  # 如果配置失敗，則終止程式
+    logging.critical(f"Failed to initialize configurations: {e}")
+    exit(1)  # Exit if critical configuration fails
 
-# 從環境變數取得是否啟用對話模式，預設為 true
+# Default talking state from environment variable
 working_status = os.getenv("DEFALUT_TALKING", default="true").lower() == "true"
 
-# --- 2. FastAPI 應用程式 ---
+# --- 2. FastAPI App ---
 app = FastAPI()
 
-# 設定 CORS，允許跨域請求
+# Setup CORS
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -121,12 +66,9 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# --- 3. 路由設定 ---
+# --- 3. Routes ---
 @app.get("/")
 def root():
-    """
-    處理根路徑請求，返回 Line Bot 的標題
-    """
     return {"title": "Line Bot"}
 
 
@@ -134,112 +76,78 @@ def root():
 async def webhook(
     request: Request, background_tasks: BackgroundTasks, x_line_signature=Header(None)
 ):
-    """
-    處理 Line Webhook 請求
-    接收 Line 的訊息並驗證簽章，然後將訊息處理加入背景任務
-    """
     body = await request.body()
     try:
         background_tasks.add_task(
-             self.run_sync_handler, line_bot_manager.line_handler, body.decode("utf-8"), x_line_signature # 修改這裡
+            line_handler.handle, body.decode("utf-8"), x_line_signature
         )
     except InvalidSignatureError:
-        logging.warning("收到無效的簽章")
-        raise HTTPException(status_code=400, detail="無效的簽章")
+        logging.warning("Invalid signature received.")
+        raise HTTPException(status_code=400, detail="Invalid signature")
     return "ok"
 
-# 定義一個方法讓 line_handler 可以正確處理非同步事件
-async def run_sync_handler(handler, body, signature):
-    await asyncio.to_thread(handler.handle, body, signature)
-
-# --- 4. 對話 Session 管理 ---
-class ChatSessionManager:
-    """
-    負責管理對話 Session 的類別
-    包括建立、取得、刪除對話 session，以及處理訊息
-    """
-    def __init__(self):
-        """
-        初始化對話 Session 管理器，建立一個儲存對話 Session 的字典
-        """
-        self.chat_sessions = {}
-
-    def get_or_create_session(self, user_id):
-        """
-        取得或建立特定使用者的對話 Session
-        如果不存在，則建立一個新的 Session
-        """
-        if user_id not in self.chat_sessions:
-            logging.info(f"為使用者 {user_id} 建立新的對話 Session")
-            self.chat_sessions[user_id] = gemini_manager.create_chat_session()
-        return self.chat_sessions[user_id]
-    
-    def remove_session(self, user_id):
-      """
-      刪除特定使用者的對話 Session
-      """
-      if user_id in self.chat_sessions:
-        del self.chat_sessions[user_id]
-        logging.info(f"刪除使用者 {user_id} 的對話 Session")
+# --- 4. Chat Session Management ---
+chat_sessions = {}
+
+def get_or_create_chat_session(user_id: str):
+    if user_id not in chat_sessions:
+        logging.info(f"Creating new chat session for user: {user_id}")
+        chat_sessions[user_id] = model.start_chat(
+            history=[
+                {
+                    "role": "user",
+                    "parts": ["hi"],
+                },
+                {
+                    "role": "model",
+                    "parts": ["Hi there! How can I help you today?\n"],
+                },
+            ]
+        )
+    return chat_sessions[user_id]
+
+def handle_text_message(event):
+    global working_status
+
+    user_id = event.source.user_id
+    if event.message.text == "再見":
+        line_bot_api.reply_message(event.reply_token, TextSendMessage(text="Bye!"))
+        if user_id in chat_sessions:
+            del chat_sessions[user_id]
+            logging.info(f"Chat session deleted for user: {user_id}")
+        return
+
+    if not working_status:
+        logging.info("Bot is not in working status, ignoring message.")
+        return
+
+    try:
+        prompt = event.message.text
+        logging.info(f"Received user prompt: {prompt}, user: {user_id}")
+        chat_session = get_or_create_chat_session(user_id)
+        response = chat_session.send_message(prompt)
+        if response.text:
+            out = response.text
+            logging.info(f"Gemini response: {out}, user: {user_id}")
+        else:
+            out = "Gemini 沒答案！請換個說法！"
+            logging.warning(f"Gemini returned no text response for user: {user_id}")
+    except Exception as e:
+        out = "Gemini 執行出錯！請換個說法！"
+        logging.error(f"Gemini error: {e}, user: {user_id}")
     
-    async def handle_message(self, event):
-        """
-        處理 Line 訊息，取得使用者輸入，並回覆 Gemini 模型的訊息
-        """
-        user_id = event.source.user_id
-        logging.info(f"開始處理來自使用者 {user_id} 的訊息")
-        if event.message.text == "再見":
-            await line_bot_manager.reply_message(event, TextSendMessage(text="Bye!"))
-            self.remove_session(user_id)
-            logging.info(f"使用者 {user_id} 說再見，已刪除對話 Session")
-            return
-        
-        if not working_status:
-           logging.info(f"對話模式關閉，忽略來自使用者 {user_id} 的訊息")
-           return
-        
-        try:
-           prompt = event.message.text
-           logging.info(f"收到使用者 {user_id} 的訊息: {prompt}")
-           chat_session = self.get_or_create_session(user_id)
-           response = await gemini_manager.send_message(chat_session, prompt)
-           if response.text:
-                out = response.text
-                logging.info(f"Gemini 回覆使用者 {user_id}: {out}")
-           else:
-                out = "Gemini 沒答案！請換個說法！"
-                logging.warning(f"Gemini 沒有回覆文字內容給使用者 {user_id}")
-           
-           await line_bot_manager.reply_message(event, TextSendMessage(text=out))
-
-        except Exception as e:
-           out = "Gemini 執行出錯！請換個說法！"
-           logging.error(f"Gemini 發生錯誤 (使用者: {user_id}): {e}")
-           try:
-                await line_bot_manager.reply_message(event, TextSendMessage(text=out))
-           except Exception as e:
-                logging.error(f"回覆使用者時發生錯誤: {e}")
-        logging.info(f"訊息處理完成 (使用者: {user_id})")
-
-
-chat_session_manager = ChatSessionManager()
-
-@line_bot_manager.line_handler.add(MessageEvent, message=TextMessage)
-async def handle_message(event):
-    """
-    Line Bot 訊息處理的主要函式
-    驗證訊息類型，然後將文字訊息轉發給對話管理員處理
-    """
-    logging.info(f"收到來自 {event.source.user_id} 的訊息，訊息類型:{event.message.type}")
+    line_bot_api.reply_message(event.reply_token, TextSendMessage(text=out))
+
+@line_handler.add(MessageEvent, message=TextMessage)
+def handle_message(event):
     if event.type != "message" or event.message.type != "text":
-        logging.warning(f"事件類型錯誤：不是文字訊息 (使用者: {event.source.user_id})")
-        await line_bot_manager.reply_message(event, TextSendMessage(text="事件類型錯誤：不是文字訊息。"))
+        logging.warning("Event type error: not a text message.")
+        line_bot_api.reply_message(event.reply_token, TextSendMessage(text="Event type error: not a text message."))
         return
 
-    await chat_session_manager.handle_message(event)
-    logging.info(f"完成來自 {event.source.user_id} 的訊息處理")
+    handle_text_message(event)
 
-# --- 5. 啟動應用程式 ---
+# --- 5. Start App ---
 if __name__ == "__main__":
     import uvicorn