Update app.py

app.py

@@ -3,10 +3,21 @@ import streamlit as st
 
 from langchain.embeddings import HuggingFaceInstructEmbeddings, HuggingFaceEmbeddings
 from langchain.vectorstores.faiss import FAISS
-from langchain.chains import VectorDBQA
+from langchain.chains import ChatVectorDBChain
 from huggingface_hub import snapshot_download
-from langchain import OpenAI
-from langchain import PromptTemplate
+from langchain.chat_models import ChatOpenAI
+from langchain.prompts.chat import (
+    ChatPromptTemplate,
+    SystemMessagePromptTemplate,
+    AIMessagePromptTemplate,
+    HumanMessagePromptTemplate,
+)
+from langchain.schema import (
+    AIMessage,
+    HumanMessage,
+    SystemMessage
+)
+
 
 
 st.set_page_config(page_title="CFA Level 1", page_icon="📖")
@@ -144,7 +155,7 @@ def get_answer(question):
 api_key = os.environ["OPENAI_API_KEY"]   
 
 ##### main ####
-user_input = st.text_input("Your question", "Who are you?", key="input")
+user_input = st.text_input("Your question", "What is an MBS and who are the main issuer and investors of the MBS market?", key="input")
 
 col1, col2 = st.columns([10, 1])
 
@@ -160,8 +171,8 @@ if ask:
     with st.spinner("this can take about a minute for your first question because some models have to be downloaded 🥺👉🏻👈🏻"):
         try:
             answer, pages, extract = get_answer(question=user_input)
-        except:
-            st.write(f"Error with Download")
+        except as e:
+            st.write(f"Error with Download: {e}")
             st.stop()
 
     st.write(f"{answer}")