Update main.py

main.py

@@ -52,18 +52,18 @@ def LLModel():
     
 def VectorDatabase(categorie):
     if categorie != "year":
-        os.environ['PINECONE_API_KEY'] = ""
-        os.environ['PINECONE_API_KEY'] = os.getenv('PINECONE_API_KEY')
         index_name = "all-venus"
+        embeddings = HuggingFaceEmbeddings()
+        vectorstore = PineconeVectorStore(
+            index_name=index_name, embedding=embeddings, pinecone_api_key=os.getenv('PINECONE_API_KEY')
+        )
     else:
-        os.environ['PINECONE_API_KEY'] = ""
-        os.environ['PINECONE_API_KEY'] = os.getenv('PINECONE_API_KEYJDLP')
         index_name = "all-jdlp"
-
-    embeddings = HuggingFaceEmbeddings()
-    vectorstore = PineconeVectorStore(
-        index_name=index_name, embedding=embeddings
-    )
+        embeddings = HuggingFaceEmbeddings()
+        vectorstore = PineconeVectorStore(
+            index_name=index_name, embedding=embeddings, pinecone_api_key=os.getenv('PINECONE_API_KEYJDLP')
+        )
+    
     return vectorstore
 
 def Retriever(categorie):