update references

app.py

@@ -242,10 +242,33 @@ Question: {query}
 
 Answer:"""
                     
+                    # Build mapping from context index to source label for resolving references
+                    context_index_to_source = {}
+                    for i, result in enumerate(results, 1):
+                        meta = result.get("metadata") or {}
+                        context_index_to_source[i] = (
+                            meta.get("document_title") or meta.get("source") or f"Source {i}"
+                        )
+
                     # Generate response using chat/comversational API (Mistral instruct uses this)
                     try:
                         response_text = self._generate_with_chat(prompt, max_new_tokens=512)
                         if response_text:
+                            # Resolve [Context N] to actual source labels in the body
+                            for i, source_label in context_index_to_source.items():
+                                response_text = response_text.replace(
+                                    f"[Context {i}]",
+                                    f"({source_label})",
+                                )
+                            # Append a References section so users see what each source is
+                            ref_lines = [
+                                "",
+                                "---",
+                                "**References**",
+                            ]
+                            for i, source_label in context_index_to_source.items():
+                                ref_lines.append(f"{i}. {source_label}")
+                            response_text = response_text.rstrip() + "\n\n" + "\n".join(ref_lines)
                             return response_text
                         raise ValueError("Empty response from model")
                     except Exception as api_error: