Update app.py

app.py

@@ -1,268 +1,728 @@
 #!/usr/bin/env python3
-# -*- coding: utf-8 -*-
 """
 YouTube Video Analyzer & Downloader Pro
-───────────────────────────────────────
-· app.py 와 같은 폴더에  𝐰𝐰𝐰.𝐲𝐨𝐮𝐭𝐮𝐛𝐞.𝐜𝐨𝐨𝐤𝐢𝐞𝐬.𝐭𝐱𝐭  이 있으면 자동 사용
-· Gradio UI에서 쿠키 파일 업로드 시 → 업로드 파일이 우선
-· Gemini-1.5 Flash 장면 분석(선택적), 전체 Transcript, 다운로드 지원
+(쿠키 자동 처리 버전)
+
+· `www.youtube.com_cookies.txt` 파일이 **app.py**와 같은 폴더에 있으면
+  자동으로 사용합니다.
+· Gradio UI에서 쿠키 파일을 업로드하면, 업로드된 파일이 **우선** 적용됩니다.
 """
 
-# ───────────────────────── 표준 라이브러리 ─────────────────────────
-import os, re, shutil, tempfile
-from datetime import datetime, timedelta
+# ──────────────────────────────────────────────────────────────
+# 표준 라이브러리
+# ──────────────────────────────────────────────────────────────
+import os
+import re
+import json
+import uuid
+import shutil
+import tempfile
+from datetime import datetime
 from pathlib import Path
 
-# ───────────────────────── 서드파티 라이브러리 ────────────────────
+# ──────────────────────────────────────────────────────────────
+# 외부 라이브러리
+# ──────────────────────────────────────────────────────────────
 import gradio as gr
 import yt_dlp
 import google.generativeai as genai
-from youtube_transcript_api import YouTubeTranscriptApi
-
-# ───────────────────────── 상수 (절대경로) ─────────────────────────
-DEFAULT_COOKIE_FILE = (Path(__file__).resolve().parent / "www.youtube.com_cookies.txt").resolve()
-
-# 유튜브 URL 정규식
-_YT_RE = re.compile(
-    r"(https?://)?(www\.)?"
-    r"(youtube|youtu|youtube-nocookie)\.(com|be)/"
-    r"(watch\?v=|embed/|v/|.+\?v=)?([^&=%\?]{11})"
-)
-
-# ───────────────────────── Helper (ID, 자막) ─────────────────────
-def extract_video_id(url: str) -> str | None:
-    m = _YT_RE.match(url)
-    return m.group(6) if m else None
-
-def fetch_transcript(video_id: str, pref=("ko", "en")) -> str:
-    tr = None
-    for lang in pref:
-        try:
-            tr = YouTubeTranscriptApi.get_transcript(video_id, languages=[lang])
-            break
-        except Exception:
-            continue
-    if tr is None:
-        tr = YouTubeTranscriptApi.get_transcript(video_id)
-    lines = []
-    for seg in tr:
-        t = str(timedelta(seconds=int(seg["start"])))
-        mmss = ":".join(t.split(":")[-2:])
-        lines.append(f"**[{mmss}]** {seg['text']}")
-    return "\n".join(lines)
+
+# ──────────────────────────────────────────────────────────────
+# 기본 쿠키 파일 경로 ― 파일명이 동일하면 자동 사용
+# ──────────────────────────────────────────────────────────────
+DEFAULT_COOKIE_FILE = Path(__file__).with_name("www.youtube.com_cookies.txt")
+
 
 # =================================================================
 # Main Class
 # =================================================================
 class YouTubeDownloader:
     def __init__(self):
-        # temp dirs
-        self.temp_dir = tempfile.mkdtemp(prefix="yt_tmp_")
-        self.temp_downloads = tempfile.mkdtemp(prefix="yt_dl_")
-        # Downloads target
-        self.dl_folder = Path.home() / "Downloads" / "YouTube_Downloads"
-        self.dl_folder.mkdir(parents=True, exist_ok=True)
-        # Gemini
+        # 임시 디렉터리 (Gradio 호환)
+        self.download_dir = tempfile.mkdtemp()
+        self.temp_downloads = tempfile.mkdtemp(prefix="youtube_downloads_")
+
+        # 사용자 Downloads 하위 폴더
+        self.downloads_folder = os.path.join(
+            os.path.expanduser("~"), "Downloads", "YouTube_Downloads"
+        )
+        os.makedirs(self.downloads_folder, exist_ok=True)
+
         self.gemini_model = None
 
-    # ───────── Gemini 설정 ─────────
-    def configure_gemini(self, api_key: str):
+    # ---------------------------------------------------------
+    # Google Gemini API
+    # ---------------------------------------------------------
+    def configure_gemini(self, api_key):
         try:
             genai.configure(api_key=api_key)
