Update app.py

app.py

@@ -397,20 +397,20 @@ def chatbot_response(image_urls_str, product_name_by_user, data_extractor_url, e
     image_urls = []
     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:
+        if similar_product_list_json != {} and extract_info == False:
             print(f"similar_product_list_json : {similar_product_list_json}")
-            similar_product_list = json.loads(similar_product_list_json)['product_list']
-            return similar_product_list
+            similar_product_list = similar_product_list_json['product_list']
+            return similar_product_list, "Product list found"
             
         elif extract_info == True:
             with st.spinner("Analyzing the product... This may take a moment."):
                 product_info_raw = get_product_data_from_db(product_name_by_user, data_extractor_url)
                 print(f"DEBUG product_info_raw : {product_info_raw}")
                 final_analysis = analyze_product(product_info_raw)
-                return final_analysis
+                return [], final_analysis
                 
         else:
-            return ""
+            return [], "Product not found"
                 
     elif "http:/" in image_urls_str.lower() or "https:/" in image_urls_str.lower():
         # Extract image URL from user input
@@ -425,10 +425,10 @@ def chatbot_response(image_urls_str, product_name_by_user, data_extractor_url, e
             product_info_raw = extract_data_from_product_image(image_urls, data_extractor_url)
             print(f"DEBUG product_info_raw : {product_info_raw}")
             final_analysis = analyze_product(product_info_raw)
-            return final_analysis
+            return [], final_analysis
             
     else:
-        return "I'm here to analyze food products. Please provide an image URL (Example : http://example.com/image.jpg) or product name (Example : Harvest Gold Bread)"
+        return [], "I'm here to analyze food products. Please provide an image URL (Example : http://example.com/image.jpg) or product name (Example : Harvest Gold Bread)"
 
 # Streamlit app
 st.title("ConsumeWise")
@@ -460,13 +460,14 @@ if product_name_by_user != "":
     st.session_state["response_pending"] = True
     
     # Generate and display assistant response. Do not allow user to send prompts while chatbot is still generating a response
-    similar_product_list = chatbot_response("", product_name_by_user, data_extractor_url, extract_info = False)
-    
+    similar_product_list, msg = chatbot_response("", product_name_by_user, data_extractor_url, extract_info = False)
+
+    choice = ""
     if len(similar_product_list) > 0:
         choice = st.radio("Select a product:", similar_product_list.append("None of the above"))
         if choice != "None of the above":
-            response = chatbot_response("", product_name_by_user, data_extractor_url, extract_info = True)
-            
+            _, response = chatbot_response("", product_name_by_user, data_extractor_url, extract_info = True)
+    
     if len(similar_product_list) == 0 or choice == "None of the above":
         image_urls_str = st.text_input("Enter image URLs separated by commas")
         
@@ -475,8 +476,8 @@ if product_name_by_user != "":
         with st.chat_message("user"):
             st.markdown(image_urls_str)
             
-        response = chatbot_response(image_urls_str, "", data_extractor_url, extract_info = True)
-        
+        _, response = chatbot_response(image_urls_str, "", data_extractor_url, extract_info = True)
+
     st.session_state.messages.append({"role": "assistant", "content": response})
     with st.chat_message("assistant"):
         st.markdown(response)