Update main.py

main.py

@@ -23,11 +23,13 @@ user_states = defaultdict(lambda: {
     "upper_body_images": [],
     "lower_body_images": [],
     "current_mode": None, # "upper" or "lower"
-    "is_ready": False # 新增狀態，標記圖片是否收集完畢
+    "is_ready_for_outfit": False, # 標記衣物是否收集完畢
+    "is_ready_for_photo": False, # 標記個人照片是否上傳
+    "user_info": {}, # 儲存身高、三圍、場合
+    "personal_photo": None # 儲存個人照片
 })
 
 MAX_IMAGES_PER_TYPE = 3
-MAX_TOTAL_IMAGES = MAX_IMAGES_PER_TYPE * 2
 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"]
 
@@ -88,103 +90,136 @@ def handle_text_message(event):
     text = event.message.text.lower()
     reply_token = event.reply_token
     
-    # 檢查是否已準備好處理場合選擇
-    if user_states[user_id]["is_ready"]:
-        prompt = (
-            f"我提供了三件上衣圖片和三件下半身圖片。請根據這些衣服，為我推薦一個適合「{text}」的穿搭建議，並詳細描述你推薦的上衣與下衣組合。請以繁體中文回答。"
-            "請在描述中以圖片連結的形式（例如：'https://example.com/outfit1.jpg'）顯示你搭配好的衣物組合。"
-            "請將圖片連結放在一個新行，並使用 markdown 語法。"
-        )
-        
-        all_images = user_states[user_id]["upper_body_images"] + user_states[user_id]["lower_body_images"]
-        
-        # 準備要回覆的訊息列表
-        messages_to_send = [
-            TextSendMessage(text="已收到場合資訊，正在為您生成穿搭建議... 請稍候。")
-        ]
-        
-        # 在這裡顯示使用者上傳的原始圖片
-        messages_to_send.append(
-            TextSendMessage(text="這是您上傳的所有衣物：")
-        )
-        
-        # 使用一個佔位符來顯示所有上傳的圖片
-        upper_placeholders = [
-            f"https://placehold.co/1024x1024?text=Upper_Body_{i+1}" for i in range(MAX_IMAGES_PER_TYPE)
-        ]
-        lower_placeholders = [
-            f"https://placehold.co/1024x1024?text=Lower_Body_{i+1}" for i in range(MAX_IMAGES_PER_TYPE)
-        ]
-        
-        for url in upper_placeholders:
-            messages_to_send.append(ImageSendMessage(original_content_url=url, preview_image_url=url))
-        
-        for url in lower_placeholders:
-            messages_to_send.append(ImageSendMessage(original_content_url=url, preview_image_url=url))
-        
-        line_bot_api.push_message(
-            user_id,
-            messages_to_send
-        )
-        
-        response_text = get_gemini_response(prompt, all_images)
-        
-        # 解析 Gemini 的回應，將圖片 URL 和文字分開
-        lines = response_text.split('\n')
-        gemini_messages = []
-        for line in lines:
-            if line.startswith('http'):
-                # 這是圖片 URL，將其轉為 ImageSendMessage
-                gemini_messages.append(ImageSendMessage(original_content_url=line, preview_image_url=line))
-            else:
-                # 這是文字，將其轉為 TextSendMessage
-                gemini_messages.append(TextSendMessage(text=line))
-        
-        line_bot_api.push_message(
-            user_id,
-            TextSendMessage(text="這是為您推薦的搭配：")
-        )
+    # 處理個人資訊輸入
+    if not user_states[user_id]["is_ready_for_photo"] and not user_states[user_id]["is_ready_for_outfit"]:
+        try:
+            parts = text.split(',')
+            if len(parts) == 5:
+                height = float(parts[0])
+                bust = float(parts[1])
+                waist = float(parts[2])
+                hip = float(parts[3])
+                occasion = parts[4].strip()
+                
+                user_states[user_id]["user_info"] = {
+                    "height": height,
+                    "bust": bust,
+                    "waist": waist,
+                    "hip": hip,
+                    "occasion": occasion
+                }
+                user_states[user_id]["is_ready_for_outfit"] = True
+                
+                line_bot_api.reply_message(
+                    reply_token,
+                    TextSendMessage(text=f"已收到您的資訊！接下來請上傳三件上衣和三件褲子的圖片。請先輸入「上衣」或「褲子」來開始。")
+                )
+                return
+        except ValueError:
+            line_bot_api.reply_message(
+                reply_token,
+                TextSendMessage(text="請依照格式輸入：身高,三圍胸圍,腰圍,臀圍,場合。例如：165,85,65,90,約會")
+            )
+            return
 
