Update app.py

app.py

@@ -606,13 +606,29 @@ def wait_for_key(key_name="OPENAI_API_KEY", timeout=10):
 # =============================
 # Step 5: Chat Function
 # =============================
+
+def format_answer(result):
+    answer = result["answer"]
+    sources = result.get("sources", [])
+
+    formatted_sources = "\n".join([
+        f"- {s['document']} → {s['section']} / {s['subsection']} / {s['subsubsection']}"
+        for s in sources
+    ])
+
+    return f"""{answer}
+
+📄 **Sources**
+{formatted_sources}
+"""
+
 def chat_fn(message, history):
     global retriever
     wait_for_key()
     if retriever is None:
         return "⚠️ Retriever not initialized. Please rebuild or check vector DB."
     answer = generate_rag_response(message, retriever)
-    return f"{answer}\n\n🧠 (Context retrieved from {pdf_path})"
+    return format_answer #f"{answer}\n\n🧠 (Context retrieved from {pdf_path})"
 
 
 # =============================