Upload 3 files

Dockerfile

@@ -0,0 +1,16 @@
+# Read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
+# you will also find guides on how best to write your Dockerfile
+
+FROM python:3.9
+
+RUN useradd -m -u 1000 user
+USER user
+ENV PATH="/home/user/.local/bin:$PATH"
+
+WORKDIR /app
+
+COPY --chown=user ./requirements.txt requirements.txt
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
+
+COPY --chown=user . /app
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -0,0 +1,41 @@
+from fastapi import FastAPI, HTTPException
+from fastapi.responses import StreamingResponse
+import yt_dlp
+import os
+
+app = FastAPI()
+
+# Temporary download directory
+DOWNLOAD_DIR = "downloads/"
+
+# Ensure the directory exists
+if not os.path.exists(DOWNLOAD_DIR):
+    os.makedirs(DOWNLOAD_DIR)
+
+@app.get("/")
+def greet_json():
+    return {"Hello": "World!"}
+
+@app.get("/download_video")
+def download_video(url: str):
+    try:
+        # Set up yt-dlp options to avoid downloading .webm files
+        ydl_opts = {
+            'format': 'bestvideo+bestaudio/best',  # Download the best video and audio
+            'outtmpl': DOWNLOAD_DIR + '%(id)s.%(ext)s',  # Output path template
+            'noplaylist': True,  # Avoid downloading playlists, if it's a playlist URL
+            'quiet': True,  # Suppress unnecessary output
+        }
+
+        # Download video using yt-dlp
+        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+            info_dict = ydl.extract_info(url, download=True)
+            video_filename = ydl.prepare_filename(info_dict)
+
+        # Return the video file as a response
+        video_file = open(video_filename, "rb")
+        return StreamingResponse(video_file, media_type="video/mp4")
+
+    except Exception as e:
+        raise HTTPException(status_code=400, detail=str(e))
+

requirements.txt

@@ -0,0 +1,3 @@
+fastapi
+uvicorn
+yt-dlp