fix: dotenv and pinecone index name

db_update/Update.py

@@ -1,16 +1,18 @@
 import sys
 import os
-from dotenv import dotenv_values
+from dotenv import load_dotenv
 from pinecone import Pinecone, ServerlessSpec
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
 
+load_dotenv()
+
 from cybernews.CyberNews import CyberNews
 
-PINECONE_API = dotenv_values(".env").get("PINECONE_API_KEY")
+PINECONE_API = os.getenv("PINECONE_API_KEY")
 
 # Configure client
 pc = Pinecone(api_key=PINECONE_API)
-index_name = str.lower(dotenv_values(".env").get("PINECONE_INDEX_NAME")) # pinecone index name must be in lowercase
+index_name = os.getenv("PINECONE_INDEX_NAME").lower() 
 
 
 

models/NewsModel.py

@@ -1,11 +1,20 @@
 from config.Database import client
 from datetime import datetime
+import os
+from dotenv import load_dotenv
+
+load_dotenv()
 
 class CybernewsDB:
     def __init__(self):
         from sentence_transformers import SentenceTransformer, SparseEncoder
         self.client = client
-        self.index_name = "cybernews-hybrid-test-2"
+
+        index_name = os.getenv("PINECONE_INDEX_NAME")
+        if not index_name:
+            raise ValueError("PINECONE_INDEX_NAME environment variable is missing in .env")
+
+        self.index_name = index_name.lower()
         self.namespace = "c2si" 
         self.index = self.client.Index(self.index_name)