add this

app.py

@@ -1,9 +1,9 @@
 import os
 import streamlit as st
 import pandas as pd
-import numpy as np
 import huggingface_hub as hfh
 import requests
+import time
 
 os.makedirs("labels", exist_ok=True)
 
@@ -60,6 +60,7 @@ def get_one_from_queue(voter: str):
 def update_vote(
     voter: str,
     image: str,
+    gender: str,
     healthiness: int,
     fat_level: int,
     muscle_level: int,
@@ -75,13 +76,19 @@ def update_vote(
     df = pd.read_csv(file_path)
     print(df)
 
-    new_row = {
-        "image_path": image,
-        "healthiness": healthiness,
-        "fat_level": fat_level,
-        "muscle_level": muscle_level,
-    }
-    df = df.append(new_row, ignore_index=True)
+    new_row = pd.DataFrame(
+        {
+            "image_path": [image],
+            "gender": [ gender ],
+            "healthiness": [healthiness],
+            "fat_level": [fat_level],
+            "muscle_level": [muscle_level],
+            "vote_date": [int(time.time())],
+            "voter_name": [voter],
+        }
+    )
+
+    df = df.concat([df, new_row])
     df.to_csv(file_path, index=False)
 
     # push the file to the dataset
@@ -111,7 +118,12 @@ if submitted or st.session_state['loggedin'] == 'true':
             st.write("You have voted for all the images")
             st.stop()
 
-        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"))
+        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)
         gender = st.selectbox("Gender", [
             "Male",
@@ -125,7 +137,14 @@ if submitted or st.session_state['loggedin'] == 'true':
         submitted_second = st.form_submit_button("Submit")
 
     if submitted_second:
-        update_vote(username, image_path, healthiness, fat_level, muscle_level)
+        update_vote(
+            voter=username,
+            image=image_path,
+            gender=gender,
+            healthiness=healthiness,
+            fat_level=fat_level,
+            muscle_level=muscle_level
+        )
         st.write("Vote submitted")
         # push the data to the database