updates on the requirements.txt

app.py

@@ -72,12 +72,12 @@ with gr.Blocks(theme="dark") as demo:
             # Download
             with gr.Group():
                 gr.Markdown(f'<center><h1>Download</h1></center>')
-                gr.Markdown(f'<center><p>Download the processed data and corrected data</p></center>')
-                answer_file = gr.File(label='Download processed data', file_types=['.xlsx'])
-                btn_download_answer = gr.Button(value='Download processed data')
+                gr.Markdown(f'<center><p>Download the original LLM answers and corrected LLM answers</p></center>')
+                answer_file = gr.File(label='Download original LLM answers', file_types=['.xlsx'])
+                btn_download_answer = gr.Button(value='Download original LLM answers')
                 btn_download_answer.style(full_width=True)
                 corrected_file = gr.File(label='Download corrected data', file_types=['.xlsx'])
-                btn_download_corrected = gr.Button(value='Download corrected data')
+                btn_download_corrected = gr.Button(value='Download corrected LLM answers')
                 btn_download_corrected.style(full_width=True)
             
         

backend.py

@@ -155,6 +155,7 @@ class Backend:
     def download_answer(self, path = './tmp', name = 'answer.xlsx'):
         os.makedirs(path, exist_ok = True)
         path = os.path.join(path, name)
+        self.ori_answer_df['questions'] = self.questions
         self.ori_answer_df.to_excel(path, index = False)
 
         return path
@@ -162,6 +163,7 @@ class Backend:
     def download_corrected(self, path = './tmp', name = 'corrected_answer.xlsx'):
         os.makedirs(path, exist_ok = True)
         path = os.path.join(path, name)
+        self.answer_df['questions'] = self.questions
         self.answer_df.to_excel(path, index = False)
 
         return path