chore: Modify log_blurb_and_vote function to handle already voted responses

app.py

@@ -20,7 +20,6 @@ MODELS = [
     "NousResearch/Nous-Hermes-2-Mixtral-8x7B-DPO",
 ]
 
-
 # Set up dataset storage
 dataset_folder = Path("dataset")
 dataset_folder.mkdir(exist_ok=True)
@@ -124,8 +123,17 @@ def generate_blurb(prompt):
         yield full_text
 
 
-# Function to log blurb and vote
-def log_blurb_and_vote(prompt, blurb, vote, user_info: gr.OAuthProfile | None, *args):
+# Modified function to log blurb and vote
+def log_blurb_and_vote(
+    prompt, blurb, vote, user_info: gr.OAuthProfile | None, has_voted, *args
+):
+    if has_voted:
+        return (
+            "You've already voted on this response.",
+            has_voted,
+            gr.Row.update(visible=True),
+        )
+
     user_id = user_info.username if user_info is not None else str(uuid.uuid4())
     log_entry = {
         "timestamp": datetime.now().isoformat(),
@@ -139,7 +147,7 @@ def log_blurb_and_vote(prompt, blurb, vote, user_info: gr.OAuthProfile | None, *
         with dataset_file.open("a") as f:
             f.write(json.dumps(log_entry) + "\n")
     gr.Info("Thank you for voting!")
-    return f"Logged: {vote} by user {user_id}"
+    return f"Logged: {vote} by user {user_id}", True, gr.Row.update(visible=False)
 
 
 # Create custom theme
@@ -161,6 +169,7 @@ with gr.Blocks(theme=tufte_theme) as demo:
 
     prompt_state = gr.State()
     blurb_output = gr.Markdown(label="Book blurb")
+    has_voted = gr.State(False)
 
     with gr.Row(visible=False) as voting_row:
         upvote_btn = gr.Button("👍 would read")
@@ -169,15 +178,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)
+        return "Generating...", gr.Row.update(visible=False), False
 
     def show_voting_buttons(blurb):
-        return blurb, gr.Row.update(visible=True)
+        return blurb, gr.Row.update(visible=True), False
 
     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,
+        outputs=[blurb_output, voting_row, has_voted],
     ).then(generate_blurb, inputs=prompt_state, outputs=blurb_output).then(
-        show_voting_buttons, inputs=blurb_output, outputs=[blurb_output, voting_row]
+        show_voting_buttons,
+        inputs=blurb_output,
+        outputs=[blurb_output, voting_row, has_voted],
     )
 
     upvote_btn.click(
@@ -187,8 +200,9 @@ with gr.Blocks(theme=tufte_theme) as demo:
             blurb_output,
             gr.Textbox(value="upvote", visible=False),
             login_btn,
+            has_voted,
         ],
-        outputs=vote_output,
+        outputs=[vote_output, has_voted, voting_row],
     )
     downvote_btn.click(
         log_blurb_and_vote,
@@ -197,8 +211,9 @@ with gr.Blocks(theme=tufte_theme) as demo:
             blurb_output,
             gr.Textbox(value="downvote", visible=False),
             login_btn,
+            has_voted,
         ],
-        outputs=vote_output,
+        outputs=[vote_output, has_voted, voting_row],
     )
 
 if __name__ == "__main__":