Update app.py

app.py

@@ -135,7 +135,7 @@ def get_answer(question):
     # pages
     unique_sources = set()
     for item in result['source_documents']:
-        unique_sources.add(item.metadata['page'])
+        unique_sources.add(item.metadata['source'].split(',')[1])
 
     unique_pages = ""
     for item in unique_sources:
@@ -147,7 +147,7 @@ def get_answer(question):
     # source text
     full_source = ""
     for item in result['source_documents']:
-        full_source += f"- **Page: {item.metadata['page']}**" + "\n" + item.page_content + "\n\n"
+        full_source += f"- **{item.metadata['source']}**" + "\n" + item.page_content + "\n\n"
 
     # will look like:
     # - Page: {number}