Update app.py

app.py

@@ -398,9 +398,10 @@ def chatbot_response(image_urls_str, product_name_by_user, data_extractor_url, e
     if product_name_by_user != "":
         similar_product_list_json = get_product_list(product_name_by_user, data_extractor_url)
         if similar_product_list_json != {} and extract_info == False:
-            print(f"similar_product_list_json : {similar_product_list_json}")
-            similar_product_list = similar_product_list_json['product_list']
-            return similar_product_list, "Product list found"
+            with st.spinner("Fetching product information from our database... This may take a moment."):
+                print(f"similar_product_list_json : {similar_product_list_json}")
+                similar_product_list = similar_product_list_json['product_list']
+                return similar_product_list, "Product list found from our database"
             
         elif extract_info == True:
             with st.spinner("Analyzing the product... This may take a moment."):
@@ -410,7 +411,7 @@ def chatbot_response(image_urls_str, product_name_by_user, data_extractor_url, e
                 return [], final_analysis
                 
         else:
-            return [], "Product not found"
+            return [], "Product not found in our database. Please provide an image URL (Example : http://example.com/image.jpg)"
                 
     elif "http:/" in image_urls_str.lower() or "https:/" in image_urls_str.lower():
         # Extract image URL from user input
@@ -456,10 +457,12 @@ if prompt := st.chat_input("What product would you like to analyze?"):
         full_response = ""
 
         # Simulate stream of response with milliseconds delay
-        similar_products, _ = chatbot_response(prompt, "", data_extractor_url, extract_info=False)
+        similar_products, msg = chatbot_response(prompt, "", data_extractor_url, extract_info=False)
         for product in similar_products:
             full_response += product + " "
             message_placeholder.markdown(full_response + "▌")
+        if len(similar_products) == 0:
+            full_response = msg
         message_placeholder.markdown(full_response)
 
     # Add assistant response to chat history
@@ -476,14 +479,14 @@ if prompt := st.chat_input("What product would you like to analyze?"):
 
             # Generate detailed response for selected product
             with st.chat_message("assistant"):
-                detailed_response = chatbot_response("", choice, data_extractor_url, extract_info=True)
+                _, detailed_response = chatbot_response("", choice, data_extractor_url, extract_info=True)
                 st.markdown(detailed_response)
             st.session_state.messages.append({"role": "assistant", "content": detailed_response})
         elif choice == "None of the above":
             st.text_input("Please provide image URLs separated by commas:")
 
 # Option to clear chat history
-if st.button("Clear Chat History"):
-    st.session_state.messages = []
-    st.session_state.product_selected = False
-    st.rerun()
+#if st.button("Clear Chat History"):
+#    st.session_state.messages = []
+#    st.session_state.product_selected = False
+#    st.rerun()