add func

app.py

@@ -1,17 +1,8 @@
 import os
-
-if not os.path.exists("data/"):
-    os.makedirs("data/")
-    # now curl -L -o data/human-segmentation-dataset.zip https://www.kaggle.com/api/v1/datasets/download/trainingdatapro/human-segmentation-dataset
-    os.system("curl -L -o data/human-segmentation-dataset.zip https://www.kaggle.com/api/v1/datasets/download/trainingdatapro/human-segmentation-dataset")
-    os.system("unzip data/human-segmentation-dataset.zip -d data/")
-    os.system("rm data/human-segmentation-dataset.zip")
-
-
 import streamlit as st
 import pandas as pd
 import numpy as np
-
+import huggingface_hub as hfh
 
 voters = [
     "osman",
@@ -33,15 +24,16 @@ with st.form("login"):
     submitted = st.form_submit_button("Login")
 
 
-def get_one_from_queue(voter: str):
-    # get an image for the voter or return False if no image is left
-
-    df = pd.read_csv("data.csv")
+def get_list_of_images():
+    all_files = hfh.list_files_info(repo_id = "aifred-smart-life-coach/capstone-images", repo_type = "dataset")
+    return all_files
 
-    # select all the rows where the voters votes is not None
-    df = df[df[voter].notnull()]
 
+def get_one_from_queue(voter: str):
+    # get an image for the voter or return False if no image is left
+    return None
 
+st.write(get_list_of_images())
 
 if submitted:
     if not password == os.environ.get("app_password"):
@@ -66,4 +58,5 @@ if submitted:
             st.write("slideers", healthiness, fat_level, muscle_level)
             # push the data to the database
 
-    st.write("Outside the form")
+    st.write("Outside the form")
+