Update src/database.py

src/database.py

@@ -7,9 +7,17 @@ from dotenv import load_dotenv
 load_dotenv()
 
 # MongoDB Configuration
-MONGO_URI = os.getenv("MONGO_URI")
-DB_NAME = os.getenv("MONGO_DB")
-COLLECTION_NAME = os.getenv("MONGO_COLLECTION")
+MONGO_USERNAME = os.getenv("MONGO_USERNAME")
+MONGO_PASSWORD = os.getenv("MONGO_PASSWORD")
+MONGO_HOST = os.getenv("MONGO_HOST")
+MONGO_DB = os.getenv("MONGO_DB")
+COLLECTION_NAME = os.getenv("MONGO_COLLECTION", "analyses")
+
+# Build MongoDB URI
+if MONGO_USERNAME and MONGO_PASSWORD:
+    MONGO_URI = f"mongodb://{MONGO_USERNAME}:{MONGO_PASSWORD}@{MONGO_HOST}/{MONGO_DB}"
+else:
+    MONGO_URI = f"mongodb://{MONGO_HOST}/{MONGO_DB}"
 
 # Global variables for lazy initialization
 _client = None
@@ -23,7 +31,7 @@ def get_database_connection():
     if _client is None:
         _client = MongoClient(MONGO_URI, serverSelectionTimeoutMS=5000)
         _client.admin.command('ping')
-        _db = _client[DB_NAME]
+        _db = _client[MONGO_DB]
         _collection = _db[COLLECTION_NAME]
     
     return _client, _db, _collection