init project

.gitignore

@@ -0,0 +1,7 @@
+__pycache__/
+*.pyc
+*.pyo
+*.pyd
+*.log
+*.mp4
+/tmp/*

Dockerfile

@@ -0,0 +1,19 @@
+FROM python:3.10-slim
+
+# Cài đặt hệ thống và yt-dlp
+RUN apt-get update && apt-get install -y wget ffmpeg && \
+    pip install yt-dlp
+
+# Tạo thư mục và sao chép mã nguồn
+WORKDIR /app
+COPY app /app
+COPY requirements.txt .
+
+# Cài đặt thư viện Python
+RUN pip install -r requirements.txt
+
+# Mở cổng API
+EXPOSE 7860
+
+# Chạy FastAPI bằng uvicorn
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

app/main.py

@@ -0,0 +1,40 @@
+from fastapi import FastAPI, Request, BackgroundTasks
+from fastapi.responses import FileResponse, JSONResponse
+from pydantic import BaseModel
+import uuid
+import os
+from app.utils import download_video
+
+app = FastAPI()
+VIDEO_DIR = "/tmp"  # RAM-based storage (HFS cho phép ghi tại đây)
+
+class DownloadRequest(BaseModel):
+    url: str
+    cookie: str  # Nội dung file cookies.txt
+
+@app.post("/download")
+async def download(req: DownloadRequest, background_tasks: BackgroundTasks):
+    video_id = str(uuid.uuid4())
+    filepath = f"{VIDEO_DIR}/{video_id}.mp4"
+    cookie_path = f"/tmp/{video_id}_cookies.txt"
+
+    with open(cookie_path, "w") as f:
+        f.write(req.cookie)
+
+    success = download_video(req.url, cookie_path, filepath)
+    if not success:
+        return JSONResponse({"error": "Download failed"}, status_code=500)
+
+    return {
+        "id": video_id,
+        "download_url": f"/video/{video_id}"
+    }
+
+@app.get("/video/{video_id}")
+def get_video(video_id: str, background_tasks: BackgroundTasks):
+    filepath = f"/tmp/{video_id}.mp4"
+    if not os.path.exists(filepath):
+        return JSONResponse({"error": "Not found"}, status_code=404)
+
+    background_tasks.add_task(os.remove, filepath)
+    return FileResponse(filepath, filename=f"{video_id}.mp4", media_type="video/mp4")

app/utils.py

@@ -0,0 +1,16 @@
+import subprocess
+
+def download_video(url, cookie_file, output_path):
+    try:
+        cmd = [
+            "yt-dlp",
+            "--cookies", cookie_file,
+            "--max-filesize", "500M",
+            "-f", "bestvideo+bestaudio",
+            "-o", output_path,
+            url
+        ]
+        result = subprocess.run(cmd, capture_output=True, text=True)
+        return result.returncode == 0
+    except Exception:
+        return False

requirements.txt

@@ -0,0 +1,2 @@
+fastapi
+uvicorn