Update app.py

app.py

@@ -160,6 +160,12 @@ def demo():
         
         gr.Markdown("<b>Chat with Your Document</b>")
         chatbot = gr.Chatbot(height=505)
+        doc_source1 = gr.Textbox(label="Reference 1", lines=2, interactive=False)
+        source1_page = gr.Number(label="Page", interactive=False)
+        doc_source2 = gr.Textbox(label="Reference 2", lines=2, interactive=False)
+        source2_page = gr.Number(label="Page", interactive=False)
+        doc_source3 = gr.Textbox(label="Reference 3", lines=2, interactive=False)
+        source3_page = gr.Number(label="Page", interactive=False)
         msg = gr.Textbox(placeholder="Ask a question", container=True)
         submit_btn = gr.Button("Submit")
         clear_btn = gr.ClearButton([msg, chatbot], value="Clear")
@@ -168,14 +174,14 @@ def demo():
         db_btn.click(initialize_database, inputs=[document], outputs=[vector_db, db_progress])
         qachain_btn.click(initialize_LLM, inputs=[llm_btn, slider_temperature, slider_maxtokens, slider_topk, vector_db], outputs=[qa_chain, llm_progress]).then(lambda: [None, "", 0, "", 0, "", 0],
             inputs=None,
-            outputs=[chatbot, None, None, None, None, None, None],
+            outputs=[chatbot, doc_source1, source1_page, doc_source2, source2_page, doc_source3, source3_page],
             queue=False)
 
-        msg.submit(conversation, inputs=[qa_chain, msg, chatbot, persona_text], outputs=[qa_chain, msg, chatbot, None, None, None, None], queue=False)
-        submit_btn.click(conversation, inputs=[qa_chain, msg, chatbot, persona_text], outputs=[qa_chain, msg, chatbot, None, None, None, None], queue=False)
-        clear_btn.click(lambda: [None, "", 0, "", 0, "", 0], inputs=None, outputs=[chatbot])
+        msg.submit(conversation, inputs=[qa_chain, msg, chatbot, persona_text], outputs=[qa_chain, msg, chatbot, doc_source1, source1_page, doc_source2, source2_page, doc_source3, source3_page], queue=False)
+        submit_btn.click(conversation, inputs=[qa_chain, msg, chatbot, persona_text], outputs=[qa_chain, msg, chatbot, doc_source1, source1_page, doc_source2, source2_page, doc_source3, source3_page], queue=False)
+        clear_btn.click(lambda: [None, "", 0, "", 0, "", 0], inputs=None, outputs=[chatbot, doc_source1, source1_page, doc_source2, source2_page, doc_source3, source3_page])
 
     demo.queue().launch(debug=True)
 
 if __name__ == "__main__":
-    demo()
+    demo()