Update app.py

app.py

@@ -145,6 +145,7 @@ def update_cow():
         st.write(f"**Health History**: {cow_info[4]}")
         st.write(f"**Health Score**: {cow_info[5]}")
 
+        # Pre-fill the form with the existing data
         new_breed = st.selectbox("Select New Breed", ['Girolando', 'Jersey', 'Holstein Friesian (HF)', 'Sahiwal', 'Kankrej', 'Gir / Gyr', 'Red Sindhi'], index=['Girolando', 'Jersey', 'Holstein Friesian (HF)', 'Sahiwal', 'Kankrej', 'Gir / Gyr', 'Red Sindhi'].index(cow_info[1]))
         new_age = st.number_input("Enter New Age (in years)", min_value=0, value=cow_info[2])
         new_tag_id = st.text_input("Enter New Tag ID", value=cow_info[3])
@@ -163,10 +164,14 @@ def update_cow():
 
         new_previous_illness = st.selectbox("Previous Illness", ['No', 'Yes'], index=['No', 'Yes'].index(cow_info[4].split('\n')[2].split(": ")[1]))
         new_previous_diseases = []
+
+        # Fixing the Previous Disease selection
         if new_previous_illness == 'Yes':
             existing_previous_diseases = cow_info[4].split('\n')[2].split(": ")[1]
+            # Ensure the values passed as default are part of the available diseases list
             existing_previous_diseases = existing_previous_diseases.split(", ") if existing_previous_diseases != 'None' else []
-            new_previous_diseases = st.multiselect("Select Previous Disease(s)", list(disease_points.keys()), default=existing_previous_diseases)
+            # Only provide valid options that exist in the disease list
+            new_previous_diseases = st.multiselect("Select Previous Disease(s)", list(disease_points.keys()), default=[disease for disease in existing_previous_diseases if disease in disease_points.keys()])
 
         # Reproductive Status handling (as string)
         reproductive_status_list = ['Not Applicable', 'Estrus Period', 'Gestation Period', 'Adult/Milking Stage', 'Pregnant/Calving Stage']