Update main.py

main.py

@@ -9,10 +9,15 @@ import requests
 import base64
 from collections import defaultdict
 import uvicorn
+import logging
+
+# 設置日誌記錄
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
 # 檢查環境變數
-if not all(k in os.environ for k in ["CHANNEL_ACCESS_TOKEN", "CHANNEL_SECRET", "GOOGLE_API_KEY", "IMGUR_CLIENT_ID"]):
-    raise ValueError("Missing environment variables. Please set CHANNEL_ACCESS_TOKEN, CHANNEL_SECRET, GOOGLE_API_KEY, and IMGUR_CLIENT_ID.")
+if not all(k in os.environ for k in ["CHANNEL_ACCESS_TOKEN", "CHANNEL_SECRET", "GOOGLE_API_KEY", "IMGBB_API_KEY"]):
+    logging.error("Missing environment variables. Please set CHANNEL_ACCESS_TOKEN, CHANNEL_SECRET, GOOGLE_API_KEY, and IMGBB_API_KEY.")
+    raise ValueError("Missing environment variables. Please set CHANNEL_ACCESS_TOKEN, CHANNEL_SECRET, GOOGLE_API_KEY, and IMGBB_API_KEY.")
 
 # Line Bot API 設定
 line_bot_api = LineBotApi(os.environ["CHANNEL_ACCESS_TOKEN"])
