JUNGU's picture
Update app.py
6b782ad verified
raw
history blame
1.6 kB
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()