Update app.py

Test HuggingFaceDatasetSaver

app.py

@@ -95,46 +95,46 @@ Masking:"""
 
    
 hf_writer = gr.HuggingFaceDatasetSaver("criteria-feedback-demo",token_w, private=True)
-with gr.Blocks() as demo:
-    with gr.Row():
-        with gr.Column():
-            prompt_box = gr.Textbox(
-                label="Research Information",
-                placeholder=place_holder,
-                value=prefilled_value,
-                lines=10)
-            submit_button = gr.Button("Generate")
-        with gr.Column():
-            output_box = gr.Textbox(
-                label="Eligiblecriteria Criteria",
-                lines=21,
-                interactive=False)
-    with gr.Row():
-        with gr.Column():
-            feedback_box = gr.Textbox(label="Enter your feedback here...", lines=3, interactive=True)
-            feedback_button = gr.Button("Submit Feedback")
-            status_text = gr.Textbox(label="Status", lines=1, interactive=False)
-
-    submit_button.click(
-        run_model_on_text,
-        inputs=prompt_box,
-        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(
-        submit_feedback,
-        inputs=[prompt_box, output_box, feedback_box],
-        outputs=status_text
-    )
+# with gr.Blocks() as demo:
+#     with gr.Row():
+#         with gr.Column():
+#             prompt_box = gr.Textbox(
+#                 label="Research Information",
+#                 placeholder=place_holder,
+#                 value=prefilled_value,
+#                 lines=10)
+#             submit_button = gr.Button("Generate")
+#         with gr.Column():
+#             output_box = gr.Textbox(
+#                 label="Eligiblecriteria Criteria",
+#                 lines=21,
+#                 interactive=False)
+#     with gr.Row():
+#         with gr.Column():
+#             feedback_box = gr.Textbox(label="Enter your feedback here...", lines=3, interactive=True)
+#             feedback_button = gr.Button("Submit Feedback")
+#             status_text = gr.Textbox(label="Status", lines=1, interactive=False)
+
+#     submit_button.click(
+#         run_model_on_text,
+#         inputs=prompt_box,
+#         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(
+#         submit_feedback,
+#         inputs=[prompt_box, output_box, feedback_box],
+#         outputs=status_text
+#     )
 
     # feedback_button.click(
     #     hf_writer.flag([prompt_box,output_box,feedback_box]),
@@ -151,28 +151,31 @@ with gr.Blocks() as demo:
     #     outputs=status_text
     # )
 
-demo.launch()
+# demo.launch()
 
 
 
 #----------------------------------
-# prompt_box = gr.Textbox(
-#     lines=25,
-#     label="Research Information",
-#     placeholder=place_holder,
-#     value=prefilled_value,
-# )
-
-# output_box = gr.Textbox(
-#     lines=25,
-#     label="Eligiblecriteria Criteria",
-# )
-
-# demo = gr.Interface(
-#   fn=run_model_on_text,
-#   inputs=prompt_box,
-#   outputs=output_box,
-#   allow_flagging='auto',
-# )
-
-# demo.queue(max_size=20).launch(debug=True, share=True)
+prompt_box = gr.Textbox(
+    lines=25,
+    label="Research Information",
+    placeholder=place_holder,
+    value=prefilled_value,
+)
+
+output_box = gr.Textbox(
+    lines=25,
+    label="Eligiblecriteria Criteria",
+)
+
+demo = gr.Interface(
+  fn=run_model_on_text,
+  inputs=prompt_box,
+  outputs=output_box,
+  allow_flagging="manual",
+    flagging_options=["incorrect", "inappropriate", "appropriate"],
+    flagging_callback=hf_writer
+)
+)
+
+demo.queue(max_size=20).launch(debug=True, share=True)