Update db/mongo.py

db/mongo.py

@@ -1,19 +1,9 @@
-import motor.motor_asyncio
+from motor.motor_asyncio import AsyncIOMotorClient
 import os
-from fastapi import HTTPException
-from bson.objectid import ObjectId
 
-# Load MongoDB URI from Hugging Face Space secrets
-MONGODB_URI = os.environ.get("MONGODB_URI")
-if not MONGODB_URI:
-    raise HTTPException(status_code=500, detail="MONGODB_URI not set in environment")
+MONGODB_URI = os.getenv("MONGODB_URI")  # Make sure it's added to your Hugging Face Space secrets
+client = AsyncIOMotorClient(MONGODB_URI)
+db = client["cps_db"]  # Adjust if your database name is different
 
-# Create MongoDB client
-client = motor.motor_asyncio.AsyncIOMotorClient(MONGODB_URI)
-
-# Access the database (you can change the DB name here if needed)
-db = client.get_default_database()
-
-# Access collections
 patients_collection = db["patients"]
-
+results_collection = db["txagent_results"]