Update app.py

app.py

@@ -92,23 +92,69 @@ Allocation:
 Interventional Model:
 Masking:"""
 
-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)
+hf_writer = gr.HuggingFaceDatasetSaver("criteria-feedback-demo",token, 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
+    )   
+
+    feedback_button.click(
+        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)
+
+    # feedback_button.click(
+    #     save_feedback,
+    #     inputs=[prompt_box, output_box, feedback_box],
+    #     outputs=status_text
+    # )
+
+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)