refactor update_vote function to streamline image voting process and improve login handling

app.py

@@ -74,14 +74,7 @@ def update_vote(
 
     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,
@@ -94,44 +87,29 @@ def update_vote(
     # 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"))
-
-
-
 if 'loggedin' not in st.session_state:
     st.session_state['loggedin'] = 'false'
 
-# 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")
 
-# save the logged in status on the machine
-
 if submitted or st.session_state['loggedin'] == 'true':
-    st.session_state['loggedin'] = 'true'
-
     if not password == os.environ.get("app_password"):
         st.error("The password you entered is incorrect")
         st.stop()
     else:
         st.success("Welcome, " + username)
         st.write("You are now logged in")
+        st.session_state['loggedin'] = 'true'
+
 
     with st.form("images"):
         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
-        # print(image_path)
-        # image_url = f"https://huggingface.co/datasets/aifred-smart-life-coach/capstone-images/resolve/main/{image_path}"
-        # # url has whitespace, replace it with %20
-        # image_url = image_url.replace(" ", "%20")
-        # print(image_url)
-        # st.image(image_url, width=300)
-        # st.markdown(f'<img src="{image_url}">', unsafe_allow_html=True)
 
         path = hfh.hf_hub_download(repo_id="aifred-smart-life-coach/capstone-images", repo_type="dataset",filename=image_path, token=os.environ.get("hf_token"))
         st.image(path, width=300)
@@ -146,7 +124,6 @@ if submitted or st.session_state['loggedin'] == 'true':
         # Every form must have a submit button.
         submitted_second = st.form_submit_button("Submit")
 
-
     if submitted_second:
         update_vote(username, image_path, healthiness, fat_level, muscle_level)
         st.write("Vote submitted")