Update config.py

config.py

@@ -1,9 +1,9 @@
 import os
 import logging
-import asyncio
 from fastapi import FastAPI
 from txagent.txagent import TxAgent
 from db.mongo import get_mongo_client
+import asyncio
 
 # Logging
 logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
@@ -20,13 +20,20 @@ users_collection = None
 SECRET_KEY = os.getenv("SECRET_KEY", "your-secret-key")
 ALGORITHM = "HS256"
 
+# Initialize collections synchronously
+db = get_mongo_client()["cps_db"]
+users_collection = db["users"]
+patients_collection = db["patients"]
+analysis_collection = db["patient_analysis_results"]
+alerts_collection = db["clinical_alerts"]
+logger.info("📡 Connected to MongoDB synchronously")
+
 def setup_app(app: FastAPI):
-    global agent, patients_collection, analysis_collection, alerts_collection, users_collection
+    global agent
 
     @app.on_event("startup")
     async def startup_event():
-        global agent, patients_collection, analysis_collection, alerts_collection, users_collection
-
+        global agent
         agent = TxAgent(
             model_name="mims-harvard/TxAgent-T1-Llama-3.1-8B",
             rag_model_name="mims-harvard/ToolRAG-T1-GTE-Qwen2-1.5B",
@@ -43,17 +50,11 @@ def setup_app(app: FastAPI):
         agent.init_model()
         logger.info("✅ TxAgent initialized")
 
-        db = get_mongo_client()["cps_db"]
-        users_collection = db["users"]
-        patients_collection = db["patients"]
-        analysis_collection = db["patient_analysis_results"]
-        alerts_collection = db["clinical_alerts"]
-        logger.info("📡 Connected to MongoDB")
-
+        # Run async analysis task
         asyncio.create_task(analyze_all_patients())
 
     async def analyze_all_patients():
-        from analysis import analyze_patient  # Move import inside to avoid circular imports
+        from analysis import analyze_patient
         patients = await patients_collection.find({}).to_list(length=None)
         for patient in patients:
             await analyze_patient(patient)