update to log prompt

app.py

@@ -26,10 +26,14 @@ def generate_prompt():
         return basic_prompt()
 
 
-def generate_blurb():
-    max_tokens = random.randint(100, 1000)
+def get_and_store_prompt():
     prompt = generate_prompt()
-    print(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(
         model="tgi",
         messages=[
@@ -45,10 +49,11 @@ def generate_blurb():
 
 
 # Function to log blurb and vote
-def log_blurb_and_vote(blurb, vote, user_info: gr.OAuthProfile | None, *args):
+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(),
+        "prompt": prompt,
         "blurb": blurb,
         "vote": vote,
         "user_id": user_id,
@@ -75,26 +80,46 @@ with gr.Blocks(theme=tufte_theme) as demo:
 
     with gr.Row():
         generate_btn = gr.Button("Create a book", variant="primary")
+
+    prompt_state = gr.State()
     blurb_output = gr.Markdown(label="Book blurb")
+
     with gr.Row(visible=False) as voting_row:
         upvote_btn = gr.Button("👍 would read")
         downvote_btn = gr.Button("👎 wouldn't read")
+
     vote_output = gr.Textbox(label="Vote Status", interactive=False, visible=False)
 
+    def generate_and_show(prompt):
+        return gr.Markdown.update(value="Generating..."), gr.Row(visible=False)
+
     def show_voting_buttons(blurb):
         return blurb, gr.Row(visible=True)
 
-    generate_btn.click(generate_blurb, outputs=blurb_output).then(
+    generate_btn.click(get_and_store_prompt, outputs=prompt_state).then(
+        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]
     )
+
     upvote_btn.click(
         log_blurb_and_vote,
-        inputs=[blurb_output, gr.Textbox(value="upvote", visible=False), login_btn],
+        inputs=[
+            prompt_state,
+            blurb_output,
+            gr.Textbox(value="upvote", visible=False),
+            login_btn,
+        ],
         outputs=vote_output,
     )
     downvote_btn.click(
         log_blurb_and_vote,
-        inputs=[blurb_output, gr.Textbox(value="downvote", visible=False), login_btn],
+        inputs=[
+            prompt_state,
+            blurb_output,
+            gr.Textbox(value="downvote", visible=False),
+            login_btn,
+        ],
         outputs=vote_output,
     )