Upload 4 files

agent_t5.py

@@ -69,12 +69,12 @@ class Agent:
             sequence = self.tokenizer.decode(output, skip_special_tokens=True)
             answers.append(sequence)
 
-        self.temp = contexts
+        self.temp = [contexts, answers]
         t_mess = "t_query: {:.2f}\t t_token: {:.2f}\t t_gen: {:.2f}\t t_decode: {:.2f}\t".format(
             s_token-s_query, s_gen-s_token, s_de-s_gen, time.time()-s_de
         )
         print(t_mess)
-        return answers, contexts
+        return answers
 
 
     

app.py

@@ -8,8 +8,7 @@ from retrieval.retrieval import BM25
 
 args = Config()
 chatbot = Agent(args)
-answer_areas = list(range(args.choices))
-context_areas = list(range(args.choices))
+RLHF = []
 
 with gr.Blocks() as demo:
     # gr.Markdown("Flip text or image files using this demo.")
@@ -19,10 +18,12 @@ with gr.Blocks() as demo:
                 chatbot_area = gr.Chatbot().style(height=700)
                 msg = gr.Textbox(label='Your prompt')
 
-            with gr.Column(scale=0.15, min_width=300):
+            with gr.Column(scale=0.3, min_width=500):
                 for i in range(args.choices):
-                    with gr.Accordion(f"Answer: {i+1}", open=False) as answer_areas[i]:
-                        context_areas[i] = gr.Markdown(f"Context {i+1}") 
+                    with gr.Accordion(f"Answer {i+1}", open=False) as answer_area:
+                        context_area = gr.Markdown(f"Context {i+1}")
+                        RLHF.append(answer_area)
+                        RLHF.append(context_area)
                 clear_chat = gr.Button("Clear history")
 
     with gr.Tab("Your context"):
@@ -45,11 +46,7 @@ with gr.Blocks() as demo:
     def bot(history):
         question = history[-1][0]
         print('User mess:', question)
-        answers, contexts = chatbot.asking(question)
-
-        for i in range(chatbot.choices):
-            context_areas[i].value = contexts[i]['context']
-            answer_areas[i].value = answers[i]
+        answers = chatbot.asking(question)
 
 
         print(answers)
@@ -58,14 +55,32 @@ with gr.Blocks() as demo:
             history[-1][1] += character
             time.sleep(0.01)
             yield history
+    
+    def fill_feedback(prompt):
+        contexts, answers = chatbot.temp
+        outptus = []
+        for i in range(args.choices):
+            if i < chatbot.choices:
+                outptus.append(RLHF[2*i].update(label=answers[i], visible=True))
+                outptus.append(contexts[i]['context'])
+            else:
+                outptus.append(RLHF[2*i].update(visible=False))
+                outptus.append("No context")
+
+        return outptus
+
+    def clear_history(history):
+        history = []
+        return history
 
     msg.submit(user, [msg, chatbot_area], [msg, chatbot_area], queue=False).then(
         bot, chatbot_area, chatbot_area
-    )
+    ).then(fill_feedback, msg, RLHF)
 
     context_btt.click(chatbot.get_context, [context_box, ])
     upload_btt.upload(chatbot.load_context, [upload_btt, ], context_box)
     clear_context_btt.click(chatbot.clear_context, outputs=context_box)
+    clear_chat.click(clear_history, chatbot_area, chatbot_area)
 
 demo.queue()
 demo.launch()