Update db/mongo.py

db/mongo.py

@@ -1,25 +1,22 @@
 import os
 from motor.motor_asyncio import AsyncIOMotorClient
 
+# MongoDB URI (must be set in Hugging Face Secrets)
 MONGO_URI = os.environ.get("MONGODB_URI")
 
-# Extract database name from the URI manually
 if not MONGO_URI:
-    raise ValueError("MONGODB_URI environment variable not set")
+    raise ValueError("MONGODB_URI environment variable is not set.")
 
-# Example: mongodb+srv://user:pass@host/mydatabase
-db_name = MONGO_URI.rsplit("/", 1)[-1].split("?")[0]
-if not db_name:
-    raise ValueError("No default database found in MONGODB_URI")
-
-# Initialize client
+# Initialize MongoDB client
 client = AsyncIOMotorClient(MONGO_URI)
-db = client[db_name]  # use the extracted database
+
+# Use the correct database name explicitly
+db = client["cps_db"]
 
 # Collections
 patients_collection = db["patients"]
 results_collection = db["patient_analysis_results"]
 
-# Optional: export client or db if needed elsewhere
+# Optional utility function to reuse the client
 def get_mongo_client():
     return client