Update app.py

app.py

@@ -155,11 +155,22 @@ if st.session_state.step == 'input':
         else:
             st.warning("Please enter an image URL.")
 
+# Update the 'select_category' step
 elif st.session_state.step == 'select_category':
     st.image(st.session_state.query_image, caption="Query Image", use_column_width=True)
     st.subheader("Detected Clothing Items:")
+    
+    for detection in st.session_state.detections:
+        col1, col2 = st.columns([1, 3])
+        with col1:
+            st.write(f"{detection['category']} (Confidence: {detection['confidence']:.2f})")
+        with col2:
+            cropped_image = crop_image(st.session_state.query_image, detection['bbox'])
+            st.image(cropped_image, caption=detection['category'], use_column_width=True)
+    
     options = [f"{d['category']} (Confidence: {d['confidence']:.2f})" for d in st.session_state.detections]
     selected_option = st.selectbox("Select a category to search:", options)
+    
     if st.button("Search Similar Items"):
         st.session_state.selected_category = selected_option
         st.session_state.step = 'show_results'