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 contact@osbm.dev)", 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 = list(files)
    print(type(files[1]))
    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

st.write(get_list_of_images())

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")