Update main.py

main.py

@@ -23,9 +23,11 @@ user_states = defaultdict(lambda: {
     "upper_body_images": [],
     "lower_body_images": [],
     "current_mode": None, # "upper" or "lower"
+    "is_ready": False # 新增狀態，標記圖片是否收集完畢
 })
 
-MAX_IMAGES = 3
+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"]
 
@@ -84,28 +86,52 @@ def get_gemini_response(prompt: str, images: list):
 def handle_text_message(event):
     user_id = event.source.user_id
     text = event.message.text.lower()
+    reply_token = event.reply_token
     
+    # 檢查是否已準備好處理場合選擇
+    if user_states[user_id]["is_ready"]:
+        prompt = (
+            f"我提供了三件上衣圖片和三件下半身圖片。請根據這些衣服，為我推薦一個適合「{text}」的穿搭建議，並詳細描述你推薦的上衣與下衣組合。請以繁體中文回答。"
+        )
+        
+        all_images = user_states[user_id]["upper_body_images"] + user_states[user_id]["lower_body_images"]
+        
+        line_bot_api.reply_message(
+            reply_token,
+            TextSendMessage(text=f"正在為您生成適合「{text}」的穿搭建議... 請稍候。")
+        )
+        
+        response_text = get_gemini_response(prompt, all_images)
+        
+        line_bot_api.push_message(
+            user_id,
+            TextSendMessage(text=response_text)
+        )
+        # 重置狀態以便下一次使用
+        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(
-            event.reply_token,
-            TextSendMessage(text=f"請上傳三件上衣圖片，您已上傳 {len(user_states[user_id]['upper_body_images'])}/{MAX_IMAGES} 張。")
+            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(
-            event.reply_token,
-            TextSendMessage(text=f"請上傳三件褲子/裙子圖片，您已上傳 {len(user_states[user_id]['lower_body_images'])}/{MAX_IMAGES} 張。")
+            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})[user_id]
+        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(
-            event.reply_token,
+            reply_token,
             TextSendMessage(text="狀態已重置。請傳送「上衣」或「褲子」來開始上傳。")
         )
     else:
         line_bot_api.reply_message(
-            event.reply_token,
+            reply_token,
             TextSendMessage(text="請先傳送「上衣」或「褲子」來選擇要上傳的衣服類型。")
         )
 
@@ -128,7 +154,7 @@ def handle_image_message(event):
         base64_img = get_base64_image(image_id)
         
         if mode == "upper":
-            if len(user_states[user_id]["upper_body_images"]) < MAX_IMAGES:
+            if len(user_states[user_id]["upper_body_images"]) < MAX_IMAGES_PER_TYPE:
                 user_states[user_id]["upper_body_images"].append(base64_img)
             else:
                 line_bot_api.reply_message(
@@ -137,7 +163,7 @@ def handle_image_message(event):
                 )
                 return
         else: # mode == "lower"
-            if len(user_states[user_id]["lower_body_images"]) < MAX_IMAGES:
+            if len(user_states[user_id]["lower_body_images"]) < MAX_IMAGES_PER_TYPE:
                 user_states[user_id]["lower_body_images"].append(base64_img)
             else:
                 line_bot_api.reply_message(
@@ -149,38 +175,19 @@ def handle_image_message(event):
         upper_count = len(user_states[user_id]["upper_body_images"])
         lower_count = len(user_states[user_id]["lower_body_images"])
         
-        if upper_count < MAX_IMAGES or lower_count < MAX_IMAGES:
+        if upper_count < MAX_IMAGES_PER_TYPE or lower_count < MAX_IMAGES_PER_TYPE:
             line_bot_api.reply_message(
                 reply_token,
-                TextSendMessage(text=f"已接收。上衣: {upper_count}/{MAX_IMAGES}，褲子/裙子: {lower_count}/{MAX_IMAGES}。")
+                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="已收到所有圖片！請告訴我您想參加什麼樣的場合？例如：約會、工作、結婚典禮。")
             )
             
-            all_images = user_states[user_id]["upper_body_images"] + user_states[user_id]["lower_body_images"]
-            
-            prompt = (
-                f"我提供了 {MAX_IMAGES} 張上衣圖片和 {MAX_IMAGES} 張下半身圖片（共 {MAX_IMAGES*2} 張）。"
-                "請根據這些衣服，為我推薦三種不同場合的穿搭建議，並盡可能將不同的上衣和下衣進行搭配。"
-                "請考慮以下場合：1. 約會，2. 結婚典禮，3. 工作。\n\n"
-                "對於每種場合，請提供一個簡短的段落說明，解釋為何這個搭配適合該場合，並詳細描述你推薦的上衣與下衣組合。請以繁體中文回答。"
-            )
-            
-            response_text = get_gemini_response(prompt, all_images)
-            
-            line_bot_api.push_message(
-                user_id,
-                TextSendMessage(text=response_text)
-            )
-
-            # 重置狀態以便下一次使用
-            user_states[user_id]["upper_body_images"] = []
-            user_states[user_id]["lower_body_images"] = []
-            user_states[user_id]["current_mode"] = None
-            
     except Exception as e:
         line_bot_api.reply_message(
             reply_token,