Update

app.py

@@ -1,3 +1,4 @@
+import base64
 from datetime import datetime
 import os
 import streamlit as st
@@ -58,11 +59,11 @@ def get_conversation_chain(vectorstore):
 
 def user_input(user_question):
     # log user question with timestamp
-    print(f"[{datetime.now()}]:{user_question}")
+    print(f"[{datetime.now()}]ask:{user_question}")
     with st.spinner("Thinking ..."):
         response = st.session_state.conversation({'question': user_question})
     # log bot answer with timestamp
-    print(f"\n[{datetime.now()}]:{response['answer']}")
+    print(f"\n[{datetime.now()}]ans:{response['answer']}")
     st.session_state.chat_history = response['chat_history']
     for i, messages in enumerate(st.session_state.chat_history):
         if i % 2 == 0:
@@ -71,36 +72,61 @@ def user_input(user_question):
             message(messages.content)
 
 
+def display_pdf(pdf):
+    with open("./docs/"+pdf, "rb") as f:
+        base64_pdf = base64.b64encode(f.read()).decode('utf-8')
+    pdf_display = F'<embed src="data:application/pdf;base64,{base64_pdf}" width="700" height="1000" type="application/pdf">'
+    st.markdown(pdf_display, unsafe_allow_html=True)
+
+
+examples = [
+    "CO2 Emissions trend in 2022?",
+    "What is the average CO2 emissions in 2022?",
+    "CO2 Emissions of the United States in 2022?",
+    "Compare CO2 Emissions of the United States and China in 2022?",
+]
+
+
 def main():
     load_dotenv()
     if "ggml-gpt4all-j-v1.3-groovy.bin" not in os.listdir("/tmp"):
         hf_hub_download(repo_id="dnato/ggml-gpt4all-j-v1.3-groovy.bin",
                         filename="ggml-gpt4all-j-v1.3-groovy.bin", local_dir="/tmp")
-    st.set_page_config(page_title="Document Chatbot")
-    if "conversation" not in st.session_state:
-        st.session_state.conversation = None
-    if "chat_history" not in st.session_state:
-        st.session_state.chat_history = None
-
-    st.header("Query your CO2 documents")
-    user_question = st.text_input("Ask a question about your documents...")
-    if user_question and st.session_state.conversation:
-        user_input(user_question)
+    st.set_page_config(page_title="CO2 Emission Document Chatbot")
+
+    if "pdf" not in st.session_state:
+        st.session_state.pdf = None
+        st.header("Choose document")
+    for pdf in os.listdir("./docs"):
+        if st.button(pdf):
+            with st.spinner("Loading document..."):
+                st.session_state.pdf = pdf
+                pdf_text = get_pdf_text(["./docs/"+pdf])
+                text_chunks = get_text_chunks(pdf_text)
+                vectorstore = get_vectorstore(text_chunks)
+                conversation_chain = get_conversation_chain(vectorstore)
+                st.session_state.conversation = conversation_chain
+                st.session_state.chat_history = None
+
+    if st.session_state.pdf:
+        if st.success("<-- Chat about the document on the left sidebar"):
+            display_pdf(st.session_state.pdf)
+
     with st.sidebar:
-        st.subheader("Your CO2 documents")
-        pdfs = st.file_uploader(
-            "Upload here", accept_multiple_files=True, type=["pdf"],)
-        if st.button("Study"):
-            with st.spinner("Studying ..."):
-                raw_text = get_pdf_text(pdfs)
-                # print(raw_text)
-                chunks = get_text_chunks(raw_text)
-                # print(chunks)
-                vectorstore = get_vectorstore(chunks)
-                # print(vectorstore)
-                st.session_state.conversation = get_conversation_chain(
-                    vectorstore)
-                st.success("Done!")
+        if "conversation" not in st.session_state:
+            st.session_state.conversation = None
+        if "chat_history" not in st.session_state:
+            st.session_state.chat_history = None
+        st.header("Query your document")
+        user_question = st.text_input(
+            "Ask a question about the document...", disabled=not st.session_state.conversation)
+
+        if st.session_state.conversation:
+            for example in examples:
+                if st.button(example):
+                    user_question = example
+        if user_question and st.session_state.conversation:
+            user_input(user_question)
 
 
 if __name__ == '__main__':