Update app.py

app.py

@@ -37,6 +37,10 @@ def infer(question):
     response = query_engine.query(question)
     return str(response)
 
+def summarize(history, text='Give a brief summary of the paper under the title `Summary` and the summarize each section of this paper seperately in detailed points.'):
+    history = history + [(text, None)]
+    return history, ''
+
 with gr.Blocks(theme='WeixuanYuan/Soft_dark') as demo:
     with gr.Column():
         chatbot = gr.Chatbot([], elem_id='chatbot')
@@ -45,12 +49,14 @@ with gr.Blocks(theme='WeixuanYuan/Soft_dark') as demo:
             paper_id = gr.Textbox(label='ArXiv Paper Id', placeholder='1706.03762')
             langchain_status = gr.Textbox(label='Status', placeholder='', interactive=False)
             load_paper = gr.Button('Load Paper to LLaMa-Index')
+            summary = gr.Button('Generate Summary')
 
         with gr.Row():
             question = gr.Textbox(label='Question', placeholder='Type your query...')
             submit_btn = gr.Button('Submit')
 
     load_paper.click(paper_changes, inputs=[paper_id], outputs=[langchain_status], queue=False)
+    summary.click(summarize, [chatbot], [chatbot]).then(bot, chatbot, chatbot)
     question.submit(add_text, [chatbot, question], [chatbot, question]).then(bot, chatbot, chatbot)
     submit_btn.click(add_text, [chatbot, question], [chatbot, question]).then(bot, chatbot, chatbot)