Update app.py

app.py

@@ -1,45 +1,11 @@
 import gradio as gr
-import yt_dlp
+from youtube_extractor import YouTubeExtractor
 
-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'),
-            }
-            
-            formats = info.get('formats', [])
-            
-            best_video_only = next((f for f in formats if f['ext'] == 'mp4' and f.get('vcodec') != 'none' and f.get('acodec') == 'none'), None)
-            
-            # 간단히 비디오와 오디오가 모두 있는 첫 번째 mp4 포맷을 선택
-            combined = next((f for f in formats if f['ext'] == 'mp4' and f.get('vcodec') != 'none' and f.get('acodec') != 'none'), None)
-            
-            metadata_str = "\n".join([f"{k}: {v}" for k, v in metadata.items()])
-            
-            video_only_link = f'<a href="{best_video_only["url"]}" target="_blank">비디오만 다운로드</a>' if best_video_only else ""
-            combined_link = f'<a href="{combined["url"]}" target="_blank">비디오+오디오 다운로드</a>' if combined else ""
-            
-            download_links = f"{video_only_link}<br>{combined_link}" if video_only_link or combined_link else "다운로드 링크를 찾을 수 없습니다."
-            
-            return metadata_str, download_links
+extractor = YouTubeExtractor()
 
-    except Exception as e:
-        return f"오류 발생: {str(e)}", "다운로드 링크를 찾을 수 없습니다."
+def process_youtube_url(youtube_url):
+    metadata, best_video_only, combined = extractor.extract_info(youtube_url)
+    return extractor.format_output(metadata, best_video_only, combined)
 
 with gr.Blocks() as demo:
     gr.Markdown("## YouTube 메타데이터 및 다운로드 링크 추출기")
@@ -51,8 +17,8 @@ with gr.Blocks() as demo:
     download_links = gr.HTML(label="다운로드 링크")
     
     extract_button.click(
-        fn=extract_info, 
-        inputs=youtube_url_input, 
+        fn=process_youtube_url,
+        inputs=youtube_url_input,
         outputs=[output, download_links]
     )