Update app.py

app.py

@@ -139,29 +139,14 @@ if search_type == "Image URL":
                 detections = detect_clothing(query_image)
                 
                 if detections:
+                    st.session_state.detections = detections  # 세션 상태에 detections 저장
                     st.subheader("Detected Clothing Items:")
                     selected_category = st.selectbox("Select a category to search:", 
                                                      [f"{d['category']} (Confidence: {d['confidence']:.2f})" for d in detections])
                     
                     if st.button("Search Similar Items"):
-                        selected_detection = next(d for d in detections if f"{d['category']} (Confidence: {d['confidence']:.2f})" == selected_category)
-                        cropped_image = crop_image(query_image, selected_detection['bbox'])
-                        st.image(cropped_image, caption="Cropped Image", use_column_width=True)
-                        query_embedding = get_image_embedding(cropped_image)
-                        similar_images = find_similar_images(query_embedding)
-                        
-                        st.subheader("Similar Items:")
-                        for img in similar_images:
-                            col1, col2 = st.columns(2)
-                            with col1:
-                                st.image(img['info']['image_url'], use_column_width=True)
-                            with col2:
-                                st.write(f"Name: {img['info']['name']}")
-                                st.write(f"Brand: {img['info']['brand']}")
-                                st.write(f"Category: {img['info']['category']}")
-                                st.write(f"Price: {img['info']['price']}")
-                                st.write(f"Discount: {img['info']['discount']}%")
-                                st.write(f"Similarity: {img['similarity']:.2f}")
+                        st.session_state.selected_category = selected_category  # 선택된 카테고리 저장
+                        st.session_state.search_clicked = True  # 검색 버튼 클릭 상태 저장
                 else:
                     st.warning("No clothing items detected in the image.")
             else:
@@ -169,6 +154,31 @@ if search_type == "Image URL":
         else:
             st.warning("Please enter an image URL.")
 
+    # 검색 결과 표시 로직
+    if 'search_clicked' in st.session_state and st.session_state.search_clicked:
+        selected_detection = next(d for d in st.session_state.detections 
+                                  if f"{d['category']} (Confidence: {d['confidence']:.2f})" == st.session_state.selected_category)
+        cropped_image = crop_image(query_image, selected_detection['bbox'])
+        st.image(cropped_image, caption="Cropped Image", use_column_width=True)
+        query_embedding = get_image_embedding(cropped_image)
+        similar_images = find_similar_images(query_embedding)
+        
+        st.subheader("Similar Items:")
+        for img in similar_images:
+            col1, col2 = st.columns(2)
+            with col1:
+                st.image(img['info']['image_url'], use_column_width=True)
+            with col2:
+                st.write(f"Name: {img['info']['name']}")
+                st.write(f"Brand: {img['info']['brand']}")
+                st.write(f"Category: {img['info']['category']}")
+                st.write(f"Price: {img['info']['price']}")
+                st.write(f"Discount: {img['info']['discount']}%")
+                st.write(f"Similarity: {img['similarity']:.2f}")
+
+        # 검색 완료 후 상태 초기화
+        st.session_state.search_clicked = False
+
 else:  # Text search
     query_text = st.text_input("Enter search text:")
     if st.button("Search by Text"):