check for previous vote

app.py

@@ -1,16 +1,17 @@
-import gradio as gr
+import hashlib
 import json
-from datetime import datetime
-from theme import TufteInspired
-import uuid
-from huggingface_hub import InferenceClient, CommitScheduler
-from openai import OpenAI
-from huggingface_hub import get_token, login
-from prompts import detailed_genre_description_prompt, basic_prompt
-import random
 import os
+import random
+import uuid
+from datetime import datetime
 from pathlib import Path
-import os
+
+import gradio as gr
+from huggingface_hub import CommitScheduler, InferenceClient, get_token, login
+from openai import OpenAI
+
+from prompts import basic_prompt, detailed_genre_description_prompt
+from theme import TufteInspired
 
 # Ensure you're logged in to Hugging Face
 login(os.getenv("HF_TOKEN"))
@@ -40,7 +41,7 @@ else:
     print(f"Creating new dataset file: {dataset_file}")
 
 # Set up CommitScheduler for dataset uploads
-repo_id = "davanstrien/summer-reading-preferences"  #
+repo_id = "davanstrien/summer-reading-preferences"
 scheduler = CommitScheduler(
     repo_id=repo_id,
     repo_type="dataset",
@@ -49,6 +50,9 @@ scheduler = CommitScheduler(
     every=1,  # Upload every 5 minutes
 )
 
+# Add a dictionary to store votes
+votes = {}
+
 
 def generate_prompt():
     if random.choice([True, False]):
@@ -79,9 +83,21 @@ def generate_blurb(prompt):
         yield full_text
 
 
-# Function to log blurb and vote
+def generate_vote_id(user_id, blurb):
+    # Create a unique identifier for this vote opportunity
+    return hashlib.md5(f"{user_id}:{blurb}".encode()).hexdigest()
+
+
+# Modified log_blurb_and_vote function
 def log_blurb_and_vote(prompt, blurb, vote, user_info: gr.OAuthProfile | None, *args):
     user_id = user_info.username if user_info is not None else str(uuid.uuid4())
+    vote_id = generate_vote_id(user_id, blurb)
+
+    if vote_id in votes:
+        return "You've already voted on this blurb!"
+
+    votes[vote_id] = vote
+
     log_entry = {
         "timestamp": datetime.now().isoformat(),
         "prompt": prompt,
@@ -115,6 +131,7 @@ with gr.Blocks(theme=tufte_theme) as demo:
 
     prompt_state = gr.State()
     blurb_output = gr.Markdown(label="Book blurb")
+    user_state = gr.State()
 
     with gr.Row(visible=False) as voting_row:
         upvote_btn = gr.Button("👍 would read")
@@ -122,14 +139,19 @@ with gr.Blocks(theme=tufte_theme) as demo:
 
     vote_output = gr.Textbox(label="Vote Status", interactive=False, visible=False)
 
-    def generate_and_show(prompt):
-        return "Generating...", gr.Row.update(visible=False)
+    def generate_and_show(prompt, user_info):
+        # Reset votes for new blurb
+        global votes
+        votes = {}
+        return "Generating...", gr.Row.update(visible=False), user_info
 
     def show_voting_buttons(blurb):
         return blurb, gr.Row.update(visible=True)
 
     generate_btn.click(get_and_store_prompt, outputs=prompt_state).then(
-        generate_and_show, inputs=prompt_state, outputs=[blurb_output, voting_row]
+        generate_and_show,
+        inputs=[prompt_state, login_btn],
+        outputs=[blurb_output, voting_row, user_state],
     ).then(generate_blurb, inputs=prompt_state, outputs=blurb_output).then(
         show_voting_buttons, inputs=blurb_output, outputs=[blurb_output, voting_row]
     )
@@ -140,7 +162,7 @@ with gr.Blocks(theme=tufte_theme) as demo:
             prompt_state,
             blurb_output,
             gr.Textbox(value="upvote", visible=False),
-            login_btn,
+            user_state,
         ],
         outputs=vote_output,
     )
@@ -150,7 +172,7 @@ with gr.Blocks(theme=tufte_theme) as demo:
             prompt_state,
             blurb_output,
             gr.Textbox(value="downvote", visible=False),
-            login_btn,
+            user_state,
         ],
         outputs=vote_output,
     )