Update app.py

app.py

@@ -13,6 +13,8 @@ import uvicorn
 import shutil
 import pandas as pd
 import uuid
+import atexit
+import shutil
 
 app = FastAPI()
 
@@ -219,6 +221,12 @@ async def validate_video(file: UploadFile = File(...), mission_type: str = Form(
 async def root():
     return {"status": "Video validation API is running."}
 
+def cleanup():
+    print("Membersihkan cache dan file sementara...")
+    shutil.rmtree("/tmp", ignore_errors=True)
+    shutil.rmtree("/root/.cache", ignore_errors=True)
+
+atexit.register(cleanup)
 
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)