Update app.py

app.py

@@ -273,24 +273,15 @@ if "food_items" not in st.session_state:
 # Initialize form input state variables
 if "custom_food_name" not in st.session_state:
     st.session_state["custom_food_name"] = ""
-if "form_portion_size" not in st.session_state:
-    st.session_state["form_portion_size"] = 0.1
-if "form_portion_unit" not in st.session_state:
-    st.session_state["form_portion_unit"] = UNIT_OPTIONS[0]
-if "form_cooking_method" not in st.session_state:
-    st.session_state["form_cooking_method"] = COOKING_METHODS[0]
-    
+if "form_key" not in st.session_state:
+    st.session_state["form_key"] = 0  # Add a form key to force re-rendering
+
 def reset_form_fields():
-    """Reset all form fields after adding an item"""
+    """Reset all form fields after adding an item by incrementing the form key"""
     # Reset custom food name
     st.session_state["custom_food_name"] = ""
-    # Reset portion size to default
-    st.session_state["form_portion_size"] = 0.1
-    # Reset portion unit to default
-    st.session_state["form_portion_unit"] = UNIT_OPTIONS[0]
-    # Reset cooking method to Unknown
-    st.session_state["form_cooking_method"] = COOKING_METHODS[0]
-    # We don't reset the dropdown food selection as users might want to add multiple similar items
+    # Increment the form key to force re-rendering with default values
+    st.session_state["form_key"] = st.session_state.get("form_key", 0) + 1
 
 def add_food_item(food_name, portion_size, portion_unit, cooking_method, ingredients):
     """Add a food item to the session state"""
@@ -308,7 +299,7 @@ def add_food_item(food_name, portion_size, portion_unit, cooking_method, ingredi
             "ingredients": ingredients
         })
         st.success(f"✅ Added {food_name} to your submission")
-        reset_form_fields()
+        reset_form_fields()  # Reset form by incrementing key
         return True
     else:
         st.error("❌ Please fill in all required fields")
@@ -453,15 +444,16 @@ if "original_image" in st.session_state:
     st.subheader("🍲 Add Food Details")
 
     # Use Streamlit form to capture Enter key and provide a better UX
-    with st.form(key="food_item_form"):
+    # Use a dynamic key based on form_key to force re-rendering with default values
+    form_key = st.session_state.get("form_key", 0)
+    with st.form(key=f"food_item_form_{form_key}"):
         food_selection = st.selectbox("Food Name", options=[""] + FOOD_SUGGESTIONS, index=0)
         
         # Only show custom food name if the dropdown is empty
         custom_food_name = ""
         if food_selection == "":
             custom_food_name = st.text_input("Or enter a custom food name", 
-                                          value=st.session_state["custom_food_name"], 
-                                          key="food_name_input")
+                                          value=st.session_state["custom_food_name"])
         
         # Determine the actual food name to use
         food_name = food_selection if food_selection else custom_food_name
@@ -472,27 +464,23 @@ if "original_image" in st.session_state:
                                           min_value=0.1, 
                                           step=0.1, 
                                           format="%.2f",
-                                          value=st.session_state["form_portion_size"],
-                                          key="portion_size_input")
+                                          value=0.1)  # Always use default values
         with col2:
             portion_unit = st.selectbox("Unit", 
                                        options=UNIT_OPTIONS,
-                                       index=UNIT_OPTIONS.index(st.session_state["form_portion_unit"]),
-                                       key="portion_unit_input")
+                                       index=0)  # Always use default values
         
         # Set Cooking Method with "Unknown" as the default (index 0)
         cooking_method = st.selectbox("Cooking Method (optional)", 
                                      options=COOKING_METHODS, 
-                                     index=COOKING_METHODS.index(st.session_state["form_cooking_method"]),
-                                     key="cooking_method_input")
+                                     index=0)  # Always use default values
         
         ingredients = st_tags.st_tags(
             label="Main Ingredients (Add up to 5)",
             text="Press enter to add",
             value=[],
             suggestions=["Salt", "Pepper", "Olive Oil", "Butter", "Garlic", "Onion", "Tomato"],
-            maxtags=5,
-            key="ingredients_input"
+            maxtags=5
         )
         
         # Submit button inside the form
@@ -502,11 +490,7 @@ if "original_image" in st.session_state:
                 # Store the custom food name if needed for future use
                 if custom_food_name:
                     st.session_state["custom_food_name"] = custom_food_name
-                    
-                # Reset all form fields to their defaults
-                reset_form_fields()
-                
-                # Rerun to refresh with cleared fields
+                # Don't call reset_form_fields() here, it's already called in add_food_item
                 st.rerun()
     
     # Make submit button more prominent