Spaces:
Sleeping
Sleeping
import streamlit as st | |
import streamlit.components.v1 as components | |
import random | |
from pathlib import Path | |
import huggingface_hub | |
from huggingface_hub import Repository | |
import json | |
import os | |
DATASET_REPO_URL = "https://huggingface.co/spaces/simonn8/email_rater" | |
DATA_FILENAME = "rate_results.jsonl" | |
HF_TOKEN = os.environ.get("HF_TOKEN") | |
print("is none?", HF_TOKEN is None) | |
repo = Repository( | |
local_dir=".", clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN | |
) | |
repo.git_pull() | |
user = st.sidebar.text_input("User", "your.email@provider.com") | |
paths_to_html = random.shuffle( | |
[Path(filename) for filename in Path("./html_files/").glob("*.html")] | |
) | |
label_to_number = {"Human": 0, "AI": 1} | |
if "paths_to_html" not in st.session_state: | |
st.session_state.paths_to_html = [ | |
Path(filename) for filename in Path("./html_files/").glob("*.html") | |
] | |
random.shuffle(st.session_state.paths_to_html) | |
if "html_idx" not in st.session_state: | |
st.session_state.html_idx = 0 | |
if "html_key" not in st.session_state: | |
st.session_state.html_key = "" | |
if "html_ratings" not in st.session_state: | |
st.session_state.html_ratings = {} | |
# Display the image | |
path_to_html = st.session_state.paths_to_html[st.session_state.html_idx] | |
with open(path_to_html, "r") as f: | |
html_data = f.read() | |
st.components.v1.html(html_data, height=2100) | |
if st.session_state["html_key"] in st.session_state.html_ratings: | |
rating = st.session_state.html_ratings[st.session_state["html_key"]] | |
else: | |
rating = "" | |
if rating := st.sidebar.radio( | |
"Written by a human or generated by AI ?", | |
["Human", "AI"], | |
horizontal=True, | |
index=None if rating == "" else label_to_number[rating], | |
key=path_to_html, | |
): | |
key = st.session_state.paths_to_html[st.session_state.html_idx].stem | |
st.session_state.html_key = key | |
st.session_state.html_ratings[key] = rating | |
st.sidebar.info("Submitted!") | |
if st.sidebar.button("Next email", key="next"): | |
idx = st.session_state.html_idx | |
idx = (idx + 1) % len(st.session_state.paths_to_html) | |
st.session_state.html_idx = idx | |
st.session_state.html_key = st.session_state.paths_to_html[ | |
st.session_state.html_idx | |
].stem | |
st.rerun() | |
elif st.sidebar.button("Previous email", key="previous"): | |
idx = st.session_state.html_idx | |
idx = (idx - 1) % len(st.session_state.paths_to_html) | |
st.session_state.html_idx = idx | |
st.session_state.html_key = st.session_state.paths_to_html[ | |
st.session_state.html_idx | |
].stem | |
st.rerun() | |
else: | |
if st.sidebar.button("Quit", key="quit"): | |
# send email with data or append to jsonl file | |
if user == "your.email@provider.com": | |
st.sidebar.error(f"Please provide your email adress") | |
else: | |
with open("rate_results.jsonl", "a", encoding="utf-8") as jsonl: | |
rating_data = {"user": user, "ratings": st.session_state.html_ratings} | |
jsonl.write(json.dumps(rating_data, ensure_ascii=False)) | |
jsonl.write("\n") | |
repo.push_to_hub(commit_message=f"ratings by user {user}") | |
st.sidebar.success( | |
f"Thank you for rating {len(st.session_state.html_ratings)} emails. Registered for user {user}. You can close the webpage." | |
) | |
st.session_state | |