Update app.py

app.py

@@ -108,14 +108,14 @@ class ChatModel:
     def generate(self, message, model_name, vector_store_name, history):
         start_time = time.time()
         self.load_model(model_name)
-        self.load_vector_store(vector_store_name)
+        vectorstore = self.load_vector_store(vector_store_name)
         config = MODEL_CONFIG[model_name]
         
         # Retrieve relevant context
         context = ""
-        if self.vectorstore[vector_store_name]:
-            docs = self.vectorstore[vector_store_name].similarity_search(message, k=3)
-            context = "\n\n".join([d.page_content for d in docs])
+        # if vectorstore:
+        docs = vectorstore.similarity_search(message, k=3)
+        context = "\n\n".join([d.page_content for d in docs])
         
         # Format prompt with context
         prompt = config["template"].format(