fixed bug that clears selected parts to filter after validating annotation

app.py

@@ -10,18 +10,20 @@ car_map = declare_component("car_map", path="./car_map")
 
 if "reset" not in st.session_state:
     st.session_state["reset"] = False
+if "parts_filter" not in st.session_state:
+    st.session_state["parts_filter"] = []
 
 with st.sidebar:
     name = st.text_input("Name *", value=st.session_state.get("name", ""))
     name = name.strip().lower()
     name_error_placeholder = st.empty()
 
-    parts_filter = st.multiselect("Filter images by damaged parts (optional)", get_car_parts_count(), help="Select one or more parts to filter the images by.")
-
+    st.session_state["parts_filter"] = st.multiselect("Filter images by damaged parts (optional)", get_car_parts_count(), help="Select one or more parts to filter the images by.", default=st.session_state["parts_filter"])
+    
     if st.button("Start annotation"):
         if name:
             st.session_state["name"] = name
-            st.session_state["img"], st.session_state["img_name"] = get_random_image(parts_filter)
+            st.session_state["img"], st.session_state["img_name"] = get_random_image(st.session_state["parts_filter"])
             st.rerun()
         else:
             name_error_placeholder.error("Name is required")
@@ -104,7 +106,7 @@ if "name" in st.session_state:
             st.session_state["prev_img_name"] = img_name
             st.session_state["prev_img"] = img
             st.session_state["prev_rotation"] = st.session_state["rotation"]
-            st.session_state["img"], st.session_state["img_name"] = get_random_image(parts_filter)
+            st.session_state["img"], st.session_state["img_name"] = get_random_image(st.session_state["parts_filter"])
             st.session_state["rotation"] = 0
             st.session_state["reset"] = False
             st.rerun()