update logging

app.py

@@ -2,39 +2,28 @@ import gradio as gr
 import json
 from datetime import datetime
 from theme import TufteInspired
-import glob
-import os
 import uuid
-from pathlib import Path
 from huggingface_hub import InferenceClient
 from openai import OpenAI
-from huggingface_hub import get_token
-from huggingface_hub import CommitScheduler, hf_hub_download, login
-
+from huggingface_hub import get_token, login
 from prompts import detailed_genre_description_prompt, basic_prompt
 import random
-from huggingface_hub import login, get_token
+import os
 
+# Ensure you're logged in to Hugging Face
 login(get_token())
-# TODOs
-# 1. Add a login button
-# 2. Prompt library expand
-# 3. log user if logged in
-
 
 client = OpenAI(
     base_url="https://api-inference.huggingface.co/models/meta-llama/Meta-Llama-3-70B-Instruct/v1",
     api_key=get_token(),
 )
 
-
 def generate_prompt():
     if random.choice([True, False]):
         return detailed_genre_description_prompt()
     else:
         return basic_prompt()
 
-
 def generate_blurb():
     max_tokens = random.randint(100, 1000)
     prompt = generate_prompt()
@@ -52,15 +41,23 @@ def generate_blurb():
         full_text += message.choices[0].delta.content
         yield full_text
 
-
 # Function to log blurb and vote
-def log_blurb_and_vote(blurb, vote):
-    log_entry = {"timestamp": datetime.now().isoformat(), "blurb": blurb, "vote": vote}
+def log_blurb_and_vote(blurb, vote, user_info: gr.OAuthProfile | None, *args):
+    if user_info is not None:
+        user_id = user_info.username
+    else:
+        user_id = str(uuid.uuid4())
+
+    log_entry = {
+        "timestamp": datetime.now().isoformat(),
+        "blurb": blurb,
+        "vote": vote,
+        "user_id": user_id
+    }
     with open("blurb_log.jsonl", "a") as f:
         f.write(json.dumps(log_entry) + "\n")
-        gr.Info("Thank you for voting!")
-    return f"Logged: {vote}"
-
+    gr.Info("Thank you for voting!")
+    return f"Logged: {vote} by user {user_id}"
 
 # Create custom theme
 tufte_theme = TufteInspired()
@@ -71,10 +68,11 @@ with gr.Blocks(theme=tufte_theme) as demo:
     gr.Markdown(
         """<p style='text-align: center;'>Looking for your next summer read? 
     Would you read a book based on this LLM generated blurb? <br> Your vote will be added to <a href="https://example.com">this</a> Hugging Face dataset</p>"""
-        + """"""
     )
-    gr.LoginButton(size="sm")
-    # user_name = gr.Textbox(label="User Name", placeholder="Enter your name")
+    
+    # Add the login button
+    login_btn = gr.LoginButton()
+    
     with gr.Row():
         generate_btn = gr.Button("Create a book", variant="primary")
     blurb_output = gr.Markdown(label="Book blurb")
@@ -90,16 +88,15 @@ with gr.Blocks(theme=tufte_theme) as demo:
         show_voting_buttons, inputs=blurb_output, outputs=[blurb_output, voting_row]
     )
     upvote_btn.click(
-        lambda x: log_blurb_and_vote(x, "upvote"),
-        inputs=blurb_output,
+        log_blurb_and_vote,
+        inputs=[blurb_output, gr.Textbox(value="upvote", visible=False), login_btn],
         outputs=vote_output,
     )
     downvote_btn.click(
-        lambda x: log_blurb_and_vote(x, "downvote"),
-        inputs=blurb_output,
+        log_blurb_and_vote,
+        inputs=[blurb_output, gr.Textbox(value="downvote", visible=False), login_btn],
         outputs=vote_output,
     )
 
-
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(debug=True)