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)