Update db/mongo.py

db/mongo.py

@@ -1,15 +1,25 @@
 import os
 from motor.motor_asyncio import AsyncIOMotorClient
 
-# Get the MongoDB URI from Hugging Face secrets
 MONGO_URI = os.environ.get("MONGODB_URI")
 
-# Initialize the async MongoDB client
-client = AsyncIOMotorClient(MONGO_URI)
+# Extract database name from the URI manually
+if not MONGO_URI:
+    raise ValueError("MONGODB_URI environment variable 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")
 
-# Automatically get the default database (from the URI)
-db = client.get_default_database()
+# Initialize client
+client = AsyncIOMotorClient(MONGO_URI)
+db = client[db_name]  # use the extracted database
 
-# Expose collections
+# Collections
 patients_collection = db["patients"]
 results_collection = db["patient_analysis_results"]
+
+# Optional: export client or db if needed elsewhere
+def get_mongo_client():
+    return client