Update app.py

app.py

@@ -59,9 +59,18 @@ chat_engine = index.as_chat_engine(
 import gradio as gr
 def chat_with_ai(user_input, chat_history):
     response = chat_engine.chat(user_input)
-
-    chat_history.append((user_input, str(response)))
-
+    references = response.source_nodes
+    ref,pages = [],[]
+    for i in range(len(references)):
+      if references[i].metadata['file_name'] not in ref:
+        ref.append(references[i].metadata['file_name'])
+      pages.append(references[i].metadata['page_label'])
+    complete_response = str(response) + "\n\n" + "references:" + str(ref) + "\n\n" + "pages:" + str(pages)
+    if ref !=[] or pages!=[]:
+      chat_history.append((user_input, complete_response))
+    elif ref==[] or pages==[]:
+      chat_history.append((user_input,str(response)))
+        
     return chat_history, ""
 
 def clear_history():