Update app.py

app.py

@@ -281,64 +281,47 @@ Claims Analysis:
 def main():
     st.title("Food Product Analysis Chatbot")
 
-    # Initialize session state
-    if 'state' not in st.session_state:
-        st.session_state.state = 'ingredients'
-        st.session_state.ingredients = []
-        st.session_state.claims = []
-        st.session_state.product_info = {}
-
-    # Input fields based on current state
-    if st.session_state.state == 'ingredients':
-        ingredients = st.text_input("Enter the list of ingredients (separated by commas):")
-        if st.button("Submit Ingredients"):
-            st.session_state.ingredients = [ingredient.strip() for ingredient in ingredients.split(',')]
-            st.session_state.state = 'claims'
-            st.rerun()
-
-    elif st.session_state.state == 'claims':
-        claims = st.text_input("Enter the list of product claims (separated by commas):")
-        if st.button("Submit Claims"):
-            st.session_state.claims = [claim.strip() for claim in claims.split(',')]
-            st.session_state.state = 'product_info'
-            st.rerun()
-
-    elif st.session_state.state == 'product_info':
-        brand_name = st.text_input("Enter the brand name:")
-        product_name = st.text_input("Enter the product name:")
-        if st.button("Submit Product Info"):
-            st.session_state.product_info = {
-                "brandName": brand_name,
-                "productName": product_name
-            }
-            st.session_state.state = 'analysis'
-            st.rerun()
-
-    elif st.session_state.state == 'analysis':
-        st.write("Analyzing the product... This may take a moment.")
-        
-        # Perform analysis
-        processing_level = analyze_processing_level(st.session_state.ingredients, assistant1)
-        harmful_ingredient_analysis = analyze_harmful_ingredients(st.session_state.ingredients, assistant2)
-        claims_analysis = analyze_claims(st.session_state.claims, assistant3)
-        
-        # Generate final analysis
-        final_analysis = generate_final_analysis(
-            st.session_state.product_info,
-            processing_level,
-            harmful_ingredient_analysis,
-            claims_analysis
-        )
-        
-        st.write("Analysis complete!")
-        st.write("Final 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")
+        ingredients = st.text_area("Ingredients (separated by commas)")
+        claims = st.text_area("Product Claims (separated by commas)")
+
+        submitted = st.form_submit_button("Analyze Product")
+
+    if submitted:
+        # Process inputs
+        ingredients_list = [ingredient.strip() for ingredient in ingredients.split(',')]
+        claims_list = [claim.strip() for claim in claims.split(',')]
+        product_info = {
+            "brandName": brand_name,
+            "productName": product_name
+        }
+
+        # Display a message while analyzing
+        with st.spinner("Analyzing the product... This may take a moment."):
+            # Perform analysis
+            processing_level = analyze_processing_level(ingredients_list, assistant1.id)
+            harmful_ingredient_analysis = analyze_harmful_ingredients(ingredients_list, assistant2.id)
+            claims_analysis = analyze_claims(claims_list, assistant3.id)
+
+            # Generate final analysis
+            final_analysis = generate_final_analysis(
+                product_info,
+                processing_level,
+                harmful_ingredient_analysis,
+                claims_analysis
+            )
+
+        # 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.session_state.state = 'ingredients'
-            st.session_state.ingredients = []
-            st.session_state.claims = []
-            st.session_state.product_info = {}
             st.rerun()
 
 if __name__ == "__main__":