Update app.py

app.py

@@ -5,9 +5,9 @@ import gradio as gr
 import pickle
 from query_data import get_chain
 from threading import Lock
+from ingest_data import get_vectorstore
+
 
-with open("vectorstore.pkl", "rb") as f:
-    vectorstore = pickle.load(f)
 
 
 
@@ -21,6 +21,11 @@ class ChatWrapper:
         """Execute the chat functionality."""
         self.lock.acquire()
         try:
+            vs = get_vectorstore("vectorstore.pkl")
+            
+            with open(vs "rb") as f:
+                vectorstore = pickle.load(f)
+                
             history = history or []
             chain = get_chain(vectorstore)
             # If chain is None, that is because no API key was provided.