Update indexes.py

indexes.py

@@ -7,10 +7,13 @@ from langchain.document_loaders import DirectoryLoader, TextLoader
 from langchain.vectorstores import Pinecone
 from PyPDF2 import PdfReader
 import pinecone
+from dotenv import load_dotenv
+from langchain.vectorstores import Chroma
 
-from config import config
+load_dotenv()
+openai_api_key=os.getenv('OPENAI_API_KEY')
 
-def create_indexes(file: tempfile, pinecone_api_key: str, pinecone_environment: str, pinecone_index_name: str, openai_api_key: str) -> str:
+def create_indexes(file: tempfile, collection_name: str) -> str:
     try:
         file_path = file.name
         reader = PyPDFLoader(file_path)
@@ -18,35 +21,32 @@ def create_indexes(file: tempfile, pinecone_api_key: str, pinecone_environment:
         embeddings = OpenAIEmbeddings(
             openai_api_key=openai_api_key
         )
-        pinecone.init(
-            api_key=pinecone_api_key,
-            environment=pinecone_environment
-        )
-        indexes_list = pinecone.list_indexes()
-        if pinecone_index_name not in indexes_list:
-            pinecone.create_index(
-                name=pinecone_index_name,
-                dimension=1536
-            )
-        Pinecone.from_documents(
-            documents=documents,
-            embedding=embeddings,
-            index_name=pinecone_index_name
-        )
+        
+        persist_directory = './db_metadata'
+
+        vectordb = Chroma.from_documents(
+        collection_name=collection_name,
+        documents=documents,
+        embedding=embeddings,
+        persist_directory=persist_directory
+    )
+
+        vectordb.persist()
+        
         return 'Document uploaded and index created successfully. You can chat now.'
     except Exception as e:
         return e
 
-def clear_indexes(pinecone_api_key: str, pinecone_environment: str, pinecone_index_name: str) -> str:
-    try:
-        pinecone.init(
-            api_key=pinecone_api_key,
-            environment=pinecone_environment
-        )
-        indexes_list = pinecone.list_indexes()
-        if pinecone_index_name in indexes_list:
-            pinecone.delete_index(name=pinecone_index_name)
-        return 'Indexes cleared.', None
-    except Exception as e:
-        return e, None
+# def clear_indexes(pinecone_api_key: str, pinecone_environment: str, pinecone_index_name: str) -> str:
+#     try:
+#         pinecone.init(
+#             api_key=pinecone_api_key,
+#             environment=pinecone_environment
+#         )
+#         indexes_list = pinecone.list_indexes()
+#         if pinecone_index_name in indexes_list:
+#             pinecone.delete_index(name=pinecone_index_name)
+#         return 'Indexes cleared.', None
+#     except Exception as e:
+#         return e, None