@@ -32,7 +37,7 @@ user_states = defaultdict(lambda: {
 MAX_IMAGES_PER_TYPE = 3
 GEMINI_API_URL = "https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash-preview-05-20:generateContent"
 GOOGLE_API_KEY = os.environ["GOOGLE_API_KEY"]
-IMGUR_CLIENT_ID = os.environ["IMGUR_CLIENT_ID"]
+IMGBB_API_KEY = os.environ["IMGBB_API_KEY"]
 
 app = FastAPI()
 
@@ -62,25 +67,33 @@ async def webhook(
     return "ok"
 
 def get_base64_image(message_id: str):
+    logging.info(f"Fetching image content for message ID: {message_id}")
     message_content = line_bot_api.get_message_content(message_id)
     image_bytes = message_content.content
     return base64.b64encode(image_bytes).decode('utf-8')
 
-def upload_to_imgur(image_base64: str):
+def upload_to_imgbb(image_base64: str):
     """
-    將 Base64 編碼的圖片上傳到 Imgur。
+    將 Base64 編碼的圖片上傳到 Imgbb。
     """
-    headers = {"Authorization": f"Client-ID {IMGUR_CLIENT_ID}"}
-    payload = {"image": image_base64}
+    logging.info("Attempting to upload image to Imgbb.")
+    url = "https://api.imgbb.com/1/upload"
+    payload = {
+        "key": IMGBB_API_KEY,
+        "image": image_base64
+    }
     try:
-        response = requests.post("https://api.imgur.com/3/image", headers=headers, data=payload)
+        response = requests.post(url, data=payload)
         response.raise_for_status()
-        return response.json()["data"]["link"]
+        imgbb_link = response.json()["data"]["url"]
+        logging.info(f"Image successfully uploaded to Imgbb: {imgbb_link}")
+        return imgbb_link
     except requests.exceptions.RequestException as e:
-        print(f"Error uploading to Imgur: {e}")
+        logging.error(f"Error uploading to Imgbb: {e}")
         return None
 
 def get_gemini_response(prompt: str, images: list):
+    logging.info("Sending request to Gemini API.")
     payload = {
         "contents": [
             {
@@ -93,17 +106,28 @@ def get_gemini_response(prompt: str, images: list):
         ]
     }
     
-    response = requests.post(f"{GEMINI_API_URL}?key={GOOGLE_API_KEY}", json=payload)
-    if response.status_code == 200:
-        return response.json()['candidates'][0]['content']['parts'][0]['text']
-    else:
-        return f"Gemini API 請求失敗：{response.status_code}, {response.text}"
+    try:
+        response = requests.post(f"{GEMINI_API_URL}?key={GOOGLE_API_KEY}", json=payload, timeout=60) # Increased timeout
+        response.raise_for_status()
+        response_data = response.json()
+        if 'candidates' in response_data and response_data['candidates']:
+            gemini_text = response_data['candidates'][0]['content']['parts'][0]['text']
+            logging.info("Successfully received response from Gemini.")
+            return gemini_text
+        else:
+            logging.error(f"Gemini API response has no candidates: {response.text}")
+            return "Gemini API 暫時無法提供服務，請稍後再試。"
+    except requests.exceptions.RequestException as e:
+        logging.error(f"Gemini API request failed: {e}")
+        return f"Gemini API 請求失敗：{e}"
+
 
 def get_virtual_tryon_image(user_photo, upper_body_image, lower_body_image):
     """
     模擬虛擬試穿 API，返回一個試穿後的圖片 URL。
     在實際應用中，這裡會呼叫一個運行 IDM-VTON 的獨立服務。
     """
+    logging.info("Simulating virtual try-on and returning a placeholder image.")
     # 這裡的邏輯需要您自行實現，可以是呼叫一個公開的 API，
     # 或者將圖片傳輸到一個運行 IDM-VTON 的伺服器。
     # 由於沒有實際的 IDM-VTON 服務，我們使用佔位符圖片來展示功能。
@@ -115,12 +139,14 @@ def handle_text_message(event):
     text = event.message.text.lower()
     reply_token = event.reply_token
 
+    logging.info(f"Received text message from user {user_id}: '{text}'")
+
     # 新增 重置 功能
-    if text == "重置" or text == "重來" or text == "重新開始" or text == "再一次":
+    if text in ["重置", "重來", "重新��始", "再一次"]:
         user_states[user_id] = defaultdict(lambda: {"upper_body_images": [], "lower_body_images": [], "current_mode": None, "is_ready_for_outfit": False, "is_ready_for_photo": False, "user_info": {}, "personal_photo": None})[user_id]
         line_bot_api.reply_message(
             reply_token,
-            TextSendMessage(text="狀態已重置。請重新輸入個人資訊，格式為：身高,胸圍,腰圍,臀圍,場合。例如：165,85,65,90,約會")
+            TextSendMessage(text="狀態已重置。請重新輸入個人資訊，格式為：身高,胸圍,腰圍,臀圍,場合。")
         )
         return
     
@@ -150,6 +176,7 @@ def handle_text_message(event):
                 )
                 return
         except ValueError:
+            logging.error(f"Invalid user info format: '{text}'")
             line_bot_api.reply_message(
                 reply_token,
                 TextSendMessage(text="請依照格式輸入：身高,三圍胸圍,腰圍,臀圍,場合。例如：165,85,65,90,約會")
@@ -184,8 +211,11 @@ def handle_image_message(event):
     user_id = event.source.user_id
     reply_token = event.reply_token
     
+    logging.info(f"Received image message from user {user_id}")
+
     # 檢查是否已準備好處理圖片
     if not user_states[user_id]["is_ready_for_outfit"]:
+        logging.warning("User is not in outfit collection mode.")
         line_bot_api.reply_message(
             reply_token,
             TextSendMessage(text="請先輸入個人資訊，格式為：身高,胸圍,腰圍,臀圍,場合。例如：165,85,65,90,約會")
@@ -197,8 +227,8 @@ def handle_image_message(event):
         try:
             image_id = event.message.id
             base64_img = get_base64_image(image_id)
-            # 將個人照片上傳到 Imgur，並儲存 URL
-            photo_url = upload_to_imgur(base64_img)
+            # 將個人照片上傳到 Imgbb，並儲存 URL
+            photo_url = upload_to_imgbb(base64_img)
             user_states[user_id]["personal_photo"] = photo_url
             user_states[user_id]["is_ready_for_photo"] = True
             
@@ -250,6 +280,7 @@ def handle_image_message(event):
             return
             
         except Exception as e:
+            logging.error(f"Error processing personal photo: {e}")
             line_bot_api.reply_message(
                 reply_token,
                 TextSendMessage(text=f"圖片處理失敗，請稍後再試。錯誤：{e}")
@@ -259,6 +290,7 @@ def handle_image_message(event):
     # 處理衣物圖片上傳
     mode = user_states[user_id]["current_mode"]
     if not mode:
+        logging.warning("User is not in a defined upload mode.")
         line_bot_api.reply_message(
             reply_token,
             TextSendMessage(text="請先傳送「上衣」或「褲子」來選擇要上傳的衣服類型。")
@@ -268,12 +300,13 @@ def handle_image_message(event):
     try:
         image_id = event.message.id
         base64_img = get_base64_image(image_id)
-        # 上傳圖片到 Imgur，並儲存 URL
-        image_url = upload_to_imgur(base64_img)
+        # 上傳圖片到 Imgbb，並儲存 URL
+        image_url = upload_to_imgbb(base64_img)
         if not image_url:
+            logging.error("Failed to upload image to Imgbb.")
             line_bot_api.reply_message(
                 reply_token,
-                TextSendMessage(text="圖片上傳 Imgur 失敗，請稍後再試。")
+                TextSendMessage(text="圖片上傳失敗，請稍後再試。")
             )
             return
             
@@ -281,6 +314,7 @@ def handle_image_message(event):
             if len(user_states[user_id]["upper_body_images"]) < MAX_IMAGES_PER_TYPE:
                 user_states[user_id]["upper_body_images"].append(image_url)
             else:
+                logging.info("Upper body image count is full.")
                 line_bot_api.reply_message(
                     reply_token,
                     TextSendMessage(text="上衣數量已滿，請傳送「褲子」來上傳褲子圖片。")
@@ -290,6 +324,7 @@ def handle_image_message(event):
             if len(user_states[user_id]["lower_body_images"]) < MAX_IMAGES_PER_TYPE:
                 user_states[user_id]["lower_body_images"].append(image_url)
             else:
+                logging.info("Lower body image count is full.")
                 line_bot_api.reply_message(
                     reply_token,
                     TextSendMessage(text="褲子數量已滿，請傳送「上衣」來上傳上衣圖片。")
@@ -305,12 +340,14 @@ def handle_image_message(event):
                 TextSendMessage(text=f"已接收。上衣: {upper_count}/{MAX_IMAGES_PER_TYPE}，褲子/裙子: {lower_count}/{MAX_IMAGES_PER_TYPE}。")
             )
         else:
+            logging.info("All outfit images received. Prompting for personal photo.")
             line_bot_api.reply_message(
                 reply_token,
                 TextSendMessage(text="已收到所有衣物圖片！接下來，請上傳一張您個人的全身照片，以便進行虛擬試穿。")
             )
             
     except Exception as e:
+        logging.error(f"Error processing outfit image: {e}")
         line_bot_api.reply_message(
             reply_token,
             TextSendMessage(text=f"圖片處理失敗，請稍後再試。錯誤：{e}")