Update example questions

app.py

@@ -79,6 +79,14 @@ def display_pdf(pdf):
     st.markdown(pdf_display, unsafe_allow_html=True)
 
 
+examples = [
+    "How climate change impact the world trade in 2022?",
+    "What is trade trending in 2022?",
+    "What is the impact of covid on trade in 2022?",
+    "What is role of WTO in world trade?",
+]
+
+
 def main():
     load_dotenv()
     if "ggml-gpt4all-j-v1.3-groovy.bin" not in os.listdir("/tmp"):
@@ -86,18 +94,23 @@ def main():
                         filename="ggml-gpt4all-j-v1.3-groovy.bin", local_dir="/tmp")
     st.set_page_config(page_title="Trade Document Chatbot")
 
-    st.header("Choose document")
+    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
-                st.success("<-- Chat about the document on the left sidebar")
-                display_pdf(pdf)
+
+    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:
         if "conversation" not in st.session_state:
@@ -107,6 +120,11 @@ def main():
         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)