fix: chatgpt의 id 인식 오류로 인한 500에러 리턴 수정

doongyeop · Aug 21, 2024 · 4718bd8 · 4718bd8
1 parent 9e85b3a
commit 4718bd8
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/backend/ai/app/services/recommend.py b/backend/ai/app/services/recommend.py
@@ -63,14 +63,18 @@ def validate_outfit(self, items: list[int], id_2_clothes: dict[int, Clothes]):
         outfit_category = []
         print(str(items) + "검증 시작")
         for item in items:
-            clothes = id_2_clothes[item]
-            high_category = self.clothes_metadata.lowcategoryName_to_highcategoryId(clothes.subCategory)
-            if high_category is not None and high_category not in outfit_category:
-                outfit_category.append(high_category)
-                outfit_set.append((clothes.id, high_category, clothes.imgPath))
+            try:
+                clothes = id_2_clothes[item]
+                high_category = self.clothes_metadata.lowcategoryName_to_highcategoryId(clothes.subCategory)
+                if high_category is not None and high_category not in outfit_category:
+                    outfit_category.append(high_category)
+                    outfit_set.append((clothes.id, high_category, clothes.imgPath))
+            except Exception as e:
+                logging.error("id : "+str(item)+" 아이템을 찾을 수 없습니다. ChatGPT 의 id 인식 오류")
+                break
         if len(outfit_set) == len(items):
             print(str(items) + str(outfit_set) + str(outfit_category) + "검증 성공")
             return outfit_set
         else:
-            print(str(items) + str(outfit_set) + str(outfit_category) + "검증 성공")
+            print(str(items) + str(outfit_set) + str(outfit_category) + "검증 실패")
             return None