Update app.py

app.py

@@ -19,9 +19,18 @@ with gr.Blocks() as demo:
                 msg = gr.Textbox(label='Your prompt')
 
             with gr.Column(scale=0.3, min_width=500):
-                for i in range(args.choices):
+                for i in range(args.show_c):
                     with gr.Accordion(f"Answer {i+1}", open=False) as answer_area:
+                        with gr.Row():
+                            gr.Button('👍 Answer').style(size='sm', full_width=False)
+                            gr.Button('👎🏻 Answer').style(size='sm', full_width=False)
                         context_area = gr.Markdown(f"Context {i+1}")
+                        # gr.CheckboxGroup(choices=['Good Answer', 'Bad Answer', 'Good Context', 'Bad Context'])
+
+                        with gr.Row():
+                            gr.Button('👍 Context').style(size='sm', full_width=False)
+                            gr.Button('👎🏻 Context').style(size='sm', full_width=False)
+
                         RLHF.append(answer_area)
                         RLHF.append(context_area)
                 clear_chat = gr.Button("Clear history")
@@ -45,24 +54,24 @@ with gr.Blocks() as demo:
 
     def bot(history):
         question = history[-1][0]
-        print('User mess:', question)
         answers = chatbot.asking(question)
 
 
         print(answers)
         history[-1][1] = ""
-        for character in answers[0]:
+        for character in answers:
             history[-1][1] += character
             time.sleep(0.01)
             yield history
     
     def fill_feedback(prompt):
-        contexts, answers = chatbot.temp
+        results = chatbot.temp
         outptus = []
-        for i in range(args.choices):
+        for i in range(args.show_c):
             if i < chatbot.choices:
-                outptus.append(RLHF[2*i].update(label=answers[i], visible=True))
-                outptus.append(contexts[i]['context'])
+                answer = f"{results[i]['score']}% | {results[i]['answer']}"
+                outptus.append(RLHF[2*i].update(label=answer, visible=True, open=False))
+                outptus.append(results[i]['context'])
             else:
                 outptus.append(RLHF[2*i].update(visible=False))
                 outptus.append("No context")