with sidebar

app.py

@@ -4,84 +4,68 @@ import random
 from pathlib import Path
 import huggingface_hub
 from huggingface_hub import Repository
+import json
 
 
-paths_to_html = [Path(filename) for filename in Path("./html_files/").glob("*.html")]
-# path_to_html = random.choice(paths_to_html)
-
-
-# Show in webpage
-
-# written_by = {}
-
-# st.header(f"Judge this email")
-# with open(path_to_html, "r") as f:
-#     html_data = f.read()
-# st.components.v1.html(html_data, height=2100)
-
-# written_by[path_to_html.as_posix()] = st.sidebar.radio(
-#     "Written by a human or generated by AI ?",
-#     ["Human", "AI"],
-#     horizontal=True,
-#     index=None, #random.randint(0, 1),
-#     key=path_to_html,
-# )
-# # col1, col2 = st.columns([1,1])
-# with col1:
-#     if st.button(label="Human", key=path_to_html.as_posix()+"human"):
-#         written_by[path_to_html.as_posix()] = "human"
-# with col2:
-#     if st.button(label="GenAI", key=path_to_html.as_posix()+"ai"):
-#         written_by[path_to_html.as_posix()] = "ai"
-
-# st.components.v1.html(
-#     """<hr style="height:10px;border:none;color:#333;background-color:#333;" /> """
-# )
-# st.stop()
+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 = paths_to_html[st.session_state.html_idx]
+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_idx"] in st.session_state.html_ratings:
-    rating = st.session_state.html_ratings[st.session_state.html_idx]
+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.radio(
+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,
 ):
-    st.session_state.html_ratings[st.session_state.html_idx] = rating
-    st.info("Submitted!")
+    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.button("Next email", key="next"):
+if st.sidebar.button("Next email", key="next"):
     idx = st.session_state.html_idx
-    idx = (idx + 1) % len(paths_to_html)
+    idx = (idx + 1) % len(st.session_state.paths_to_html)
     st.session_state.html_idx = idx
-    st.experimental_rerun()
+    st.session_state.html_key = st.session_state.paths_to_html[st.session_state.html_idx].stem
+    st.rerun()
 
-elif st.button("Previous email", key="previous"):
+elif st.sidebar.button("Previous email", key="previous"):
     idx = st.session_state.html_idx
-    idx = (idx - 1) % len(paths_to_html)
+    idx = (idx - 1) % len(st.session_state.paths_to_html)
     st.session_state.html_idx = idx
-    st.experimental_rerun()
+    st.session_state.html_key = st.session_state.paths_to_html[st.session_state.html_idx].stem
+    st.rerun()
 else:
-    if st.button("Quit", key="quit"):
+    if st.sidebar.button("Quit", key="quit"):
         # send email with data or append to jsonl file
-        st.success(f"Thank you for rating {len(st.session_state.html_ratings)} emails, you can close the webpage.")
+        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.")
 
-# st.session_state
+st.session_state

pyproject.toml

@@ -0,0 +1,18 @@
+[tool.poetry]
+name = "compare-ab"
+version = "0.1.0"
+description = "compare versions of html files"
+authors = ["Lode Nachtergaele <you@example.comlode.nachtergaele@gmail.com>"]
+license = "MIT"
+readme = "README.md"
+
+[tool.poetry.dependencies]
+python = "^3.12"
+streamlit = "^1.33.0"
+watchdog = "^4.0.0"
+huggingface-hub = "^0.22.2"
+
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"

rate_results.jsonl

@@ -0,0 +1,4 @@
+{"user": "simon.nachtergaele@gmail.com", "ratings": {"newsletter_5pm_240304": "AI", "newsletter_5pm_240304_generated": "AI", "newsletter_5pm_240306_generated": "AI", "newsletter_5pm_240307": "Human"}}
+{"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"}}