Update app.py

app.py

@@ -121,7 +121,7 @@ st.title("Accessary Search App")
 # 단계별 처리
 if st.session_state.step == 'input':
     st.session_state.query_image_url = st.text_input("Enter image URL:", st.session_state.query_image_url)
-    if st.button("Detect acsseary"):
+    if st.button("Detect accesseary"):
         if st.session_state.query_image_url:
             query_image = load_image_from_url(st.session_state.query_image_url)
             if query_image is not None:
@@ -172,7 +172,9 @@ elif st.session_state.step == 'show_results':
         with col2:
             st.write(f"Name: {img['info']['name']}")
             st.write(f"Brand: {img['info']['brand']}")
-            st.write(f"Category: {img['info']['category']}")
+            category = img['info'].get('category')
+            if category:  
+                st.write(f"Category: {category}")
             st.write(f"Price: {img['info']['price']}")
             st.write(f"Discount: {img['info']['discount']}%")
             st.write(f"Similarity: {img['similarity']:.2f}")
@@ -198,7 +200,9 @@ else:  # Text search
                 with col2:
                     st.write(f"Name: {img['info']['name']}")
                     st.write(f"Brand: {img['info']['brand']}")
-                    st.write(f"Category: {img['info']['category']}")
+                    category = img['info'].get('category')
+                    if category:  
+                        st.write(f"Category: {category}")
                     st.write(f"Price: {img['info']['price']}")
                     st.write(f"Discount: {img['info']['discount']}%")
                     st.write(f"Similarity: {img['similarity']:.2f}")