Spaces:
Sleeping
Sleeping
| from fastapi import FastAPI, UploadFile, File, BackgroundTasks | |
| from fastapi.responses import HTMLResponse, FileResponse | |
| import shutil | |
| import os | |
| import uuid | |
| from pathlib import Path | |
| import time | |
| app = FastAPI() | |
| TEMP_DIR = Path("/tmp/uploads") | |
| TEMP_DIR.mkdir(parents=True, exist_ok=True) | |
| # Background cleanup: delete files older than 1 hour | |
| def cleanup_old_files(): | |
| now = time.time() | |
| for f in TEMP_DIR.iterdir(): | |
| if now - f.stat().st_mtime > 3600: | |
| try: | |
| f.unlink() | |
| except Exception: | |
| pass | |
| async def main(): | |
| html_path = Path(__file__).parent / "static" / "index.html" | |
| return HTMLResponse(content=html_path.read_text(), status_code=200) | |
| async def upload_temp_file( | |
| background_tasks: BackgroundTasks, | |
| file: UploadFile = File(...), | |
| ): | |
| # Schedule cleanup task | |
| background_tasks.add_task(cleanup_old_files) | |
| unique_name = f"{uuid.uuid4()}_{file.filename}" | |
| dest_path = TEMP_DIR / unique_name | |
| with dest_path.open("wb") as buf: | |
| shutil.copyfileobj(file.file, buf) | |
| return {"filename": unique_name, "url": f"/files/{unique_name}"} | |
| async def serve_file(filename: str): | |
| file_path = TEMP_DIR / filename | |
| if not file_path.exists(): | |
| return {"error": "File not found"} | |
| return FileResponse(path=str(file_path), filename=filename) |