update source messaging

app.py

@@ -116,8 +116,18 @@ async def main(message: cl.Message):
     )
 
     answer = response["answer"]
-    source_documents = response["source_documents"][0].metadata["source"]
-
-    content = answer + source_documents
+    source_pdf = response["source_documents"][0].metadata["source"]
+    if source_pdf == "Blueprint-for-an-AI-Bill-of-Rights.pdf":
+        source = "Blueprint for an AI Bill of Rights"
+    elif source_pdf == "AI_Risk_Management_Framework.pdf":
+        source = "AI Risk Management Framework"
+    else:
+        source = source_pdf
+
+    page_number = str(response["source_documents"][0].metadata["page"])
+
+    content = (
+        response["answer"] + f"\n\n(**Source**: {source}, **Page**: {page_number})"
+    )
 
     await cl.Message(content=content, author="AI").send()