Update main.py

main.py

@@ -8,15 +8,19 @@ from linebot.exceptions import InvalidSignatureError
 from linebot.models import MessageEvent, TextMessage, TextSendMessage, ImageSendMessage, AudioMessage
 from dotenv import load_dotenv
 
-# Load environment variables from .env file
+# 從 .env 檔案載入環境變數
 load_dotenv()
 
-# Logging configuration
-logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+# 設定日誌記錄
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s', encoding='utf-8')
 
 
-# --- 1. Configuration ---
+# --- 1. 配置設定 ---
 def configure_gemini():
+    """
+    配置 Google Gemini 模型
+    設定 API 金鑰、生成參數和模型
+    """
     try:
         genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
         generation_config = genai.types.GenerationConfig(
@@ -26,38 +30,42 @@ def configure_gemini():
             "gemini-2.0-flash-exp",
             system_instruction="你是咖啡廳員工，請使用親切的招呼語做為開頭，然後以爽朗愉悅的口氣回答問題。",
         )
-        logging.info("Gemini model configured successfully.")
+        logging.info("Gemini 模型配置完成.")
         return model, generation_config
     except Exception as e:
-        logging.error(f"Error configuring Gemini: {e}")
+        logging.error(f"Gemini 模型配置錯誤: {e}")
         raise
 
 
 def configure_line_bot():
+    """
+    配置 Line Bot API
+    設定 Channel Access Token 和 Channel Secret
+    """
     try:
       line_bot_api = LineBotApi(os.getenv("CHANNEL_ACCESS_TOKEN"))
       line_handler = WebhookHandler(os.getenv("CHANNEL_SECRET"))
-      logging.info("Line Bot API configured successfully.")
+      logging.info("Line Bot API 配置完成.")
       return line_bot_api, line_handler
     except Exception as e:
-      logging.error(f"Error configuring Line Bot API: {e}")
+      logging.error(f"Line Bot API 配置錯誤: {e}")
       raise
 
-# Initialize configurations
+# 初始化配置
 try:
     model, generation_config = configure_gemini()
     line_bot_api, line_handler = configure_line_bot()
 except Exception as e:
-    logging.critical(f"Failed to initialize configurations: {e}")
-    exit(1)  # Exit if critical configuration fails
+    logging.critical(f"初始化配置失敗: {e}")
+    exit(1)  # 如果配置失敗，則終止程式
 
-# Default talking state from environment variable
+# 從環境變數取得是否啟用對話模式，預設為 true
 working_status = os.getenv("DEFALUT_TALKING", default="true").lower() == "true"
 
-# --- 2. FastAPI App ---
+# --- 2. FastAPI 應用程式 ---
 app = FastAPI()
 
-# Setup CORS
+# 設定 CORS，允許跨域請求
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -66,9 +74,12 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# --- 3. Routes ---
+# --- 3. 路由設定 ---
 @app.get("/")
 def root():
+    """
+    處理根路徑請求，返回 Line Bot 的標題
+    """
     return {"title": "Line Bot"}
 
 
@@ -76,22 +87,30 @@ 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(
             line_handler.handle, body.decode("utf-8"), x_line_signature
         )
     except InvalidSignatureError:
-        logging.warning("Invalid signature received.")
-        raise HTTPException(status_code=400, detail="Invalid signature")
+        logging.warning("收到無效的簽章.")
+        raise HTTPException(status_code=400, detail="無效的簽章")
     return "ok"
 
-# --- 4. Chat Session Management ---
+# --- 4. 對話 Session 管理 ---
 chat_sessions = {}
 
 def get_or_create_chat_session(user_id: str):
+    """
+    取得或建立特定使用者的對話 Session
+    如果不存在，則建立一個新的 Session
+    """
     if user_id not in chat_sessions:
-        logging.info(f"Creating new chat session for user: {user_id}")
+        logging.info(f"為使用者 {user_id} 建立新的對話 Session")
         chat_sessions[user_id] = model.start_chat(
             history=[
                 {
@@ -107,6 +126,9 @@ def get_or_create_chat_session(user_id: str):
     return chat_sessions[user_id]
 
 def handle_text_message(event):
+    """
+    處理文字訊息，取得使用者輸入，並回覆 Gemini 模型的訊息
+    """
     global working_status
 
     user_id = event.source.user_id
@@ -114,40 +136,44 @@ def handle_text_message(event):
         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}")
+            logging.info(f"使用者 {user_id} 說再見，已刪除對話 Session")
         return
 
     if not working_status:
-        logging.info("Bot is not in working status, ignoring message.")
+        logging.info("對話模式關閉，忽略訊息.")
         return
 
     try:
         prompt = event.message.text
-        logging.info(f"Received user prompt: {prompt}, user: {user_id}")
+        logging.info(f"收到使用者 {user_id} 的訊息: {prompt}")
         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}")
+            logging.info(f"Gemini 回覆使用者 {user_id}: {out}")
         else:
             out = "Gemini 沒答案！請換個說法！"
-            logging.warning(f"Gemini returned no text response for user: {user_id}")
+            logging.warning(f"Gemini 沒有回覆文字內容給使用者 {user_id}")
     except Exception as e:
         out = "Gemini 執行出錯！請換個說法！"
-        logging.error(f"Gemini error: {e}, user: {user_id}")
+        logging.error(f"Gemini 發生錯誤 (使用者: {user_id}): {e}")
     
     line_bot_api.reply_message(event.reply_token, TextSendMessage(text=out))
 
 @line_handler.add(MessageEvent, message=TextMessage)
 def handle_message(event):
+    """
+    Line Bot 訊息處理的主要函式
+    驗證訊息類型，然後將文字訊息轉發給處理函數
+    """
     if event.type != "message" or event.message.type != "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."))
+        logging.warning("事件類型錯誤：不是文字訊息.")
+        line_bot_api.reply_message(event.reply_token, TextSendMessage(text="事件類型錯誤：不是文字訊息。"))
         return
 
     handle_text_message(event)
 
-# --- 5. Start App ---
+# --- 5. 啟動應用程式 ---
 if __name__ == "__main__":
     import uvicorn