-        line_bot_api.push_message(
-            user_id,
-            gemini_messages
-        )
-        
-        # 重置狀態以便下一次使用
-        user_states[user_id] = defaultdict(lambda: {"upper_body_images": [], "lower_body_images": [], "current_mode": None, "is_ready": False})[user_id]
-        return
-        
-    if text == "上衣":
-        user_states[user_id]["current_mode"] = "upper"
-        line_bot_api.reply_message(
-            reply_token,
-            TextSendMessage(text=f"請上傳三件上衣圖片，您已上傳 {len(user_states[user_id]['upper_body_images'])}/{MAX_IMAGES_PER_TYPE} 張。")
-        )
-    elif text == "褲子":
-        user_states[user_id]["current_mode"] = "lower"
-        line_bot_api.reply_message(
-            reply_token,
-            TextSendMessage(text=f"請上傳三件褲子/裙子圖片，您已上傳 {len(user_states[user_id]['lower_body_images'])}/{MAX_IMAGES_PER_TYPE} 張。")
-        )
-    elif text == "重置":
-        user_states[user_id] = defaultdict(lambda: {"upper_body_images": [], "lower_body_images": [], "current_mode": None, "is_ready": False})[user_id]
-        line_bot_api.reply_message(
-            reply_token,
-            TextSendMessage(text="狀態已重置。請傳送「上衣」或「褲子」來開始上傳。")
-        )
-    else:
-        line_bot_api.reply_message(
-            reply_token,
-            TextSendMessage(text="請先傳送「上衣」或「褲子」來選擇要上傳的衣服類型。")
-        )
+    # 處理衣物上傳模式
+    if user_states[user_id]["is_ready_for_outfit"]:
+        if text == "上衣":
+            user_states[user_id]["current_mode"] = "upper"
+            line_bot_api.reply_message(
+                reply_token,
+                TextSendMessage(text=f"請上傳三件上衣圖片，您已上傳 {len(user_states[user_id]['upper_body_images'])}/{MAX_IMAGES_PER_TYPE} 張。")
+            )
+            return
+        elif text == "褲子":
+            user_states[user_id]["current_mode"] = "lower"
+            line_bot_api.reply_message(
+                reply_token,
+                TextSendMessage(text=f"請上傳三件褲子/裙子圖片，您已上傳 {len(user_states[user_id]['lower_body_images'])}/{MAX_IMAGES_PER_TYPE} 張。")
+            )
+            return
+        elif text == "重置":
+            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="狀態已重置。請重新輸入個人資訊，格式為：身高,胸圍,腰圍,臀圍,場合")
+            )
+            return
+
+    # 如果沒有進入任何模式，給予提示
+    line_bot_api.reply_message(
+        reply_token,
+        TextSendMessage(text="請先依照格式輸入個人資訊，格式為：身高,胸圍,腰圍,臀圍,場合。")
+    )
 
 @line_handler.add(MessageEvent, message=ImageMessage)
 def handle_image_message(event):
     user_id = event.source.user_id
     reply_token = event.reply_token
     
+    # 檢查是否已準備好處理圖片
+    if not user_states[user_id]["is_ready_for_outfit"]:
+        line_bot_api.reply_message(
+            reply_token,
+            TextSendMessage(text="請先輸入個人資訊，格式為：身高,胸圍,腰圍,臀圍,場合。")
+        )
+        return
+
+    # 如果個人照片還沒上傳
+    if not user_states[user_id]["personal_photo"]:
+        try:
+            image_id = event.message.id
+            base64_img = get_base64_image(image_id)
+            user_states[user_id]["personal_photo"] = base64_img
+            user_states[user_id]["is_ready_for_photo"] = True
+            
+            line_bot_api.reply_message(
+                reply_token,
+                TextSendMessage(text="已收到您的個人照片，正在為您準備穿搭... 請稍候。")
+            )
+            
+            # 開始生成搭配建議和試穿照片
+            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}」。"
+                "請根據這些衣物，為我推薦一套最適合的穿搭，並詳細說明為何這套搭配適合這個場合。請以繁體中文回答。"
+                "接著，請生成一張虛擬試穿的照片，將推薦的上衣和下衣搭配到使用者提供的照片上。由於模型限制，我會使用佔位符圖片來模擬試穿效果。"
+            )
+            
+            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 的文字建議
+            line_bot_api.push_message(
+                user_id,
+                TextSendMessage(text=f"這是為您推薦的搭配：\n\n{response_text}")
+            )
+            
+            # 發送虛擬試穿照片
+            line_bot_api.push_message(
+                user_id,
+                ImageSendMessage(original_content_url=virtual_try_on_url, preview_image_url=virtual_try_on_url)
+            )
+
+            # 重置狀態以便下一次使用
+            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]
+            return
+            
+        except Exception as e:
+            line_bot_api.reply_message(
+                reply_token,
+                TextSendMessage(text=f"圖片處理失敗，請稍後再試。錯誤：{e}")
+            )
+            return
+
+    # 處理衣物圖片上傳
     mode = user_states[user_id]["current_mode"]
-    
     if not mode:
         line_bot_api.reply_message(
             reply_token,
@@ -224,11 +259,9 @@ def handle_image_message(event):
                 TextSendMessage(text=f"已接收。上衣: {upper_count}/{MAX_IMAGES_PER_TYPE}，褲子/裙子: {lower_count}/{MAX_IMAGES_PER_TYPE}。")
             )
         else:
-            # 所有圖片已收集完畢，改變狀態並提示使用者選擇場合
-            user_states[user_id]["is_ready"] = True
             line_bot_api.reply_message(
                 reply_token,
-                TextSendMessage(text="已收到所有圖片！請告訴我您想參加什麼樣的場合？例如：約會、工作、結婚典禮。")
+                TextSendMessage(text="已收到所有衣物圖片！接下來，請上傳一張您個人的全身照片，以便進行虛擬試穿。")
             )
             
     except Exception as e: