added consulting retrieved docs feature

app.py

@@ -84,7 +84,7 @@ else:
             full_response = ""
             # Define the basic input structure for the chains
             input_dict = {"input": prompt.lower()}
-
+            used_docs = retriever.get_relevant_documents(prompt.lower())
 
             with collect_runs() as cb:
                 for chunk in chain.stream(input_dict, config={"tags": ["SUP'ASSISTANT"]}):
@@ -93,6 +93,22 @@ else:
                 memory.save_context(input_dict, {"output": full_response})
                 st.session_state.run_id = cb.traced_runs[0].id
             message_placeholder.markdown(full_response)
+            if used_docs :
+                docs_content = "\n\n".join(
+                                        [
+                                            f"Doc {i+1}:\n"
+                                            f"Source: {doc.metadata['source']}\n"
+                                            f"Title: {doc.metadata['title']}\n"
+                                            f"Content: {doc.page_content}\n"
+                                            for i, doc in enumerate(used_docs)
+                                        ]
+                                    )
+                st.download_button(
+                    label="Consulted Documents",
+                    data=docs_content,
+                    file_name="Consulted_documents.txt",
+                    mime="text/plain",
+                )
 
             with st.spinner("Just a sec! Dont enter prompts while loading pelase!"):
                 run_id = st.session_state.run_id
@@ -181,4 +197,4 @@ else:
                     ],
                     parallel=4,
                     max_retries=3,
-                    )
+                    )