Update main.py

main.py

@@ -34,9 +34,11 @@ from google.genai import types
 
 # MongoDB Configuration
 MONGO_PASSWORD = quote_plus(os.getenv("MONGO_PASSWORD"))
-DATABASE_NAME = os.getenv("MONGO_DB_NAME")
-MONGO_URI = os.getenv("CONNECTION_STRING")
-CHAT_COLLECTION = "chat_history"
+MONGO_DATABASE_NAME = os.getenv("DATABASE_NAME")
+MONGO_COLLECTION_NAME = os.getenv("COLLECTION_NAME")
+connection_string_template = os.getenv("CONNECTION_STRING")
+MONGO_CLUSTER_URL = connection_string_template.replace("${PASSWORD}", MONGO_PASSWORD)
+CHAT_COLLECTION = MONGO_COLLECTION_NAME or "chat_history"
 USER_COLLECTION = "users"
 VIDEO_COLLECTION = "videos"
 
@@ -107,8 +109,8 @@ class VideoData(BaseModel):
 # MongoDB connection and chat management
 class MongoDB:
     def __init__(self):
-        self.client = MongoClient(MONGO_URI)
-        self.db = self.client[DATABASE_NAME]
+        self.client = MongoClient(MONGO_CLUSTER_URL)
+        self.db = self.client[MONGO_DATABASE_NAME]
         self.users = self.db[USER_COLLECTION]
         self.videos = self.db[VIDEO_COLLECTION]
         
@@ -176,7 +178,7 @@ class ChatManagement:
 
 # Global variables and instances
 mongodb = MongoDB()
-chat_manager = ChatManagement(MONGO_URI, DATABASE_NAME, CHAT_COLLECTION)
+chat_manager = ChatManagement(MONGO_CLUSTER_URL, MONGO_DATABASE_NAME, CHAT_COLLECTION)
 sessions = {}  # In-memory session storage for retrievers
 
 # Video directory for temporary storage
@@ -246,9 +248,9 @@ def get_llm():
     Returns the language model instance (LLM) using ChatGroq API.
     The LLM used is Llama 3.3 with a versatile 70 billion parameters model.
     """
-    api_key = os.getenv("GROQ_API_KEY", "")
+    api_key = os.getenv("CHATGROQ_API_KEY", "")
     if not api_key:
-        raise ValueError("GROQ_API_KEY environment variable not set")
+        raise ValueError("CHATGROQ_API_KEY environment variable not set")
     
     llm = ChatGroq(
         model="llama-3.3-70b-versatile",