Update app.py

app.py

@@ -120,12 +120,27 @@ with gr.Blocks() as demo:
         outputs=output_box
     )   
 
+    def submit_feedback(prompt, generated_text, feedback):
+        data = {
+            "prompt": prompt,
+            "generated_text": generated_text,
+            "feedback": feedback
+        }
+        hf_writer.flag(data)  
+        return "Feedback submitted."
+
     feedback_button.click(
-        hf_writer.flag([prompt_box,output_box,feedback_box]),
-        # lambda *args: hf_writer.flag(args),
+        submit_feedback,
         inputs=[prompt_box, output_box, feedback_box],
-        outputs=status_text,
-        )
+        outputs=status_text
+    )
+
+    # feedback_button.click(
+    #     hf_writer.flag([prompt_box,output_box,feedback_box]),
+    #     # lambda *args: hf_writer.flag(args),
+    #     inputs=[prompt_box, output_box, feedback_box],
+    #     outputs=status_text,
+    #     )
     
     # gr.Interface(lambda x:x, "text", "text", allow_flagging="manual", flagging_callback=hf_writer)