update source display

app.py

@@ -59,12 +59,13 @@ vectordb_p = Pinecone.from_existing_index(index_name, embeddings)
 # print(type(rr))
 # print(rr)
 def chathmi(message, history):
-    response = "I don't know"
-    print(message)
-    response = QAQuery_p(message)
+    # response = "I don't know"
+    # print(message)
+    response, source = QAQuery_p(message)
     time.sleep(0.3)
     print(history)
     yield response
+    yield source
     # yield history
 
 def chathmi2(message, history):
@@ -77,7 +78,7 @@ def chathmi2(message, history):
 # chatbot = gr.Chatbot(color_map =("blue", "pink"))
 
 demo = gr.ChatInterface(
-    chathmi2,
+    chathmi,
     title="STLA BABY - YOUR FRIENDLY GUIDE",
     )
 
@@ -121,7 +122,8 @@ def QAQuery_p(question: str):
     print("Source:", res['source_documents'])
     response = res['result']
     # response = res['source_documents']
-    return response
+    source = res['source_documents']
+    return response, source
 
 def CreatDb():
     global vectordb