Update main.py

main.py

@@ -84,6 +84,16 @@ def get_gemini_response(prompt: str, images: list):
     else:
         return f"Gemini API 請求失敗：{response.status_code}, {response.text}"
 
+def get_virtual_tryon_image(user_photo, upper_body_image, lower_body_image):
+    """
+    模擬虛擬試穿 API，返回一個試穿後的圖片 URL。
+    在實際應用中，這裡會呼叫一個運行 IDM-VTON 的獨立服務。
+    """
+    # 這裡的邏輯需要您自行實現，可以是呼叫一個公開的 API，
+    # 或者將圖片傳輸到一個運行 IDM-VTON 的伺服器。
+    # 由於沒有實際的 IDM-VTON 服務，我們使用佔位符圖片來展示功能。
+    return "https://placehold.co/1024x1024?text=Virtual+Try-On+Outfit"
+
 @line_handler.add(MessageEvent, message=TextMessage)
 def handle_text_message(event):
     user_id = event.source.user_id
@@ -142,14 +152,14 @@ def handle_text_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]
             line_bot_api.reply_message(
                 reply_token,
-                TextSendMessage(text="狀態已重置。請重新輸入個人資訊，格式為：身高,胸圍,腰圍,臀圍,場合，例如：165,85,65,90,約會")
+                TextSendMessage(text="狀態已重置。請重新輸入個人資訊，格式為：身高,胸圍,腰圍,臀圍,場合")
             )
             return
 
     # 如果沒有進入任何模式，給予提示
     line_bot_api.reply_message(
         reply_token,
-        TextSendMessage(text="請先依照格式輸入個人資訊，格式為：身高,胸圍,腰圍,臀圍,場合。例如：165,85,65,90,約會")
+        TextSendMessage(text="請先依照格式輸入個人資訊，格式為：身高,胸圍,腰圍,臀圍,場合。")
     )
 
 @line_handler.add(MessageEvent, message=ImageMessage)
@@ -161,12 +171,12 @@ def handle_image_message(event):
     if not user_states[user_id]["is_ready_for_outfit"]:
         line_bot_api.reply_message(
             reply_token,
-            TextSendMessage(text="請先輸入個人資訊，格式為：身高,胸圍,腰圍,臀圍,場合。例如：165,85,65,90,約會")
+            TextSendMessage(text="請先輸入個人資訊，格式為：身高,胸圍,腰圍,臀圍,場合。")
         )
         return
 
-    # 如果個人照片還沒上傳
-    if not user_states[user_id]["personal_photo"]:
+    # 處理個人照片上傳
+    if len(user_states[user_id]["upper_body_images"]) == MAX_IMAGES_PER_TYPE and len(user_states[user_id]["lower_body_images"]) == MAX_IMAGES_PER_TYPE and not user_states[user_id]["personal_photo"]:
         try:
             image_id = event.message.id
             base64_img = get_base64_image(image_id)
@@ -178,22 +188,30 @@ def handle_image_message(event):
                 TextSendMessage(text="已收到您的個人照片，正在為您準備穿搭... 請稍候。")
             )
             
-            # 開始生成搭配建議和試穿照片
+            # 準備 Gemini 提示
             user_info = user_states[user_id]["user_info"]
             occasion = user_info["occasion"]
             
             prompt = (
-                f"我提供了三件上衣圖片和三件下半身圖片，以及使用者的一張個人照片。使用者的身高為 {user_info['height']}，三圍為 {user_info['bust']}-{user_info['waist']}-{user_info['hip']}，想要參加的場合是「{occasion}」。"
+                f"我提供了三件上衣圖片和三件下半身圖片，以及使用者的一張個人照片。使用者資訊：身高 {user_info['height']}cm，三圍 {user_info['bust']}-{user_info['waist']}-{user_info['hip']}，場合是「{occasion}」。"
                 "請根據這些衣物，為我推薦一套最適合的穿搭，並詳細說明為何這套搭配適合這個場合。請以繁體中文回答。"
-                "接著，請生成一張虛擬試穿的照片，將推薦的上衣和下衣搭配到使用者提供的照片上。由於模型限制，我會使用佔位符圖片來模擬試穿效果。"
+                "請在描述中以圖片連結的形式（例如：'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"]]
             
             response_text = get_gemini_response(prompt, all_images)
             
-            # 虛擬試穿照片佔位符
-            virtual_try_on_url = "https://placehold.co/1024x1024?text=Virtual+Try-On+Outfit"
+            # 從 Gemini 建議中找出最佳搭配的衣物圖片（這裡需要更複雜的邏輯，我們使用佔位符）
+            # 在實際應用中，您可以設計一個提示，讓 Gemini 返回最佳搭配的圖片索引
+            best_upper_index = 0
+            best_lower_index = 0
+
+            best_upper_body_image = user_states[user_id]["upper_body_images"][best_upper_index]
+            best_lower_body_image = user_states[user_id]["lower_body_images"][best_lower_index]
+
+            # 呼叫虛擬試穿 API，並取得結果圖片 URL
+            virtual_try_on_url = get_virtual_tryon_image(user_states[user_id]["personal_photo"], best_upper_body_image, best_lower_body_image)
             
             # 發送 Gemini 的文字建議
             line_bot_api.push_message(
@@ -261,7 +279,7 @@ def handle_image_message(event):
         else:
             line_bot_api.reply_message(
                 reply_token,
-                TextSendMessage(text="已收到所有衣物圖片！接下來，請上傳一張您個人的全身照片，以便進行虛擬試穿。如果想重來請輸入：重置")
+                TextSendMessage(text="已收到所有衣物圖片！接下來，請上傳一張您個人的全身照片，以便進行虛擬試穿。")
             )
             
     except Exception as e: