Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import time
 import os
+import pandas as pd
 from collections.abc import Collection
 from langchain.memory import ChatMessageHistory
 from langchain_community.chat_message_histories import (
@@ -147,12 +148,19 @@ for msg in history.messages:
 
 # preprocessing context
 def format_docs_with_metadata(docs):
-    formatted_docs = []
-    for i, doc in enumerate(docs, start=1):
-        metadata_str = "\n".join([f"**{key}**: `{value}`\n" for key, value in doc.metadata.items() if key != "embedding"])
-        formatted_doc = f"- {doc.page_content}\n\n**Metadata:**\n{metadata_str}"
-        formatted_docs.append(formatted_doc)   
-    return "\n\n".join(formatted_docs)
+    # formatted_docs = []
+    # for i, doc in enumerate(docs, start=1):
+    #     metadata_str = "\n".join([f"**{key}**: `{value}`\n" for key, value in doc.metadata.items() if key != "embedding"])
+    #     formatted_doc = f"- {doc.page_content}\n\n**Metadata:**\n{metadata_str}"
+    #     formatted_docs.append(formatted_doc)   
+    # return "\n\n".join(formatted_docs)
+    rows = []
+    for doc in docs:
+        row = {'Content': doc.page_content}
+        row.update({k: v for k, v in doc.metadata.items() if k != 'embedding'})
+        rows.append(row)
+    return pd.DataFrame(rows)
+    
 
 def stream_data(response):
   for word in response.split(" "):
@@ -169,6 +177,7 @@ if prompt := st.chat_input():
 
     with st.chat_message("AI"):
       st.write_stream(stream_data(res['answer']))
-      with st.popover("View Source"):
-        st.markdown("### Source Alerts 📢")
-        st.markdown(format_docs_with_metadata(res['context']))
+      with st.expander("View Source"):
+        st.subheader("Source Alerts 📢")
+        df = format_docs_as_table(res['context'])
+        st.table(df)