Spaces:
Running
Running
| import os | |
| import re | |
| import uuid | |
| import glob | |
| import shutil | |
| import zipfile | |
| import subprocess | |
| import gradio as gr | |
| def backup_tiktok(url): | |
| """ | |
| Downloads TikTok video(s) from a provided URL using yt-dlp. | |
| If multiple videos are downloaded (e.g., from a TikTok profile), | |
| return a zip of all videos. Otherwise, return the single video file. | |
| """ | |
| # Log the initiation of the download | |
| print(f"DOWNLOAD INITIATED: {url}") | |
| # Validate URL | |
| if "tiktok.com" not in url.lower(): | |
| return "Please provide a valid TikTok Account or Video URL (must contain 'tiktok.com')." | |
| # Create a unique temporary folder | |
| download_id = str(uuid.uuid4()) | |
| download_folder = f"downloads_{download_id}" | |
| os.makedirs(download_folder, exist_ok=True) | |
| # Run yt-dlp command | |
| cmd = ["yt-dlp", url, "-o", f"{download_folder}/%(title)s.%(ext)s"] | |
| try: | |
| subprocess.run(cmd, check=True) | |
| except subprocess.CalledProcessError as e: | |
| print(f"yt-dlp encountered an error: {str(e)}") | |
| # Gather all downloaded files | |
| video_files = glob.glob(f"{download_folder}/*") | |
| if len(video_files) == 0: | |
| print(f"DOWNLOAD FAILED: No videos found for {url}") | |
| return "No videos found." | |
| # If only one video, return it directly | |
| if len(video_files) == 1: | |
| print(f"DOWNLOAD SUCCESSFUL: {url}") | |
| return video_files[0] | |
| # Otherwise, zip all files and return the zip | |
| zip_name = f"tiktok_backup_{download_id}.zip" | |
| with zipfile.ZipFile(zip_name, "w", zipfile.ZIP_DEFLATED) as zipf: | |
| for file_path in video_files: | |
| arcname = os.path.basename(file_path) | |
| zipf.write(file_path, arcname=arcname) | |
| print(f"DOWNLOAD SUCCESSFUL: {url}") | |
| return zip_name | |
| with gr.Blocks() as demo: | |
| gr.Markdown(""" | |
| # TikTok Account/Video Backup | |
| # Set your profile to "Public" or your videos will not download. | |
| Enter a TikTok Account or Video URL to download the video(s). | |
| """) | |
| with gr.Row(): | |
| url_input = gr.Textbox(label="TikTok URL", placeholder="https://www.tiktok.com/@username/video/...") | |
| with gr.Row(): | |
| disclaimer = gr.Checkbox(label="I agree that I am only downloading videos for the purpose of archiving my own content, that no copyrighted material is being infringed, and that I take full responsibility for this action.", value=False) | |
| with gr.Row(): | |
| download_button = gr.Button("Download", interactive=False) | |
| disclaimer.change(lambda x: gr.update(interactive=x), inputs=disclaimer, outputs=download_button) | |
| output_file = gr.File(label="Downloaded File or ZIP") | |
| download_button.click(fn=backup_tiktok, inputs=url_input, outputs=output_file) | |
| demo.launch(server_name="0.0.0.0", show_api=False) | |