ratings by user test@provider.com

app.py

@@ -5,14 +5,31 @@ 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"
 
-paths_to_html = random.shuffle([Path(filename) for filename in Path("./html_files/").glob("*.html")])
+HF_TOKEN = os.environ.get("HF_TOKEN")
+print("is none?", HF_TOKEN is None)
 
-label_to_number = {"Human":0, "AI":1}
+repo = Repository(
+    local_dir=".", clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN
+)
+
+
+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")]
+    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:
@@ -51,21 +68,34 @@ 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.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.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
-        with open("rate_results.jsonl", "a", encoding="utf-8") as jsonl:
-            rating_data = {"user": "simon.nachtergaele@gmail.com", "ratings": st.session_state.html_ratings}
-            jsonl.write(json.dumps(rating_data, ensure_ascii=False) + "\n")
-        st.sidebar.success(f"Thank you for rating {len(st.session_state.html_ratings)} emails, you can close the webpage.")
+        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

rate_results.jsonl

@@ -2,3 +2,9 @@
 {"user": "simon.nachtergaele@gmail.com", "ratings": {"newsletter_5pm_240222": "Human", "newsletter_5pm_240228_generated": "AI"}}
 {"user": "simon.nachtergaele@gmail.com", "ratings": {"newsletter_5pm_240304_generated": "AI", "newsletter_5pm_240228": "AI"}}
 {"user": "simon.nachtergaele@gmail.com", "ratings": {"newsletter_5pm_240214": "AI", "newsletter_5pm_240215": "AI", "newsletter_5pm_240306_generated": "AI"}}
+{"user": "simon@twipe.com", "ratings": {}}
+{"user": "simon@twipe.com", "ratings": {"newsletter_5pm_240222": "AI", "newsletter_5pm_240226": "AI", "newsletter_5pm_240227": "AI", "newsletter_5pm_240305": "AI"}}
+{"user": "lode@provider.com", "ratings": {"newsletter_5pm_240214_generated": "AI", "newsletter_5pm_240226": "AI"}}
+{"user": "lode@provider.com", "ratings": {"newsletter_5pm_240214_generated": "AI", "newsletter_5pm_240226": "AI", "newsletter_5pm_240304": "Human"}}
+{"user": "lode@provider.com", "ratings": {"newsletter_5pm_240214_generated": "AI", "newsletter_5pm_240226": "AI", "newsletter_5pm_240304": "Human", "newsletter_5pm_240227_generated": "AI"}}
+{"user": "test@provider.com", "ratings": {"newsletter_5pm_240223": "AI", "newsletter_5pm_240228": "AI", "newsletter_5pm_240307": "AI"}}