added count of car parts in filter

app.py

@@ -16,7 +16,7 @@ with st.sidebar:
     name = name.strip().lower()
     name_error_placeholder = st.empty()
 
-    parts_filter = st.multiselect("Filter images by damaged parts (optional)", df.columns[6:], help="Select one or more parts to filter the images by.")
+    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.")
 
     if st.button("Start annotation"):
         if name:

functions.py

@@ -15,11 +15,16 @@ folder = "car-damage-detection/scrappedImages/"
 csv_folder = "car-damage-detection/CSVs/"
 s3_df_path = csv_folder + "70k_old_annotations_fixed.csv"
 response = s3_client.get_object(Bucket=bucket_name, Key=s3_df_path)
-# df = pd.read_csv("CSVs/70k_old_annotations_fixed.csv", low_memory=False)
 with io.BytesIO(response['Body'].read()) as bio:
     df = pd.read_csv(bio, low_memory=False)
 df = df[df['s3_available'] == True]
 
+def get_car_parts_count():
+    car_parts = df.columns[6:]
+    # create a dictionary with the count of each part, +1 for a part if value > 0
+    car_parts_count = {part: len(df[df[part] > 0]) for part in car_parts}
+    return [f"{part} ({count})" for part, count in car_parts_count.items()]
+
 def get_random_image(parts_filter=False):
     not_validated_imgs = df[df["validated"] == False]["img_name"].tolist()
     if parts_filter: