Update main.py

main.py

@@ -10,6 +10,8 @@ import base64
 from collections import defaultdict
 import uvicorn
 import logging
+from PIL import Image
+import io
 
 # 設置日誌記錄
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -31,7 +33,8 @@ user_states = defaultdict(lambda: {
     "is_ready_for_outfit": False, # 標記衣物是否收集完畢
     "is_ready_for_photo": False, # 標記個人照片是否上傳
     "user_info": {}, # 儲存身高、三圍、場合
-    "personal_photo": None # 儲存個人照片
+    "personal_photo": None, # 儲存個人照片
+    "personal_photo_base64": None # 新增：儲存個人照片的 Base64 字串
 })
 
 MAX_IMAGES_PER_TYPE = 3
@@ -90,7 +93,8 @@ def upload_to_imgbb(image_base64: str):
         return imgbb_link
     except requests.exceptions.RequestException as e:
         logging.error(f"Error uploading to Imgbb: {e}")
-        logging.error(f"Imgbb API Response: {response.status_code} - {response.text}")
+        if 'response' in locals() and response is not None:
+            logging.error(f"Imgbb API Response: {response.status_code} - {response.text}")
         return None
 
 def get_gemini_response(prompt: str, images: list):
@@ -108,7 +112,7 @@ def get_gemini_response(prompt: str, images: list):
     }
     
     try:
-        response = requests.post(f"{GEMINI_API_URL}?key={GOOGLE_API_KEY}", json=payload, timeout=60) # Increased timeout
+        response = requests.post(f"{GEMINI_API_URL}?key={GOOGLE_API_KEY}", json=payload, timeout=60)
         response.raise_for_status()
         response_data = response.json()
         if 'candidates' in response_data and response_data['candidates']:
@@ -126,7 +130,6 @@ def get_gemini_response(prompt: str, images: list):
 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，
@@ -144,7 +147,7 @@ def handle_text_message(event):
 
     # 新增 重置 功能
     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]
+        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, "personal_photo_base64": None})[user_id]
         line_bot_api.reply_message(
             reply_token,
             TextSendMessage(text="狀態已重置。請重新輸入個人資訊，格式為：身高,胸圍,腰圍,臀圍,場合。例如：165,85,65,90,約會")
@@ -228,7 +231,11 @@ def handle_image_message(event):
         try:
             image_id = event.message.id
             base64_img = get_base64_image(image_id)
-            # 將個人照片上傳到 Imgbb，並儲存 URL
+            
+            # 先將個人照片的 Base64 儲存起來，用於 Gemini
+            user_states[user_id]["personal_photo_base64"] = 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
@@ -249,9 +256,13 @@ def handle_image_message(event):
                 "請在描述中以圖片連結的形式（例如：'https://example.com/outfit1.jpg'）顯示你搭配好的衣物組合。"
             )
             
-            all_images = user_states[user_id]["upper_body_images"] + user_states[user_id]["lower_body_images"] + [user_states[user_id]["personal_photo"]]
+            all_images_base64 = (
+                [get_base64_from_url(url) for url in user_states[user_id]["upper_body_images"]] +
+                [get_base64_from_url(url) for url in user_states[user_id]["lower_body_images"]] +
+                [user_states[user_id]["personal_photo_base64"]]
+            )
             
-            response_text = get_gemini_response(prompt, all_images)
+            response_text = get_gemini_response(prompt, all_images_base64)
             
             # 從 Gemini 建議中找出最佳搭配的衣物圖片（這裡需要更複雜的邏輯，我們使用佔位符）
             # 在實際應用中，您可以設計一個提示，讓 Gemini 返回最佳搭配的圖片索引
@@ -277,7 +288,7 @@ def handle_image_message(event):
             )
 
             # 重置狀態以便下一次使用
-            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]
+            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, "personal_photo_base64": None})[user_id]
             return
             
         except Exception as e: