now

app.py

@@ -22,12 +22,6 @@ voters = [
 
 api = hfh.HfApi(token=os.environ.get("hf_token"))
 
-# login page
-with st.form("login"):
-    username = st.selectbox("Select voter", voters)
-    password = st.text_input("Password (get password from [email protected])", type="password")
-    submitted = st.form_submit_button("Login")
-
 
 def get_list_of_images():
     files = api.list_repo_tree(repo_id="aifred-smart-life-coach/capstone-images", repo_type="dataset", recursive=True,)
@@ -63,8 +57,56 @@ def get_one_from_queue(voter: str):
     return {"image": images_not_voted[0]} if images_not_voted else False
 
 
+def update_vote(
+    voter: str,
+    image: str,
+    healthiness: int,
+    fat_level: int,
+    muscle_level: int,
+):
+    url = f"https://huggingface.co/datasets/aifred-smart-life-coach/labels/raw/main/{voter}.csv"
+
+    # fetch file and save it to the labels folder
+    file_path = f"labels/{voter}.csv"
+    req = requests.get(url)
+    with open(file_path, "wb") as file:
+        file.write(req.content)
+
+    df = pd.read_csv(file_path)
+    print(df)
+    num_past_votes = df.shape[0]
+    print("num_past_votes", num_past_votes)
+
+    # get the list of images that are not present in the csv file
+    images_not_voted = list(set(list_of_images) - set(df["image_path"].tolist()))
+    print("images_not_voted", len(images_not_voted))
+
+    # update the csv file with the new vote
+    new_row = {
+        "image_path": image,
+        "healthiness": healthiness,
+        "fat_level": fat_level,
+        "muscle_level": muscle_level,
+    }
+    df = df.append(new_row, ignore_index=True)
+    df.to_csv(file_path, index=False)
+
+    # push the file to the dataset
+    api.push_to_hub(file_path, repo_id="aifred-smart-life-coach/labels", repo_type="dataset", commit_message=f"Voted for {image}")
+
 print(get_one_from_queue("osman"))
 
+
+
+
+
+# login page
+with st.form("login"):
+    username = st.selectbox("Select voter", voters)
+    password = st.text_input("Password (get password from [email protected])", type="password")
+    submitted = st.form_submit_button("Login")
+
+
 if submitted:
     if not password == os.environ.get("app_password"):
         st.error("The password you entered is incorrect")
@@ -74,13 +116,16 @@ if submitted:
         st.write("You are now logged in")
 
     with st.form("images"):
-        queue = get_one_from_queue(username)
-        if not queue:
+        image_path = get_one_from_queue(username)
+        if not image_path:
             st.write("You have voted for all the images")
             st.stop()
         # https://huggingface.co/datasets/aifred-smart-life-coach/capstone-images/resolve/main/kaggle-human-segmentation-dataset/Women%20I/img/woman_image_200.jpg
-        st.image(f"https://huggingface.co/datasets/aifred-smart-life-coach/capstone-images/resolve/main/{queue['image']}", width=300)
-        gender = st.select([
+        print(image_path)
+        image_url = f"https://huggingface.co/datasets/aifred-smart-life-coach/capstone-images/resolve/main/{image_path}"
+        print(image_url)
+        st.image(image_url, width=300)
+        gender = st.selectbox("Gender", [
             "Male",
             "Female",
             "Non-defining",
@@ -89,10 +134,13 @@ if submitted:
         fat_level = st.slider("How fat is this picture?", 0, 100, 50)
         muscle_level = st.slider("How muscular is this picture?", 0, 100, 50)
         # Every form must have a submit button.
-        submitted = st.form_submit_button("Submit")
-        if submitted:
-            st.write("slideers", healthiness, fat_level, muscle_level)
-            # push the data to the database
+        submitted_second = st.form_submit_button("Submit")
+
+
+    if submitted_second:
+        update_vote(username, image_path, healthiness, fat_level, muscle_level)
+        st.write("Vote submitted")
+        # push the data to the database
 
     st.write("Outside the form")