-            self.gemini_model = genai.GenerativeModel("gemini-1.5-flash-latest")
-            return True, "✅ Gemini API configured!"
+            self.gemini_model = genai.GenerativeModel(
+                model_name="gemini-1.5-flash-latest"
+            )
+            return True, "✅ Gemini API configured successfully!"
         except Exception as e:
-            return False, f"❌ Gemini 설정 실패: {e}"
+            return False, f"❌ Failed to configure Gemini API: {e}"
 
-    # ───────── 쿠키 선택 + 디버그 ─────────
-    @staticmethod
-    def choose_cookie(ui_cookie: str | None):
-        if ui_cookie and os.path.exists(ui_cookie):
-            ck = ui_cookie
-        elif DEFAULT_COOKIE_FILE.exists():
-            ck = str(DEFAULT_COOKIE_FILE)
+    # ---------------------------------------------------------
+    # 임시 디렉터리 정리
+    # ---------------------------------------------------------
+    def cleanup(self):
+        try:
+            if hasattr(self, "download_dir") and os.path.exists(self.download_dir):
+                shutil.rmtree(self.download_dir)
+            if hasattr(self, "temp_downloads") and os.path.exists(self.temp_downloads):
+                shutil.rmtree(self.temp_downloads)
+        except Exception as e:
+            print(f"⚠️ Warning: Could not clean up temporary directory: {e}")
+
+    # ---------------------------------------------------------
+    # 유튜브 URL 검증
+    # ---------------------------------------------------------
+    def is_valid_youtube_url(self, url):
+        youtube_regex = re.compile(
+            r"(https?://)?(www\.)?"
+            r"(youtube|youtu|youtube-nocookie)\.(com|be)/"
+            r"(watch\?v=|embed/|v/|.+\?v=)?([^&=%\?]{11})"
+        )
+        return youtube_regex.match(url) is not None
+
+    # ---------------------------------------------------------
+    # Gemini-AI 장면 분석
+    # ---------------------------------------------------------
+    def generate_scene_breakdown_gemini(self, video_info):
+        if not self.gemini_model:
+            return self.generate_scene_breakdown_fallback(video_info)
+
+        try:
+            duration = video_info.get("duration", 0)
+            title = video_info.get("title", "")
+            description = video_info.get("description", "")[:1500]
+
+            if not duration:
+                return [
+                    "**[Duration Unknown]**: Unable to generate timestamped breakdown - "
+                    "video duration not available"
+                ]
+
+            prompt = f"""
+Analyze this YouTube video and create a highly detailed, scene-by-scene breakdown
+with precise timestamps and specific descriptions:
+
+Title: {title}
+Duration: {duration} seconds
+Description: {description}
+
+IMPORTANT INSTRUCTIONS:
+1. Create detailed scene descriptions that include:
+   - Physical appearance of people (age, gender, clothing, hair, etc.)
+   - Exact actions being performed
+   - Dialogue or speech (include actual lines if audible, or infer probable spoken
+     lines based on actions and setting; format them as "Character: line…")
+   - Setting and environment details
+   - Props, objects, or products being shown
+   - Visual effects, text overlays, or graphics
+   - Mood, tone, and atmosphere
+   - Camera movements or angles (if apparent)
+2. Dialogue Emphasis:
+   - Include short dialogue lines in **every scene** wherever plausible.
+   - Write lines like: Character: "Actual or inferred line…"
+   - If dialogue is not available, intelligently infer probable phrases
+     (e.g., "Welcome!", "Try this now!", "It feels amazing!").
+3. Timestamp Guidelines:
+   - For videos under 1 minute: 2-3 second segments
+   - For videos 1-5 minutes: 3-5 second segments
+   - For videos 5-15 minutes: 5-10 second segments
+   - For videos over 15 minutes: 10-15 second segments
+   - Maximum 20 scenes total for longer videos
+4. Format each scene EXACTLY like this:
+   **[MM:SS-MM:SS]**: Detailed description…
+5. Write descriptions as if you're watching the video in real-time,
+   noting everything visible and audible.
+"""
+            response = self.gemini_model.generate_content(prompt)
+
+            if response and response.text:
+                scenes = []
+                lines = response.text.split("\n")
+                current_scene = ""
+
+                for line in lines:
+                    line = line.strip()
+                    if line.startswith("**[") and "]**:" in line:
+                        if current_scene:
+                            scenes.append(current_scene.strip())
+                        current_scene = line
+                    elif current_scene:
+                        current_scene += "\n" + line
+
+                if current_scene:
+                    scenes.append(current_scene.strip())
+
+                return (
+                    scenes if scenes else self.generate_scene_breakdown_fallback(video_info)
+                )
+
+            return self.generate_scene_breakdown_fallback(video_info)
+
+        except Exception as e:
+            print(f"Gemini API error: {e}")
+            return self.generate_scene_breakdown_fallback(video_info)
+
+    # ---------------------------------------------------------
+    # Fallback 장면 분석
+    # ---------------------------------------------------------
+    def generate_scene_breakdown_fallback(self, video_info):
+        duration = video_info.get("duration", 0)
+        title = video_info.get("title", "").lower()
+        description = video_info.get("description", "").lower()
+        uploader = video_info.get("uploader", "Content creator")
+
+        if not duration:
+            return ["**[Duration Unknown]**: Unable to generate timestamped breakdown"]
+
+        if duration <= 60:
+            segment_length = 3
+        elif duration <= 300:
+            segment_length = 5
+        elif duration <= 900:
+            segment_length = 10
         else:
