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