import gradio as gr
import yt_dlp

def extract_info(youtube_url):
    ydl_opts = {
        'quiet': True,
        'no_warnings': True,
        'no_color': True,
        'youtube_include_dash_manifest': False,
        'youtube_include_hls_manifest': False,
        'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36',
    }
    try:
        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
            info = ydl.extract_info(youtube_url, download=False)
            
            metadata = {
                "제목": info.get('title', 'N/A'),
                "채널": info.get('channel', 'N/A'),
                "업로드 날짜": info.get('upload_date', 'N/A'),
                "조회수": info.get('view_count', 'N/A'),
                "길이 (초)": info.get('duration', 'N/A'),
            }
            
            # 사용 가능한 모든 포맷 중 가장 높은 품질의 mp4 찾기
            formats = info.get('formats', [])
            best_mp4 = max((f for f in formats if f['ext'] == 'mp4'), key=lambda x: x.get('quality', 0), default=None)
            
            if best_mp4:
                metadata["다운로드 URL"] = best_mp4['url']
            else:
                metadata["다운로드 URL"] = "적절한 형식을 찾을 수 없습니다."
            
            return "\n".join([f"{k}: {v}" for k, v in metadata.items()])
    except Exception as e:
        return f"오류 발생: {str(e)}"

# Gradio 인터페이스 코드는 이전과 동일

if __name__ == "__main__":
    demo.launch()