-            ck = None
-        print(">>> COOKIE =", ck, "EXISTS?", os.path.exists(ck) if ck else None)
-        return ck
-
-    # ───────── URL 검증 ─────────
-    @staticmethod
-    def valid_url(url: str) -> bool:
-        return bool(_YT_RE.match(url))
-
-    # ───────── 숫자 포맷 ─────────
-    @staticmethod
-    def fmt(n: int) -> str:
-        if n >= 1_000_000: return f"{n/1_000_000:.1f} M"
-        if n >= 1_000:     return f"{n/1_000:.1f} K"
-        return str(n)
-
-    # ───────── Scene 분석 (fallback만 간단 구현) ─────────
-    def scene_breakdown(self, info: dict):
-        dur = info.get("duration", 0)
-        seg = 3 if dur <= 60 else 5 if dur <= 300 else 10
+            segment_length = 15
+
         scenes = []
-        for s in range(0, dur, seg):
-            e = min(s + seg - 1, dur)
-            scenes.append(f"**[{s//60:02d}:{s%60:02d}-{e//60:02d}:{e%60:02d}]** …")
-        return "\n".join(scenes[:20])
-
-    # ───────── Video Info ─────────
-    def get_info(self, url: str, ui_cookie: str | None, progress):
-        if not self.valid_url(url):
-            return None, "❌ URL 오류"
-        ck = self.choose_cookie(ui_cookie)
-        if ck is None:
-            return None, "❌ 쿠키 파일을 찾을 수 없습니다"
-        ydl_opts = {"quiet": True, "noplaylist": True, "cookiefile": ck}
+        num_segments = min(duration // segment_length + 1, 20)
+        video_type = self.detect_video_type_detailed(title, description)
+
+        for i in range(num_segments):
+            start_time = i * segment_length
+            end_time = min(start_time + segment_length - 1, duration)
+
+            start_fmt = f"{start_time//60}:{start_time%60:02d}"
+            end_fmt = f"{end_time//60}:{end_time%60:02d}"
+            desc = self.generate_contextual_description(
+                i, num_segments, video_type, uploader, title
+            )
+            scenes.append(f"**[{start_fmt}-{end_fmt}]**: {desc}")
+
+        return scenes
+
+    # ---------------------------------------------------------
+    # 비디오 유형 감지(상세)
+    # ---------------------------------------------------------
+    def detect_video_type_detailed(self, title, description):
+        text = (title + " " + description).lower()
+        if any(w in text for w in ["tutorial", "how to", "guide", "learn", "diy"]):
+            return "tutorial"
+        if any(w in text for w in ["review", "unboxing", "test", "comparison", "vs"]):
+            return "review"
+        if any(w in text for w in ["vlog", "daily", "routine", "day in"]):
+            return "vlog"
+        if any(w in text for w in ["music", "song", "cover", "lyrics"]):
+            return "music"
+        if any(w in text for w in ["comedy", "funny", "prank", "challenge"]):
+            return "entertainment"
+        if any(w in text for w in ["news", "breaking", "update", "report"]):
+            return "news"
+        if any(w in text for w in ["cooking", "recipe", "food", "kitchen"]):
+            return "cooking"
+        if any(w in text for w in ["workout", "fitness", "exercise", "yoga"]):
+            return "fitness"
+        return "general"
+
+    # ---------------------------------------------------------
+    # 장면별 설명 생성
+    # ---------------------------------------------------------
+    def generate_contextual_description(
+        self, scene_index, total_scenes, video_type, uploader, title
+    ):
+        presenter_desc = "The content creator"
+        if "woman" in title or "girl" in title:
+            presenter_desc = "A woman"
+        elif "man" in title or "guy" in title:
+            presenter_desc = "A man"
+
+        if scene_index == 0:
+            if video_type == "tutorial":
+                return (
+                    f"{presenter_desc} appears on screen, introducing themselves and the "
+                    f"topic. They are in a well-lit workspace, wearing casual clothes."
+                )
+            if video_type == "vlog":
+                return (
+                    f"{presenter_desc} greets the camera cheerfully, perhaps waving, and "
+                    f"explains what today's vlog is about."
+                )
+            if video_type == "review":
+                return (
+                    f"{presenter_desc} holds up the product to be reviewed, giving a brief "
+                    f"overview of its features."
+                )
+            return (
+                f"{presenter_desc} starts the video with an engaging introduction to "
+                f"capture viewers' attention."
+            )
+
+        if scene_index == total_scenes - 1:
+            if video_type == "tutorial":
+                return (
+                    f"{presenter_desc} shows the final result, thanks viewers, and "
+                    f"encourages them to like and subscribe."
+                )
+            if video_type == "vlog":
+                return (
+                    f"{presenter_desc} wraps up the day, sharing final thoughts and "
+                    f"bidding farewell."
+                )
+            return (
+                f"{presenter_desc} concludes, summarizing key points and prompting "
+                f"engagement through likes and comments."
+            )
+
+        if video_type == "tutorial":
+            return (
+                f"{presenter_desc} demonstrates the next step, providing clear "
+                f"instructions with close-up shots."
+            )
+        if video_type == "review":
+            return (
+                f"{presenter_desc} examines a specific feature of the product, showing "
+                f"it in use and commenting on performance."
+            )
+        if video_type == "vlog":
+            return (
+                f"{presenter_desc} continues the day's activities, sharing candid "
+                f"moments and personal reflections."
+            )
+        if video_type == "cooking":
+            return (
+                f"{presenter_desc} prepares ingredients, chopping and mixing while "
+                f"explaining each step."
+            )
+        if video_type == "fitness":
+            return (
+                f"{presenter_desc} performs an exercise set, demonstrating proper form "
+                f"and offering tips."
+            )
+        return (
+            f"{presenter_desc} proceeds with the main content, engaging viewers through "
+            f"clear explanations."
+        )
+
+    # ---------------------------------------------------------
+    # 비디오 유형 (간략)
+    # ---------------------------------------------------------
+    def detect_video_type(self, title, description):
+        text = (title + " " + description).lower()
+        if any(w in text for w in ["music", "song", "album", "artist", "band", "lyrics"]):
+            return "🎵 Music Video"
+        if any(w in text for w in ["tutorial", "how to", "guide", "learn"]):
+            return "📚 Tutorial/Educational"
+        if any(w in text for w in ["funny", "comedy", "entertainment", "vlog"]):
+            return "🎭 Entertainment/Comedy"
+        if any(w in text for w in ["news", "breaking", "report", "update"]):
+            return "📰 News/Information"
+        if any(w in text for w in ["review", "unboxing", "test", "comparison"]):
+            return "⭐ Review/Unboxing"
+        if any(w in text for w in ["commercial", "ad", "brand", "product"]):
+            return "📺 Commercial/Advertisement"
+        return "🎬 General Content"
+
+    # ---------------------------------------------------------
+    # 배경 음악 추정
+    # ---------------------------------------------------------
+    def detect_background_music(self, video_info):
+        title = video_info.get("title", "").lower()
+        if "music" in title or "song" in title:
+            return "🎵 Original Music/Soundtrack"
+        if "commercial" in title or "ad" in title:
+            return "🎶 Upbeat Commercial Music"
+        if "tutorial" in title or "how to" in title:
+            return "🔇 Minimal/No Background Music"
+        if "vlog" in title or "daily" in title:
+            return "🎼 Ambient Background Music"
+        return "🎵 Background Music"
+
+    # ---------------------------------------------------------
+    # 인플루언서 규모 추정
+    # ---------------------------------------------------------
+    def detect_influencer_status(self, video_info):
+        subs = video_info.get("channel_followers", 0)
+        views = video_info.get("view_count", 0)
+        if subs > 10_000_000:
+            return "🌟 Mega Influencer (10M+)"
+        if subs > 1_000_000:
+            return "⭐ Major Influencer (1M+)"
+        if subs > 100_000:
+            return "🎯 Mid-tier Influencer (100K+)"
+        if subs > 10_000:
+            return "📈 Micro Influencer (10K+)"
+        if views > 100_000:
+            return "🔥 Viral Content Creator"
+        return "👤 Regular Content Creator"
+
+    # ---------------------------------------------------------
+    # 숫자 포맷터
+    # ---------------------------------------------------------
+    def format_number(self, num):
+        if not num:
+            return "0"
+        if num >= 1_000_000_000:
+            return f"{num/1_000_000_000:.1f}B"
+        if num >= 1_000_000:
+            return f"{num/1_000_000:.1f}M"
+        if num >= 1_000:
+            return f"{num/1_000:.1f}K"
+        return str(num)
+
+    # ---------------------------------------------------------
+    # 최종 리포트 생성
+    # ---------------------------------------------------------
+    def format_video_info(self, video_info):
+        if not video_info:
+            return "❌ No video information available."
+
+        title = video_info.get("title", "Unknown")
+        uploader = video_info.get("uploader", "Unknown")
+        duration = video_info.get("duration", 0)
+        dur_str = f"{duration//60}:{duration%60:02d}" if duration else "Unknown"
+        views = video_info.get("view_count", 0)
+        likes = video_info.get("like_count", 0)
+        comments = video_info.get("comment_count", 0)
+        upload_date = video_info.get("upload_date", "Unknown")
+
+        if len(upload_date) == 8:
+            upload_date = f"{upload_date[:4]}-{upload_date[4:6]}-{upload_date[6:8]}"
+
+        scenes = self.generate_scene_breakdown_gemini(video_info)
+        vtype = self.detect_video_type(title, video_info.get("description", ""))
+        bgm = self.detect_background_music(video_info)
+        creator = self.detect_influencer_status(video_info)
+        engagement = (likes / views) * 100 if views else 0
+
+        report = f"""
+🎬 YOUTUBE VIDEO ANALYSIS REPORT
+{'='*50}
+
+📋 BASIC INFORMATION
+{'─'*25}
+📹 **Title:** {title}
+👤 **Uploader:** {uploader}
+📅 **Upload Date:** {upload_date}
+⏱️ **Duration:** {dur_str}
+🆔 **Video ID:** {video_info.get('id', 'Unknown')}
+
+📊 PERFORMANCE METRICS
+{'─'*25}
+👀 **Views:** {self.format_number(views)} ({views:,})
+👍 **Likes:** {self.format_number(likes)} ({likes:,})
+💬 **Comments:** {self.format_number(comments)} ({comments:,})
+📈 **Engagement Rate:** {engagement:.2f}%
+
+🎯 CONTENT ANALYSIS
+{'─'*25}
+📂 **Video Type:** {vtype}
+🎵 **Background Music:** {bgm}
+👑 **Creator Status:** {creator}
+
+🎬 DETAILED SCENE BREAKDOWN
+{'─'*30}
+{chr(10).join(scenes)}
+
+📝 DESCRIPTION PREVIEW
+{'─'*25}
+{video_info.get('description', 'No description available')[:500]}
+{'...(truncated)' if len(video_info.get('description', '')) > 500 else ''}
+
+{'='*50}
+📊 **Analysis completed:** {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}
+🤖 **AI Enhancement:** {'Gemini AI' if self.gemini_model else 'Standard Analysis'}
+"""
+        return report.strip()
+
+    # ---------------------------------------------------------
+    # 메타데이터 추출
+    # ---------------------------------------------------------
+    def get_video_info(self, url, progress=gr.Progress(), cookiefile=None):
+        if not url or not url.strip():
+            return None, "❌ Please enter a YouTube URL"
+        if not self.is_valid_youtube_url(url):
+            return None, "❌ Invalid YouTube URL format"
+
+        # 쿠키 선택 순서: UI 업로드 → 기본 쿠키 → None
+        if cookiefile and os.path.exists(cookiefile):
+            cookiefile = cookiefile
+        elif DEFAULT_COOKIE_FILE.exists():
+            cookiefile = str(DEFAULT_COOKIE_FILE)
+        else:
+            cookiefile = None
+
         try:
-            progress(0.1, desc="metadata…")
+            progress(0.1, desc="Initializing YouTube extractor…")
+            ydl_opts = {"noplaylist": True, "extract_flat": False}
+            if cookiefile:
+                ydl_opts["cookiefile"] = cookiefile
+
+            progress(0.5, desc="Extracting video metadata…")
             with yt_dlp.YoutubeDL(ydl_opts) as ydl:
                 info = ydl.extract_info(url, download=False)
-            progress(1.0)
-            return info, "OK"
+
+            progress(1.0, desc="✅ Analysis complete!")
+            return info, "✅ Video information extracted successfully"
+
         except Exception as e:
-            return None, f"yt-dlp 오류: {e}"
+            return None, f"❌ Error: {e}"
 
-    # ───────── Download ─────────
-    def download(
-        self, url: str, qual: str, audio: bool, ui_cookie: str | None, progress
+    # ---------------------------------------------------------
+    # 다운로드
+    # ---------------------------------------------------------
+    def download_video(
+        self,
+        url,
+        quality="best",
+        audio_only=False,
+        progress=gr.Progress(),
+        cookiefile=None,
     ):
-        if not self.valid_url(url):
-            return None, "❌ URL 오류"
-        ck = self.choose_cookie(ui_cookie)
-        if ck is None:
-            return None, "❌ 쿠키 파일을 찾을 수 없습니다"
-        ts = datetime.now().strftime("%Y%m%d_%H%M%S")
-        ydl_opts: dict = {
-            "outtmpl": str(Path(self.temp_downloads) / f"%(title)s_{ts}.%(ext)s"),
-            "quiet": True,
-            "noplaylist": True,
-            "cookiefile": ck,
-        }
-        if audio:
-            ydl_opts["format"] = "bestaudio/best"
-            ydl_opts["postprocessors"] = [
-                {"key": "FFmpegExtractAudio", "preferredcodec": "mp3", "preferredquality": "192"}
-            ]
+        if not url or not url.strip():
+            return None, "❌ Please enter a YouTube URL"
+        if not self.is_valid_youtube_url(url):
+            return None, "❌ Invalid YouTube URL format"
+
+        # 쿠키 선택 순서
+        if cookiefile and os.path.exists(cookiefile):
+            cookiefile = cookiefile
+        elif DEFAULT_COOKIE_FILE.exists():
+            cookiefile = str(DEFAULT_COOKIE_FILE)
         else:
-            ydl_opts["format"] = (
-                "best[height<=720]" if qual == "720p"
-                else "best[height<=480]" if qual == "480p"
-                else "best[height<=1080]"
-            )
+            cookiefile = None
+
         try:
-            progress(0.1, desc="Downloading…")
+            progress(0.1, desc="Preparing download…")
+            timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+
+            ydl_opts = {
+                "outtmpl": os.path.join(
+                    self.temp_downloads, f"%(title)s_{timestamp}.%(ext)s"
+                ),
+                "noplaylist": True,
+            }
+
+            if audio_only:
+                ydl_opts["format"] = "bestaudio/best"
+                ydl_opts["postprocessors"] = [
+                    {
+                        "key": "FFmpegExtractAudio",
+                        "preferredcodec": "mp3",
+                        "preferredquality": "192",
+                    }
+                ]
+            else:
+                if quality == "720p":
+                    ydl_opts["format"] = "best[height<=720]"
+                elif quality == "480p":
+                    ydl_opts["format"] = "best[height<=480]"
+                else:  # "best"
+                    ydl_opts["format"] = "best[height<=1080]"
+
+            if cookiefile:
+                ydl_opts["cookiefile"] = cookiefile
+
+            # 진행률 훅
+            def hook(d):
+                if d["status"] == "downloading":
+                    if "total_bytes" in d:
+                        pct = d["downloaded_bytes"] / d["total_bytes"] * 100
+                        progress(0.1 + pct / 100 * 0.7, desc=f"Downloading… {pct:.1f}%")
+                    else:
+                        progress(0.5, desc="Downloading…")
+                elif d["status"] == "finished":
+                    progress(0.8, desc="Processing download…")
+
+            ydl_opts["progress_hooks"] = [hook]
+
             with yt_dlp.YoutubeDL(ydl_opts) as ydl:
                 ydl.extract_info(url, download=True)
-            # temp → Downloads
-            for f in Path(self.temp_downloads).iterdir():
-                if ts in f.name:
-                    dest = self.dl_folder / f.name
-                    try:
-                        shutil.copy2(f, dest)
-                        saved = dest
-                    except Exception:
-                        saved = f
-                    progress(1.0)
-                    return saved, f"✅ 저장 위치: {saved}"
-            return None, "❌ 파일을 찾을 수 없습니다"
+
+            progress(0.9, desc="Copying to Downloads folder…")
+
+            # temp 디렉터리에서 파일 찾기
+            downloaded_temp = None
+            for f in os.listdir(self.temp_downloads):
+                if timestamp in f:
+                    downloaded_temp = os.path.join(self.temp_downloads, f)
+                    break
+
+            if not downloaded_temp:
+                return None, "❌ Downloaded file not found"
+
+            final_name = os.path.basename(downloaded_temp)
+            final_path = os.path.join(self.downloads_folder, final_name)
+
+            try:
+                shutil.copy2(downloaded_temp, final_path)
+                saved = True
+            except Exception as e:
+                print(f"Copy warning: {e}")
+                saved = False
+                final_path = "File kept only in temp folder"
+
+            progress(1.0, desc="✅ Download complete!")
+
+            msg = (
+                "✅ Download successful!\n"
+                f"📁 Temp file: {os.path.basename(downloaded_temp)}\n"
+                f"📁 Saved to: {final_path if saved else 'Copy failed'}\n"
+                f"🎯 Size: {os.path.getsize(downloaded_temp)/(1024*1024):.1f} MB"
+            )
+            return downloaded_temp, msg
+
         except Exception as e:
-            return None, f"❌ 다운로드 실패: {e}"
+            return None, f"❌ Download failed: {e}"
+
+
+# =================================================================
+# Helper functions for Gradio
+# =================================================================
+downloader = YouTubeDownloader()
 
-# ───────────────────────── 인스턴스 ─────────────────────────
-yt_dl = YouTubeDownloader()
 
-# ───────────────────────── Gradio 함수 ──────────────────────
-def api_config(key):
-    ok, msg = yt_dl.configure_gemini(key.strip()) if key else (False, "❌ API 키 필요")
+def configure_api_key(api_key):
+    if not api_key or not api_key.strip():
+        return "❌ Please enter a valid Google API key", gr.update(visible=False)
+    ok, msg = downloader.configure_gemini(api_key.strip())
     return msg, gr.update(visible=ok)
 
-def analyze(url, cfile, progress=gr.Progress()):
-    info, err = yt_dl.get_info(url, cfile, progress)
-    if not info:
-        return err
-    report = (
-        f"**Title** : {info.get('title')}\n"
-        f"**Duration** : {info.get('duration',0)//60}:{info.get('duration',0)%60:02d}\n"
-        f"**Views / Likes** : {yt_dl.fmt(info.get('view_count',0))} / {yt_dl.fmt(info.get('like_count',0))}\n"
-        f"{'-'*40}\n"
-        f"{yt_dl.scene_breakdown(info)}"
-    )
-    return report
-
-def download(url, q, a, cfile, progress=gr.Progress()):
-    path, msg = yt_dl.download(url, q, a, cfile, progress)
-    return msg, path
-
-def transcript(url, _cfile):
-    vid = extract_video_id(url)
-    if not vid:
-        return "❌ URL 오류"
+
+def analyze_with_cookies(url, cookies_file, progress=gr.Progress()):
     try:
-        return fetch_transcript(vid)
+        progress(0.05, desc="Starting analysis…")
+        cookiefile = cookies_file if cookies_file else None
+        info, msg = downloader.get_video_info(
+            url, progress=progress, cookiefile=cookiefile
+        )
+        if info:
+            progress(0.95, desc="Generating report…")
+            return downloader.format_video_info(info)
+        return f"❌ Analysis Failed: {msg}"
+    except Exception as e:
+        return f"❌ System Error: {e}"
+
+
+def download_with_cookies(url, quality, audio_only, cookies_file, progress=gr.Progress()):
+    try:
+        progress(0.05, desc="Preparing download…")
+        cookiefile = cookies_file if cookies_file else None
+        file_path, status = downloader.download_video(
+            url, quality, audio_only, progress=progress, cookiefile=cookiefile
+        )
+        return (file_path, status) if file_path else (None, status)
     except Exception as e:
-        return f"❌ 자막 오류: {e}"
+        return None, f"❌ System Error: {e}"
 
-# ───────────────────────── UI ───────────────────────────────
-def create_ui():
-    with gr.Blocks(theme=gr.themes.Soft(), title="🎥 YouTube Video Analyzer & Downloader") as ui:
-        gr.HTML("<h1>🎥 YouTube Video Analyzer & Downloader</h1>")
 
-        # API
+# =================================================================
+# Gradio UI
+# =================================================================
+def create_interface():
+    with gr.Blocks(
+        theme=gr.themes.Soft(), title="🎥 YouTube Video Analyzer & Downloader Pro"
+    ) as iface:
+        gr.HTML("<h1>🎥 YouTube Video Analyzer & Downloader Pro</h1>")
+
+        # API 섹션
         with gr.Group():
+            gr.HTML("<h3>🔑 Google Gemini API Configuration</h3>")
             with gr.Row():
-                api_key = gr.Textbox(type="password", label="Gemini API Key")
-                api_btn = gr.Button("Configure")
-            api_stat = gr.Textbox(label="API Status", value="❌ Gemini 미설정", interactive=False)
-        api_btn.click(api_config, [api_key], [api_stat])
+                api_key_in = gr.Textbox(
+                    label="🔑 Google API Key",
+                    placeholder="Paste your Google API key…",
+                    type="password",
+                )
+                api_btn = gr.Button("🔧 Configure API", variant="secondary")
+            api_status = gr.Textbox(
+                label="API Status",
+                value="❌ Gemini API not configured – Using fallback analysis",
+                interactive=False,
+                lines=1,
+            )
 
-        # 입력
+        # 메인 UI
         with gr.Row():
-            url = gr.Textbox(label="🔗 YouTube URL")
-            cookie = gr.File(label="🍪 cookies.txt (선택)", type="filepath", file_types=[".txt"])
+            url_in = gr.Textbox(
+                label="🔗 YouTube URL",
+                placeholder="Paste YouTube video URL…",
+            )
+            cookies_in = gr.File(
+                label="🍪 Upload cookies.txt (optional)",
+                file_types=[".txt"],
+                type="filepath",
+            )
 
         with gr.Tabs():
+            with gr.TabItem("📊 Video Analysis"):
+                analyze_btn = gr.Button("🔍 Analyze Video", variant="primary")
+                analysis_out = gr.Textbox(
+                    label="📊 Analysis Report", lines=25, show_copy_button=True
+                )
+                analyze_btn.click(
+                    fn=analyze_with_cookies,
+                    inputs=[url_in, cookies_in],
+                    outputs=analysis_out,
+                    show_progress=True,
+                )
 
-            with gr.TabItem("📊 Analyze"):
-                a_btn = gr.Button("Analyze", variant="primary")
-                a_out = gr.Textbox(lines=20, label="Analysis", show_copy_button=True)
-                a_btn.click(analyze, [url, cookie], a_out, show_progress=True)
-
-            with gr.TabItem("⬇️ Download"):
+            with gr.TabItem("⬇️ Video Download"):
                 with gr.Row():
-                    q_dd = gr.Dropdown(["best", "720p", "480p"], value="best", label="Quality")
-                    a_cb = gr.Checkbox(label="Audio only (MP3)")
-                d_btn = gr.Button("Download", variant="primary")
-                d_stat = gr.Textbox(lines=5, label="Status")
-                d_file = gr.File(label="File", visible=False)
-
-                def wrap_dl(u,q,a,c,pg=gr.Progress()):
-                    msg, fp = download(u,q,a,c,pg)
-                    return msg, gr.update(value=fp, visible=bool(fp and os.path.exists(fp)))
-                d_btn.click(wrap_dl, [url,q_dd,a_cb,cookie], [d_stat,d_file], show_progress=True)
-
-            with gr.TabItem("🗒️ Transcript"):
-                t_btn = gr.Button("Get Transcript", variant="primary")
-                t_out = gr.Textbox(lines=30, label="Transcript", show_copy_button=True)
-                t_btn.click(transcript, [url, cookie], t_out, show_progress=True)
+                    quality_dd = gr.Dropdown(
+                        choices=["best", "720p", "480p"],
+                        value="best",
+                        label="📺 Quality",
+                    )
+                    audio_cb = gr.Checkbox(label="🎵 Audio only (MP3)")
+                download_btn = gr.Button("⬇️ Download Video", variant="primary")
+                dl_status = gr.Textbox(
+                    label="📥 Download Status", lines=5, show_copy_button=True
+                )
+                dl_file = gr.File(label="📁 Downloaded File", visible=False)
+
+                def wrapped_download(url, q, a, cfile, progress=gr.Progress()):
+                    fp, st = download_with_cookies(url, q, a, cfile, progress)
+                    if fp and os.path.exists(fp):
+                        return st, gr.update(value=fp, visible=True)
+                    return st, gr.update(visible=False)
+
+                download_btn.click(
+                    fn=wrapped_download,
+                    inputs=[url_in, quality_dd, audio_cb, cookies_in],
+                    outputs=[dl_status, dl_file],
+                    show_progress=True,
+                )
+
+        # API 버튼 동작
+        api_btn.click(
+            fn=configure_api_key,
+            inputs=[api_key_in],
+            outputs=[api_status],
+        )
 
         gr.HTML(
-            f"""
+            """
 <div style="margin-top:20px;padding:15px;background:#f0f8ff;border-left:5px solid #4285f4;border-radius:10px;">
-  <b>Tip</b> : <code>{DEFAULT_COOKIE_FILE.name}</code> 파일을 <b>app.py</b> 옆에 두면
-  자동으로 쿠키가 적용됩니다.
+  <h3>💡 Tip: 쿠키 파일 자동 사용</h3>
+  <p><code>www.youtube.com_cookies.txt</code> 파일을 <strong>app.py</strong>와 같은
+  폴더에 두면 자동으로 사용됩니다. 주기적으로 새 파일로 교체해 주세요.</p>
 </div>
 """
         )
-    return ui
+    return iface
 
-# ───────────────────────── Entrypoint ───────────────────────
+
+# =================================================================
+# Entrypoint
+# =================================================================
 if __name__ == "__main__":
-    demo = create_ui()
+    demo = create_interface()
     import atexit
 
-    atexit.register(lambda: shutil.rmtree(yt_dl.temp_dir, ignore_errors=True))
-    atexit.register(lambda: shutil.rmtree(yt_dl.temp_downloads, ignore_errors=True))
+    atexit.register(downloader.cleanup)
     demo.launch(debug=True, show_error=True)