Update app.py

app.py

@@ -363,103 +363,6 @@ Claims Analysis:
     return completion.choices[0].message.content
 
 
-# Streamlit app
-def old_main():
-    global data_extractor_url
-    
-    st.title("Food Product Analysis")
-
-    # Create a form for all inputs
-    with st.form("product_analysis_form"):
-        st.write("Please enter the following information about the product:")
-        #brand_name = st.text_input("Brand Name")
-        #product_name = st.text_input("Product Name")
-        image_links = st.text_area("Image Links (separated by commas)")
-
-
-        submitted = st.form_submit_button("Analyze Product")
-
-    if submitted:
-        # Process inputs
-
-        image_links_list = []
-        if image_links:
-            image_links_list = [image_link.strip() for image_link in image_links.split(',')]
-
-        # Display a message while analyzing
-        with st.spinner("Analyzing the product... This may take a moment."):
-            # Perform analysis
-            processing_level = ""
-            harmful_ingredient_analysis = ""
-            claims_analysis = ""
-            ingredients_list = []
-            claims_list = []
-            product_info = {}
-            brand_name = ""
-            brand_product = ""
-            
-            if len(image_links_list) > 0:
-                product_info_raw = extract_data_from_product_image(image_links_list, data_extractor_url)
-                if product_info_raw != "{}":
-                    product_info_from_db = json.loads(product_info_raw)
-                print(f"DEBUG : product_info_from_db : {product_info_from_db}\n\n")
-                
-                if product_info_from_db:
-                    try:
-                        brand_name = product_info_from_db["brandName"]
-                    except:
-                        print(f"Brand Name not found in product_info {product_info_from_db}")
-                        
-                    try:
-                        product_name = product_info_from_db["productName"]
-                    except:
-                        print(f"Product Name not found in product_info {product_info_from_db}")
-                        
-                    try:
-                        ingredients_list = [ingredient["name"] for ingredient in product_info_from_db["ingredients"]]
-                    except:
-                        print(f"Ingredient list not found in product_info {product_info_from_db}")
-                        
-                    try:
-                        claims_list = product_info_from_db["claims"]
-                    except:
-                        print(f"Claims list not found in product_info {product_info_from_db}")
-                        
-            if len(ingredients_list) > 0:
-                processing_level = analyze_processing_level(ingredients_list, brand_name, product_name, assistant1.id)
-                harmful_ingredient_analysis = analyze_harmful_ingredients(ingredients_list, brand_name, product_name, assistant2.id)
-            else:
-                print("No ingredients specified by the user!")
-                
-            if len(claims_list) > 0:
-                claims_analysis = analyze_claims(claims_list, assistant3.id)
-            else:
-                print("No claims specified by the user!")
-            
-            # Generate final analysis
-            if processing_level != "" or harmful_ingredient_analysis != "" or claims_analysis != "":
-                final_analysis = generate_final_analysis(
-                    brand_name,
-                    product_name,
-                    processing_level,
-                    harmful_ingredient_analysis,
-                    claims_analysis
-                )
-            else:
-                final_analysis = "Sorry, No product information found! Please re-try"
-
-        # Display results
-        st.success("Analysis complete!")
-        st.subheader("Final Analysis:")
-        st.write(final_analysis)
-
-        # Option to start over
-        if st.button("Analyze Another Product"):
-            st.rerun()
-
-#if __name__ == "__main__":
-#    main()
-
 def analyze_product(product_info_raw):
     global assistant1, assistant2, assistant3
     
@@ -480,7 +383,8 @@ def analyze_product(product_info_raw):
         return final_analysis
     else:
         return "I'm sorry, product information could not be extracted from the url."    
-                
+
+# Streamlit app
 # Initialize session state
 if 'messages' not in st.session_state:
     st.session_state.messages = []
@@ -494,6 +398,7 @@ 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 = json.loads(similar_product_list_json)['product_list']
             return similar_product_list
             
@@ -537,21 +442,8 @@ st.write("Hello! I'm your food product analysis assistant. Select one of the fol
 #If user selects 1, ask user to enter image_urls separated by commas else do nothing. Repeat the options after response is generated. 
 #Do not allow user to send prompts while chatbot is still generating a response
 
-options = [
-    "I want to know more about a product",
-    "How much % of my daily allowance of calorie, fat, protein, carb, sugar or any other nutrient will I exhaust if I eat this?",
-    "I want to understand what an ingredient name means",
-    "I want to understand what different nutrients mean",
-]
-choice = st.radio("Choose an option:", options)
-
-
 # User input
-if choice == options[0]:
-    product_name_by_user =  st.text_input("Enter product name")
-    #image_urls_str = st.text_input("Enter image URLs separated by commas")
-else:
-    product_name_by_user = ""
+product_name_by_user =  st.text_input("Enter product name")
 
 # Display chat messages
 for message in st.session_state.messages:
@@ -590,7 +482,7 @@ if product_name_by_user != "":
         st.markdown(response)
 
     st.session_state["response_pending"] = False
-
+    
 # Option to clear chat history
 if st.button("Clear Chat History"):
     st.session_state.messages = []