Spaces:
Running
Running
File size: 1,633 Bytes
0b5dc96 6cc6875 0b5dc96 6cc6875 0b5dc96 6cc6875 0b5dc96 6cc6875 0b5dc96 6cc6875 0b5dc96 6cc6875 0b5dc96 6cc6875 0b5dc96 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 |
from fastapi import FastAPI, HTTPException
from fastapi.responses import StreamingResponse
import yt_dlp
import os
from pathlib import Path
app = FastAPI()
# Temporary directory for downloaded videos
DOWNLOAD_DIR = Path("downloads/")
# Ensure the directory exists
if not DOWNLOAD_DIR.exists():
DOWNLOAD_DIR.mkdir(parents=True)
@app.get("/")
def greet_json():
return {"Hello": "World!"}
@app.get("/download_video")
def download_video(url: str):
try:
# Set up yt-dlp options to download the best video/audio
ydl_opts = {
'format': 'bestvideo+bestaudio/best', # Download the best video and audio
'outtmpl': str(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
}
# Create a temporary filename for the video
video_filename = DOWNLOAD_DIR / "temp_video.mp4"
# Use yt-dlp to download the video
with yt_dlp.YoutubeDL(ydl_opts) as ydl:
info_dict = ydl.extract_info(url, download=True)
video_filename = ydl.prepare_filename(info_dict)
# Check if the video was successfully downloaded
if not video_filename.exists():
raise HTTPException(status_code=500, detail="Video download failed.")
# Open the video file and stream it
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))
|