Update app.py

app.py

@@ -203,15 +203,21 @@ def save_feedback(inputs):
     return "Thanks your advice!"
         
 
-def yes(message, history):
-    return "yes"
+feedback_data = []
+def chatbot_respond(message, history=[]):
+    response = "yes"
+    return response, history + [(message, response)]
 
+# Functions to handle like/dislike
+def upvote(vote_id):
+    feedback_data.append((vote_id, "upvote"))
+    print(f"Current feedback data: {feedback_data}")
+    return "You liked this response"
 
-def vote(data: gr.LikeData):
-    if data.liked:
-        print("You upvoted this prediction: " + data.value["value"])
-    else:
-        print("You downvoted this prediction: " + data.value["value"])
+def downvote(vote_id):
+    feedback_data.append((vote_id, "downvote"))
+    print(f"Current feedback data: {feedback_data}")
+    return "You disliked this response"
 
         
 # Define the FAPM interface
@@ -242,6 +248,7 @@ css = """
 
 with gr.Blocks(css=css) as demo:
     gr.Markdown(description)
+    vote_id = gr.State(0)
     with gr.Tab(label="Protein caption"):
         with gr.Row():
             with gr.Column():
@@ -252,14 +259,13 @@ with gr.Blocks(css=css) as demo:
                 # output_text = gr.Textbox(label="Output Text")
                 with gr.Accordion('Prediction:', open=True):
                     output_markdown = gr.Markdown(label="Output")
-        with gr.Row():
-            chatbot = gr.Chatbot(placeholder="<strong>Is this prediction helpful?</strong><br>")
-            chatbot.like(vote, None, None)
-        # with gr.Row(label="Your feedback"):
-        #     with gr.Column():
-        #         inputs = gr.Textbox(type="text")
-        #         output_markdown = gr.Markdown(label="Output")
-        #         feedback_btn = gr.Button(value="Submit")
+            upvote_button = gr.Button("👍")
+            downvote_button = gr.Button("👎")
+
+            with gr.Column():
+                inputs = gr.Textbox(type="text")
+                feedback_markdown = gr.Markdown(label="Output")
+                feedback_btn = gr.Button(value="Submit")
         # O14813 train index 127, 266, 738, 1060 test index 4
         gr.Examples(
             examples=[
@@ -277,7 +283,9 @@ with gr.Blocks(css=css) as demo:
             label='Try examples'
         )
         submit_btn.click(generate_caption, [input_protein, prompt], [output_markdown])
-        # feedback_btn.click(save_feedback, [inputs], [output_markdown])
+        upvote_button.click(upvote, vote_id, None)
+        downvote_button.click(downvote, vote_id, None)
+        feedback_btn.click(save_feedback, [inputs], [feedback_markdown])
 
 demo.launch(debug=True)