feat: Add source documents on right of display

app.py

@@ -19,6 +19,7 @@ st.title("Sorbobot - Le futur de la recherche scientifique interactive")
 
 chat_column, doc_column = st.columns([2, 1])
 
+
 def initialize_session_state():
     if "history" not in st.session_state:
         st.session_state.history = []
@@ -53,16 +54,20 @@ def initialize_session_state():
             return_source_documents=True,
         )
 
+
 def on_click_callback():
     with get_openai_callback() as cb:
         human_prompt = st.session_state.human_prompt
         llm_response = st.session_state.conversation(human_prompt)
         st.session_state.history.append(Message("human", human_prompt))
         st.session_state.history.append(
-            Message("ai", llm_response["answer"], llm_response["source_documents"])
+            Message(
+                "ai", llm_response["answer"], documents=llm_response["source_documents"]
+            )
         )
         st.session_state.token_count += cb.total_tokens
 
+
 load_css()
 initialize_session_state()
 
@@ -142,4 +147,7 @@ with chat_column:
 with doc_column:
     if len(st.session_state.history) > 0:
         st.markdown("**Source document**")
-        st.markdown(st.session_state.history[-1].metadata)
+        for doc in st.session_state.history[-1].documents:
+            expander = st.expander(doc.metadata["title"])
+            expander.markdown("**" + doc.metadata["doi"] + "**")
+            expander.markdown(doc.page_content)

message.py

@@ -1,5 +1,7 @@
 from dataclasses import dataclass
-from typing import Literal, Optional
+from typing import Literal, Optional, List
+
+from langchain.schema import Document
 
 
 @dataclass
@@ -8,4 +10,4 @@ class Message:
 
     origin: Literal["human", "ai"]
     message: str
-    metadata: Optional[dict] = None
+    documents: Optional[List[Document]] = None