Update api.py

api.py

@@ -5,7 +5,9 @@ import random
 import string
 import numpy as np
 import logging
+import aysncio
 import datetime
+from contextlib import asynccontextmanager
 from fastapi import FastAPI, HTTPException, Request, Response
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel, constr, conint
@@ -711,6 +713,7 @@ async def generate_video(req: GenerateImageRequest):
                 raise HTTPException(status_code=500, detail=f"Failed to generate video after multiple attempts: {str(e)}")
             continue  # Retry on transient errors
 
+@asynccontextmanager
 @app.on_event("shutdown")
 def shutdown_event():
     """ Perform any cleanup activities on shutdown. """