Spaces:
Building
Building
| import gradio as gr | |
| import yt_dlp | |
| import os | |
| import browser_cookie3 | |
| from huggingface_hub import hf_hub_download, login | |
| if "cookies.txt" not in os.listdir("."): | |
| hf_hub_download("not-lain/secrets","cookies.txt",local_dir=".",repo_type="dataset") | |
| def get_cookies(): | |
| try: | |
| # Try to get cookies from multiple browsers | |
| cookies = [] | |
| for get_cookies in [ | |
| browser_cookie3.firefox, | |
| browser_cookie3.chrome, | |
| browser_cookie3.brave, | |
| ]: | |
| try: | |
| cookies.extend([c for c in get_cookies(domain_name=".youtube.com")]) | |
| except: | |
| continue | |
| return cookies | |
| except: | |
| return None | |
| def download_video(url): | |
| try: | |
| downloads_dir = "/downloads" | |
| if not os.path.exists(downloads_dir): | |
| os.makedirs(downloads_dir, exist_ok=True) | |
| # Use a pre-provided cookies file | |
| cookie_file = "cookies.txt" | |
| cookie_args = {} | |
| if os.path.exists(cookie_file): | |
| cookie_args["cookiefile"] = cookie_file | |
| ydl_opts = { | |
| "outtmpl": os.path.join(downloads_dir, "%(title)s.%(ext)s"), | |
| "format": "mp4/bestvideo+bestaudio/best", | |
| "merge_output_format": "mp4", | |
| **cookie_args, | |
| "paths": {"home": "/downloads", "temp": "/tmp"}, | |
| } | |
| with yt_dlp.YoutubeDL(ydl_opts) as ydl: | |
| info = ydl.extract_info(url, download=True) | |
| video_path = ydl.prepare_filename(info) | |
| if os.path.exists(video_path): | |
| return f"Successfully downloaded: {info['title']}", video_path | |
| else: | |
| return f"Download completed but file not found: {info['title']}", None | |
| except Exception as e: | |
| return f"Error: {str(e)}", None | |
| iface = gr.Interface( | |
| fn=download_video, | |
| inputs=gr.Textbox(label="YouTube URL"), | |
| outputs=[gr.Textbox(label="Status"), gr.Video(label="Downloaded Video")], | |
| title="YouTube Video Downloader", | |
| description="Cookies created using the following extension : https://chromewebstore.google.com/detail/get-cookiestxt-clean/ahmnmhfbokciafffnknlekllgcnafnie?hl=en \n\nEnter a YouTube URL to download and view the video", | |
| examples=["https://www.youtube.com/watch?v=UiyDmqO59QE"], | |
| ) | |
| if __name__ == "__main__": | |
| iface.launch(server_name="0.0.0.0", allowed_paths=["/"], server_port=7860) |