Spaces:
Sleeping
Sleeping
| import os | |
| import streamlit as st | |
| import pandas as pd | |
| import numpy as np | |
| import huggingface_hub as hfh | |
| import requests | |
| os.makedirs("labels", exist_ok=True) | |
| voters = [ | |
| "osman", | |
| "eren", | |
| "robin", | |
| "mira", | |
| "bilal", | |
| "volunteer-1", | |
| "volunteer-2", | |
| "volunteer-3", | |
| "volunteer-4", | |
| "volunteer-5", | |
| ] | |
| 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 contact@osbm.dev)", 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,) | |
| files = [file.path for file in files if file.path.endswith((".png", ".jpg"))] | |
| return files | |
| def get_one_from_queue(voter: str): | |
| # get an image for the voter or return False if no image is left | |
| # aifred-smart-life-coach/labels labels dataset | |
| # labels dataset multiple csv files named as [voter name].csv | |
| # each csv file has the image image path vote date, votes | |
| 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) | |
| list_of_images = get_list_of_images() | |
| print("list_of_images", len(list_of_images)) | |
| # 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)) | |
| return {"image": images_not_voted[0]} if images_not_voted else False | |
| print(get_one_from_queue("osman")) | |
| 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() | |
| # 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([ | |
| "Male", | |
| "Female", | |
| "Non-defining", | |
| ]) | |
| 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") | |