Spaces:
Running
Running
File size: 2,007 Bytes
6841e1f 035be31 6841e1f 035be31 6841e1f |
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 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 |
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.
"""
# 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:
return f"Error downloading: {str(e)}"
# Gather all downloaded files
video_files = glob.glob(f"{download_folder}/*")
if len(video_files) == 0:
return "No videos found."
# If only one video, return it directly
if len(video_files) == 1:
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)
return zip_name
with gr.Blocks() as demo:
gr.Markdown("# TikTok Account/Video Backup\n\nEnter 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/...")
download_button = gr.Button("Download")
output_file = gr.File(label="Downloaded File or ZIP")
download_button.click(fn=backup_tiktok, inputs=url_input, outputs=output_file)
demo.launch()
|