revert

app.py

@@ -20,6 +20,7 @@ MODELS = [
     "NousResearch/Nous-Hermes-2-Mixtral-8x7B-DPO",
 ]
 
+
 # Set up dataset storage
 dataset_folder = Path("dataset")
 dataset_folder.mkdir(exist_ok=True)
@@ -123,17 +124,8 @@ def generate_blurb(prompt):
         yield full_text
 
 
-# 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),
-        )
-
+# Function to log blurb and vote
+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())
     log_entry = {
         "timestamp": datetime.now().isoformat(),
@@ -147,7 +139,7 @@ def log_blurb_and_vote(
         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}", True, gr.Row.update(visible=False)
+    return f"Logged: {vote} by user {user_id}"
 
 
 # Create custom theme
@@ -169,7 +161,6 @@ 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")
@@ -178,19 +169,15 @@ 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), False
+        return "Generating...", gr.Row.update(visible=False)
 
     def show_voting_buttons(blurb):
-        return blurb, gr.Row.update(visible=True), False
+        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, has_voted],
+        generate_and_show, inputs=prompt_state, outputs=[blurb_output, voting_row]
     ).then(generate_blurb, inputs=prompt_state, outputs=blurb_output).then(
-        show_voting_buttons,
-        inputs=blurb_output,
-        outputs=[blurb_output, voting_row, has_voted],
+        show_voting_buttons, inputs=blurb_output, outputs=[blurb_output, voting_row]
     )
 
     upvote_btn.click(
@@ -200,9 +187,8 @@ with gr.Blocks(theme=tufte_theme) as demo:
             blurb_output,
             gr.Textbox(value="upvote", visible=False),
             login_btn,
-            has_voted,
         ],
-        outputs=[vote_output, has_voted, voting_row],
+        outputs=vote_output,
     )
     downvote_btn.click(
         log_blurb_and_vote,
@@ -211,9 +197,8 @@ with gr.Blocks(theme=tufte_theme) as demo:
             blurb_output,
             gr.Textbox(value="downvote", visible=False),
             login_btn,
-            has_voted,
         ],
-        outputs=[vote_output, has_voted, voting_row],
+        outputs=vote_output,
     )
 
 if __name__ == "__main__":