added ability to filter images by damaged parts

app.py

@@ -12,14 +12,16 @@ if "reset" not in st.session_state:
     st.session_state["reset"] = False
 
 with st.sidebar:
-    name = st.text_input("What's your name?", value=st.session_state.get("name", ""))
+    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)", df.columns[6:], help="Select one or more parts to filter the images by.")
+
     if st.button("Start annotation"):
         if name:
             st.session_state["name"] = name
-            st.session_state["img"], st.session_state["img_name"] = get_random_image()
+            st.session_state["img"], st.session_state["img_name"] = get_random_image(parts_filter)
             st.rerun()
         else:
             name_error_placeholder.error("Name is required")
@@ -102,7 +104,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()
+            st.session_state["img"], st.session_state["img_name"] = get_random_image(parts_filter)
             st.session_state["rotation"] = 0
             st.session_state["reset"] = False
             st.rerun()

custom_style.css

@@ -1,5 +1,5 @@
 .st-emotion-cache-1jicfl2 {
-    padding-top: 4rem;
+    padding-top: 3.8rem;
 }
 
 /* .st-emotion-cache-vdokb0 p {
@@ -10,14 +10,14 @@
 .st-dc, .st-cx {
     margin-top: 5px !important;
 }
-.st-af {
+/* .st-af {
     justify-content: center;
     margin-top: 0px;
     margin-bottom: 0px;
-}
+} */
 div.row-widget.stRadio > div[role="radiogroup"] > label[data-baseweb="radio"] {
     background-color: #f4ca9a;
-    padding-right: 38px;
+    padding-right: 36px;
     padding-left: 30px;
     padding-bottom: 6px;
     padding-top: 4px;
@@ -26,6 +26,6 @@ div.row-widget.stRadio > div[role="radiogroup"] > label[data-baseweb="radio"] {
 div.row-widget.stRadio > div[role="radiogroup"] > label[data-baseweb="radio"]:hover {
     background-color: #eb9a68;
 }
-.st-cj {
+/* .st-cj {
     background-color: rgb(255, 75, 75);
-}
+} */

functions.py

@@ -20,8 +20,12 @@ with io.BytesIO(response['Body'].read()) as bio:
     df = pd.read_csv(bio, low_memory=False)
 df = df[df['s3_available'] == True]
 
-def get_random_image():
+def get_random_image(parts_filter=False):
     not_validated_imgs = df[df["validated"] == False]["img_name"].tolist()
+    if parts_filter:
+        # get rows where all selected parts are damaged (> 0)
+        filtered_imgs = df[(df[parts_filter] > 0).all(axis=1)]["img_name"].tolist()
+        not_validated_imgs = list(set(not_validated_imgs) & set(filtered_imgs))
     if len(not_validated_imgs) == 0:
         return None, None
     image_name = random.choice(not_validated_imgs)