fix

app.py

@@ -7,7 +7,7 @@ from datetime import datetime
 from pathlib import Path
 
 import gradio as gr
-from huggingface_hub import CommitScheduler, InferenceClient, get_token, login
+from huggingface_hub import CommitScheduler, get_token, login
 from openai import OpenAI
 
 from prompts import basic_prompt, detailed_genre_description_prompt
@@ -16,6 +16,8 @@ from theme import TufteInspired
 # Ensure you're logged in to Hugging Face
 login(os.getenv("HF_TOKEN"))
 
+
+
 client = OpenAI(
     base_url="https://api-inference.huggingface.co/models/meta-llama/Meta-Llama-3-70B-Instruct/v1",
     api_key=get_token(),
@@ -25,13 +27,11 @@ client = OpenAI(
 dataset_folder = Path("dataset")
 dataset_folder.mkdir(exist_ok=True)
 
-
 # Function to get the latest dataset file
 def get_latest_dataset_file():
     files = list(dataset_folder.glob("data_*.jsonl"))
     return max(files, key=os.path.getctime) if files else None
 
-
 # Check for existing dataset and create or append to it
 if latest_file := get_latest_dataset_file():
     dataset_file = latest_file
@@ -47,26 +47,23 @@ scheduler = CommitScheduler(
     repo_type="dataset",
     folder_path=dataset_folder,
     path_in_repo="data",
-    every=1,  # Upload every 5 minutes
+    every=1,  # Upload every minute
 )
 
-# Add a dictionary to store votes
+# Global dictionary to store votes
 votes = {}
 
-
 def generate_prompt():
     if random.choice([True, False]):
         return detailed_genre_description_prompt()
     else:
         return basic_prompt()
 
-
 def get_and_store_prompt():
     prompt = generate_prompt()
     print(prompt)  # Keep this for debugging
     return prompt
 
-
 def generate_blurb(prompt):
     max_tokens = random.randint(100, 1000)
     chat_completion = client.chat.completions.create(
@@ -82,19 +79,16 @@ def generate_blurb(prompt):
         full_text += message.choices[0].delta.content
         yield full_text
 
-
 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:
         gr.Info("You've already voted on this blurb!")
+        return None, gr.Row.update(visible=False)
 
     votes[vote_id] = vote
 
@@ -108,9 +102,9 @@ def log_blurb_and_vote(prompt, blurb, vote, user_info: gr.OAuthProfile | None, *
     with scheduler.lock:
         with dataset_file.open("a") as f:
             f.write(json.dumps(log_entry) + "\n")
+    
     gr.Info("Thank you for voting! Your feedback will be synced to the dataset.")
-    return f"Logged: {vote} by user {user_id}"
-
+    return f"Logged: {vote} by user {user_id}", gr.Row.update(visible=False)
 
 # Create custom theme
 tufte_theme = TufteInspired()
@@ -123,7 +117,6 @@ with gr.Blocks(theme=tufte_theme) as demo:
     Would you read a book based on this LLM generated blurb? <br> Your vote will be added to <a href="https://huggingface.co/datasets/your-username/your-dataset-repo">this</a> Hugging Face dataset</p>"""
     )
 
-    # Add the login button
     with gr.Row():
         login_btn = gr.LoginButton(size="sm")
     with gr.Row():
@@ -137,12 +130,12 @@ with gr.Blocks(theme=tufte_theme) as demo:
         upvote_btn = gr.Button("👍 would read")
         downvote_btn = gr.Button("👎 wouldn't read")
 
-    vote_output = gr.Textbox(label="Vote Status", interactive=False, visible=False)
+    vote_output = gr.Textbox(label="Vote Status", interactive=False, visible=True)
 
     def generate_and_show(prompt, user_info):
-        # Reset votes for new blurb
-        global votes
-        votes = {}
+        # Optionally clear votes for the previous blurb if needed
+        # global votes
+        # votes = {k: v for k, v in votes.items() if not k.endswith(hash(previous_blurb))}
         return "Generating...", gr.Row.update(visible=False), user_info
 
     def show_voting_buttons(blurb):
@@ -164,7 +157,7 @@ with gr.Blocks(theme=tufte_theme) as demo:
             gr.Textbox(value="upvote", visible=False),
             user_state,
         ],
-        outputs=vote_output,
+        outputs=[vote_output, voting_row],
     )
     downvote_btn.click(
         log_blurb_and_vote,
@@ -174,7 +167,7 @@ with gr.Blocks(theme=tufte_theme) as demo:
             gr.Textbox(value="downvote", visible=False),
             user_state,
         ],
-        outputs=vote_output,
+        outputs=[vote_output, voting_row],
     )
 
 if __name__ == "__main__":