Update functions.py

functions.py

@@ -50,7 +50,22 @@ async def handle_userinput(user_question, custom_graph):
             st.subheader("Your documents")
             with st.spinner("Processing"):
                 for doc in docs:
-                    st.write(f"Document: {doc.page_content}")  # Assuming doc.page_content contains the text
+                    # Extract document content
+                    content = doc.page_content  # Assuming the document has a `page_content` attribute
+                    
+                    # Extract document metadata if available
+                    metadata = doc.metadata if hasattr(doc, 'metadata') else {}
+
+                    # Display content and metadata
+                    st.write(f"Document: {content}")
+                    
+                    # Display metadata (assuming it's a dictionary)
+                    if metadata:
+                        st.write("Metadata:")
+                        for key, value in metadata.items():
+                            st.write(f"- {key}: {value}")
+                    else:
+                        st.write("No metadata available.")
 
         # Check if a response (generation) was produced by the graph
         if 'generation' in state_dict and state_dict['generation']:
@@ -66,8 +81,7 @@ async def handle_userinput(user_question, custom_graph):
     except Exception as e:
         # Display an error message in case of failure
         st.chat_message("assistant").write("An error occurred: Try to change the question.")
-        st.chat_message("assistant").write(e) 
-
+        st.chat_message("assistant").write(e)
 
 
 def create_retriever_from_chroma(vectorstore_path="./docs/chroma/", search_type='mmr', k=7, chunk_size=550, chunk_overlap=40):