Update app.py

app.py

@@ -30,10 +30,13 @@ chat_prompt = ChatPromptTemplate.from_messages(
 HF_TOKEN = os.getenv('HF_TOKEN')  # replace 'HF_TOKEN' with the name of your environment variable
 hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "CenterFor/HeyJesus_Feedback")
 
+# This needs to be called at some point prior to the first call to hf_writer.flag()
+hf_writer.setup([chatbot, msg], "flagged_conversations")
+
 with gr.Blocks() as demo:
     chatbot = gr.Chatbot()
     msg = gr.Textbox()
-    clear = gr.ClearButton([msg, chatbot])
+    flag = gr.Button("Flag")
 
     def user(user_message, history):
         return gr.update(value="", interactive=False), history + [[user_message, None]]
@@ -55,8 +58,8 @@ with gr.Blocks() as demo:
     )
     response.then(lambda: gr.update(interactive=True), None, [msg], queue=False)
 
-demo.allow_flagging = "manual"  # allow manual flagging
-demo.flagging_callback = hf_writer  # set the flagging callback to hf_writer
+    # We can choose which components to flag -- in this case, we'll flag all of them
+    flag.click(lambda *args: hf_writer.flag(args), [chatbot, msg], None, preprocess=False)
 
 demo.queue()
 demo.launch()