import tkinter as tk
from tkinter import messagebox
from tkinter import ttk
import yt_dlp
import os

def download_with_ytdlp(url, status_label):
    if not url:
        messagebox.showwarning("入力エラー", "URLを入力してください。")
        return

    status_label.config(text="📥 ダウンロード中...")
    root.update()

    try:
        ydl_opts = {
            'format': 'bv*+ba/best',
            'outtmpl': '%(title).80s.%(ext)s',  # 現在のフォルダにタイトル名で保存
            'merge_output_format': 'mp4',
        }

        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
            info = ydl.extract_info(url, download=True)
            filename = ydl.prepare_filename(info).replace('.webm', '.mp4').replace('.mkv', '.mp4')

        status_label.config(text="✅ ダウンロード完了！")
        messagebox.showinfo("完了", f"動画が保存されました:\n{filename}")

    except Exception as e:
        status_label.config(text="❌ エラーが発生しました")
        messagebox.showerror("エラー", f"処理中に問題が発生しました:\n{str(e)}")

# GUIセットアップ
root = tk.Tk()
root.title("動画ダウンロードツール")
root.geometry("500x200")

tk.Label(root, text="🎯 ダウンロードする動画のURLを入力:").pack(pady=10)
url_entry = tk.Entry(root, width=60)
url_entry.pack()

status_label = tk.Label(root, text="", fg="green")
status_label.pack(pady=10)

download_button = ttk.Button(root, text="ダウンロード", command=lambda: download_with_ytdlp(url_entry.get(), status_label))
download_button.pack(pady=5)

root.mainloop()