Spaces:
Sleeping
Sleeping
| import os | |
| import streamlit as st | |
| import pandas as pd | |
| import numpy as np | |
| import huggingface_hub as hfh | |
| voters = [ | |
| "osman", | |
| "eren", | |
| "robin", | |
| "mira", | |
| "bilal", | |
| "volunteer-1", | |
| "volunteer-2", | |
| "volunteer-3", | |
| "volunteer-4", | |
| "volunteer-5", | |
| ] | |
| # 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(): | |
| # fs = hfh.HfFileSystem(token=os.environ.get("hf_token")) | |
| # return fs.ls("datasets/aifred-smart-life-coach/capstone-images", detail=False, refresh=True) | |
| api = hfh.HfApi() | |
| files = api.list_repo_tree(repo_id="aifred-smart-life-coach/capstone-images", repo_type="dataset", recursive=True) | |
| files = [file.path for file in files] | |
| print(type(files[1].path)) | |
| return files | |
| def get_one_from_queue(voter: str): | |
| # get an image for the voter or return False if no image is left | |
| return None | |
| result = get_list_of_images() | |
| st.write(result) | |
| print(result) | |
| if submitted: | |
| 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") | |
| with st.form("images"): | |
| queue = get_one_from_queue(username) | |
| if not queue: | |
| st.write("You have voted for all the images") | |
| st.stop() | |
| st.image(queue["image"]) | |
| healthiness = st.slider("How healthy is this picture?", 0, 100, 50) | |
| 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 | |
| st.write("Outside the form") | |