Update main.py

main.py

@@ -114,6 +114,15 @@ def handle_text_message(event):
     user_id = event.source.user_id
     text = event.message.text.lower()
     reply_token = event.reply_token
+
+    # 新增 重置 功能
+    if text == "重置" or text == "重來" or text == "重新開始" or 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
     
     # 處理個人資訊輸入
     if not user_states[user_id]["is_ready_for_photo"] and not user_states[user_id]["is_ready_for_outfit"]:
@@ -163,14 +172,7 @@ def handle_text_message(event):
                 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,