Update matt_function.py

matt_function.py

@@ -1,15 +1,5 @@
-from langchain.llms import OpenAI
-from langchain.document_loaders import PyPDFLoader
-from langchain.embeddings import OpenAIEmbeddings
-from langchain.vectorstores import Chroma
-
-# Import OpenAI as main LLM service
-from langchain.llms import OpenAI
-from langchain.embeddings import OpenAIEmbeddings
-
-# Import PDF document loaders...there's other ones as well!
-from langchain.document_loaders import PyPDFLoader
-from langchain.indexes import VectorstoreIndexCreator
+from llama_index.llms import OpenAI
+from llama_index import VectorStoreIndex, SimpleDirectoryReader
 
 
 
@@ -17,11 +7,10 @@ def create_VSI(files, index):   #add vsi and filepaths
     
     pathnames = [file.name for file in files]
 
-    loaders = []
-    for pdf in pathnames:
-        loader = PyPDFLoader(pdf)
-        loaders.append(loader)
+    documents = SimpleDirectoryReader(input_files=pathnames).load_data()
+
+    index = VectorStoreIndex.from_documents(documents)
 
-    index = VectorstoreIndexCreator().from_loaders(loaders)
+    query_engine = index.as_query_engine()
 
-    return index, "Upload complete"
+    return query_engine, "Upload complete"