Spaces:
Running
Running
| """Endpoint dùng chung: trạng thái job, tải file kết quả.""" | |
| from __future__ import annotations | |
| import os | |
| from urllib.parse import quote | |
| from fastapi import APIRouter, HTTPException | |
| from fastapi.responses import FileResponse | |
| from ..core.files import session_dir, delete_session | |
| from ..core.jobs import jobs | |
| router = APIRouter(prefix="/api", tags=["common"]) | |
| def drop_session(sid: str): | |
| """Xoá phiên tạm sau khi user đã tải kết quả về máy (không bắt buộc).""" | |
| return {"deleted": delete_session(sid), "session": sid} | |
| def job_status(job_id: str): | |
| job = jobs.get(job_id) | |
| if not job: | |
| raise HTTPException(404, "job không tồn tại") | |
| return job.to_dict() | |
| def queue_stats(): | |
| """Số liệu hàng đợi: số job đang chạy/chờ theo loại + tổng job trong bộ nhớ.""" | |
| return jobs.stats() | |
| def job_cancel(job_id: str): | |
| jobs.cancel(job_id) | |
| return {"ok": True} | |
| def list_inputs(sid: str): | |
| """Liệt kê video trong thư mục inputs của phiên (dùng khi đẩy từ Down list).""" | |
| d = os.path.join(session_dir(sid), "inputs") | |
| if not os.path.isdir(d): | |
| return {"session": sid, "files": []} | |
| out = [] | |
| for name in sorted(os.listdir(d)): | |
| full = os.path.join(d, name) | |
| if os.path.isfile(full): | |
| out.append({ | |
| "file": f"inputs/{name}", | |
| "name": name, | |
| "size": os.path.getsize(full), | |
| "url": f"/api/files/{sid}/inputs/{quote(name)}", | |
| }) | |
| return {"session": sid, "files": out} | |
| def get_file(sid: str, path: str): | |
| base = os.path.abspath(session_dir(sid)) | |
| # Chấp cả "/" lẫn "\" trong path (Windows) -> chuẩn hoá trước khi nối. | |
| rel = path.replace("\\", "/").strip("/") | |
| full = os.path.abspath(os.path.join(base, *rel.split("/"))) | |
| # Phải nằm trong phiên VÀ là FILE thật: trỏ vào THƯ MỤC thì FileResponse | |
| # ném lỗi 500, trong khi đúng nghĩa là 404. | |
| if not full.startswith(base) or not os.path.isfile(full): | |
| raise HTTPException(404, "file không tồn tại") | |
| # inline (KHÔNG attachment) -> trình duyệt render video/ảnh trực tiếp trong <video>. | |
| # Nút tải vẫn hoạt động nhờ thuộc tính `download` phía frontend. | |
| return FileResponse(full, filename=os.path.basename(full), | |
| content_disposition_type="inline") | |