Update ccc.py

ccc.py

@@ -7,11 +7,21 @@ import os
 import logging
 from fastapi import FastAPI
 import uvicorn
+from contextlib import asynccontextmanager
 
 # Disable all logging except for critical errors
 logging.basicConfig(level=logging.CRITICAL)
 
-app = FastAPI()
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    # Startup
+    initialize_base_ids()
+    asyncio.create_task(start_scanner())
+    yield
+    # Shutdown
+    # Add any cleanup code here if needed
+
+app = FastAPI(lifespan=lifespan)
 
 @app.get("/")
 async def root():
@@ -150,7 +160,6 @@ async def print_status_periodically():
         await asyncio.sleep(60)
 
 async def start_scanner():
-    initialize_base_ids()
     tasks = []
     instances_per_digit = 20000
     
@@ -161,9 +170,5 @@ async def start_scanner():
     tasks.append(print_status_periodically())
     await asyncio.gather(*tasks)
 
-@app.on_event("startup")
-async def startup_event():
-    asyncio.create_task(start_scanner())
-
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    uvicorn.run(app, host="0.0.0.0", port=7860)