| import os | |
| import tempfile | |
| from yt_dlp import YoutubeDL | |
| import gradio as gr | |
| def download_video(url): | |
| if not url.strip(): | |
| raise gr.Error("⚠️ Please enter a valid YouTube or TikTok URL.") | |
| temp_dir = tempfile.mkdtemp() | |
| output_path = os.path.join(temp_dir, "%(title)s.%(ext)s") | |
| ydl_opts = { | |
| 'format': 'best', | |
| 'outtmpl': output_path, | |
| 'quiet': True, | |
| 'no_warnings': True, | |
| 'noplaylist': True, | |
| 'socket_timeout': 15, | |
| } | |
| try: | |
| with YoutubeDL(ydl_opts) as ydl: | |
| info = ydl.extract_info(url, download=True) | |
| downloaded_file = ydl.prepare_filename(info) | |
| if not os.path.exists(downloaded_file): | |
| raise gr.Error("❌ Download failed. File not found.") | |
| return downloaded_file | |
| except Exception as e: | |
| err = str(e) | |
| if "Failed to extract" in err or "player response" in err: | |
| raise gr.Error( | |
| "❌ Extraction error — YouTube/TikTok may have changed.\n" | |
| "Try updating yt-dlp using:\n\n`pip install -U yt-dlp`" | |
| ) | |
| elif "resolve" in err: | |
| raise gr.Error("❌ This Space doesn't have internet access. Enable internet in Settings.") | |
| else: | |
| raise gr.Error(f"❌ Error: {err}") | |
| demo = gr.Interface( | |
| fn=download_video, | |
| inputs=gr.Textbox(label="🎥 Enter YouTube or TikTok URL"), | |
| outputs=gr.File(label="📁 Downloaded Video File"), | |
| title="🎬 Video Downloader", | |
| description="Download videos from YouTube or TikTok. Paste a valid video link and get the file.", | |
| theme="default" | |
| ) | |
| if __name__ == "__main__": | |
| demo.launch() | |