comeon

main.py

@@ -1,7 +1,8 @@
-from fastapi import FastAPI, File, UploadFile
+from fastapi import FastAPI, File, UploadFile, BackgroundTasks
 from fastapi.responses import HTMLResponse, FileResponse
 import shutil
 import os
+import time
 
 app = FastAPI()
 
@@ -32,7 +33,7 @@ async def read_root():
     return html_content.format(file_list=file_list)
 
 @app.post("/uploadfile/")
-async def upload_file(file: UploadFile = File(...)):
+async def upload_file(file: UploadFile = File(...), background_tasks: BackgroundTasks):
     # Save the uploaded file to the uploads directory
     upload_directory = "uploads"
     os.makedirs(upload_directory, exist_ok=True)  # Create the directory if it doesn't exist
@@ -41,6 +42,9 @@ async def upload_file(file: UploadFile = File(...)):
     with open(file_location, "wb") as buffer:
         shutil.copyfileobj(file.file, buffer)
 
+    # Schedule the file deletion after 10 minutes
+    background_tasks.add_task(delete_file_after_delay, file_location, 600)  # 600 seconds = 10 minutes
+
     return {"filename": file.filename}
 
 def get_uploaded_files():
@@ -54,3 +58,9 @@ def get_uploaded_files():
 @app.get("/files/{filename}")
 async def get_file(filename: str):
     return FileResponse(f"uploads/{filename}")
+
+def delete_file_after_delay(file_path: str, delay: int):
+    time.sleep(delay)  # Wait for the specified delay
+    if os.path.exists(file_path):
+        os.remove(file_path)  # Delete the file
+        print(f"Deleted file: {file_path}")  # Log the deletion (optional)