add topic

app.py

@@ -807,11 +807,12 @@ def chat_paper_main(args):
         reader1.summary_with_chat(paper_list=paper_list)
 
 
-def upload(file):
+def upload(topic_str, file):
     global reader1, paper_list
 
     args = parser.parse_args()
     args.pdf_path = file.name
+    args.key_word = 'education' if len(topic_str) < 1 else topic_str
     reader1, paper_list = chat_paper_main(args=args)
 
     return reader1.summary(paper_list)
@@ -844,6 +845,7 @@ if __name__ == '__main__':
 
     with gr.Blocks(css=".output-image {height: 800px !important}") as demo:
         with gr.Column():
+            topic = gr.Text(label='Topic')
             pdf_file = gr.File(label="上传论文(必须为PDF)")
             summary_input = gr.Text(label='Summary Input')
             summary_output = gr.Text(label='Summary Output')
@@ -855,7 +857,7 @@ if __name__ == '__main__':
             conclusion_output = gr.Text(label='Conclusion Output')
             result_button = gr.Button(value='Next')
             result = gr.Text(label='Result')
-            pdf_file.upload(upload, [pdf_file], [summary_input])
+            pdf_file.upload(upload, [topic, pdf_file], [summary_input])
             method_input_button.click(lambda x: reader1.method(paper_list, x), [summary_output], [method_input])
             conclusion_input_button.click(lambda x, y: reader1.conclusion(paper_list, x, y), [summary_output, method_output], [conclusion_input])
             result_button.click(lambda x, y, z: '\n'.join([x, y, z]), [summary_output, method_output, conclusion_output], [result])