updates on the new prompt and better features

app.py

@@ -44,7 +44,7 @@ with gr.Blocks(theme="dark") as demo:
                 gr.Markdown(f'<center><p>The answer to your question is :</p></center>')
                 question_box = gr.Textbox(label='Question')
                 answer_box = gr.Textbox(label='Answer')
-                reference_box = gr.Textbox(label='Reference')
+                # reference_box = gr.Textbox(label='Reference')
 
                 highlighted_text = gr.outputs.HTML(label="Highlighted Text")
                 with gr.Row():
@@ -103,7 +103,7 @@ with gr.Blocks(theme="dark") as demo:
     btn_submit_txt.click(
             backend.process_file,
             inputs=[file, questions, openai_key],
-            outputs=[question_box, answer_box, reference_box, highlighted_text, correct_answer, correct_reference],
+            outputs=[question_box, answer_box, highlighted_text, correct_answer, correct_reference],
         )
     
     btn_submit_correctness.click(   # TODO
@@ -115,12 +115,12 @@ with gr.Blocks(theme="dark") as demo:
     # Switch question button
     btn_last_question.click(
             backend.process_last,
-            outputs=[question_box, answer_box, reference_box, highlighted_text, correct_answer, correct_reference, save_results, answer_correct, reference_correct],
+            outputs=[question_box, answer_box, highlighted_text, correct_answer, correct_reference, save_results, answer_correct, reference_correct],
         )
 
     btn_next_question.click(
             backend.process_next,
-            outputs=[question_box, answer_box, reference_box, highlighted_text, correct_answer, correct_reference, save_results, answer_correct, reference_correct],
+            outputs=[question_box, answer_box, highlighted_text, correct_answer, correct_reference, save_results, answer_correct, reference_correct],
         )
     
     # Download button

backend.py

@@ -75,7 +75,7 @@ class Backend:
         highlighted_out_html = self.highlight_text(self.text, self.highlighted_out)
         self.highlighted_out = '\n'.join(self.highlighted_out)
         
-        return question, self.answer, self.highlighted_out, highlighted_out_html, self.answer, self.highlighted_out
+        return question, self.answer, highlighted_out_html, self.answer, self.highlighted_out
     
     def process_results(self, answer_correct, correct_answer, reference_correct, correct_reference):
         if not hasattr(self, 'clicked_correct_answer'):
@@ -122,7 +122,7 @@ class Backend:
 
         if self.curret_question >= self.totel_question:
             # self.curret_question -= 1
-            return "No more questions!", "No more questions!", "No more questions!", "No more questions!", 'No more questions!', 'No more questions!', 'Still need to click the button above to save the results', None, None
+            return "No more questions!", "No more questions!", "No more questions!", 'No more questions!', 'No more questions!', 'Still need to click the button above to save the results', None, None
         else:
             res = self.gpt_result[f'Question {self.curret_question + 1}']
             question = self.questions[self.curret_question]
@@ -130,7 +130,7 @@ class Backend:
             self.highlighted_out = res['original sentences']
             highlighted_out_html = self.highlight_text(self.text, self.highlighted_out)
             self.highlighted_out = '\n'.join(self.highlighted_out)
-            return question, self.answer,self.highlighted_out, highlighted_out_html, 'Please judge on the generated answer', 'Please judge on the generated answer', 'Still need to click the button above to save the results', None, None
+            return question, self.answer, highlighted_out_html, 'Please judge on the generated answer', 'Please judge on the generated answer', 'Still need to click the button above to save the results', None, None
 
     def process_last(self):
         self.curret_question -= 1
@@ -142,7 +142,7 @@ class Backend:
         
         if self.curret_question < 0:
             # self.curret_question += 1
-            return "No more questions!", "No more questions!", "No more questions!", "No more questions!", 'No more questions!', 'No more questions!', 'Still need to click the button above to save the results', None, None
+            return "No more questions!", "No more questions!", "No more questions!", 'No more questions!', 'No more questions!', 'Still need to click the button above to save the results', None, None
         else:
             res = self.gpt_result[f'Question {self.curret_question + 1}']
             question = self.questions[self.curret_question]
@@ -150,7 +150,7 @@ class Backend:
             self.highlighted_out = res['original sentences']
             highlighted_out_html = self.highlight_text(self.text, self.highlighted_out)
             self.highlighted_out = '\n'.join(self.highlighted_out)
-            return question, self.answer, self.highlighted_out, highlighted_out_html, 'Please judge on the generated answer', 'Please judge on the generated answer', 'Still need to click the button above to save the results', None, None
+            return question, self.answer, highlighted_out_html, 'Please judge on the generated answer', 'Please judge on the generated answer', 'Still need to click the button above to save the results', None, None
         
     def download_answer(self, path = './tmp', name = 'answer.xlsx'):
         os.makedirs(path, exist_ok = True)