Update app.py

app.py

@@ -1,52 +1,40 @@
 import gradio as gr
-from pytube import YouTube
-import os
+from yt_dlp import YoutubeDL
 
-def download_youtube_videos(urls):
-    """Downloads multiple YouTube videos and returns a list of downloaded file paths."""
-    downloaded_files = []
-    for url in urls:
-        try:
-            yt = YouTube(url)
-            video = yt.streams.filter(progressive=True, file_extension='mp4').order_by('resolution').desc().first()
-            filename = f"{yt.title}.mp4"
-            video.download(filename=filename)
-            downloaded_files.append(filename)
-        except Exception as e:
-            print(f"Error downloading {url}: {str(e)}")
-    
-    return downloaded_files
+def extract_metadata(youtube_url):
+    ydl_opts = {'quiet': True}
+    try:
+        with 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('like_count', 'N/A'),
+                "길이 (초)": info.get('duration', 'N/A'),
+                "설명": info.get('description', 'N/A')[:500] + '...'  # 설명 일부만 표시
+            }
+            
+            return "\n".join([f"{k}: {v}" for k, v in metadata.items()])
+    except Exception as e:
+        return f"오류 발생: {str(e)}"
+
+def check_yt_dlp_version():
+    return f"yt-dlp 버전: {YoutubeDL.version.__version__}"
 
-def app(urls):
-    # Split the input string into a list of URLs
-    url_list = [url.strip() for url in urls.split('\n') if url.strip()]
+with gr.Blocks() as demo:
+    gr.Markdown("## YouTube 메타데이터 추출기")
+    gr.Markdown(check_yt_dlp_version())
     
-    if not url_list:
-        return None, "No valid URLs provided. Please enter at least one YouTube URL."
+    with gr.Row():
+        youtube_url_input = gr.Textbox(label="YouTube URL 입력")
+        extract_button = gr.Button("메타데이터 추출")
     
-    downloaded_files = download_youtube_videos(url_list)
+    output = gr.Textbox(label="추출된 메타데이터", lines=10)
     
-    if downloaded_files:
-        # Create a list of (filename, file path) tuples for Gradio File component
-        file_tuples = [(f, f) for f in downloaded_files if os.path.exists(f)]
-        message = f"Successfully downloaded {len(file_tuples)} videos."
-        return file_tuples, message
-    else:
-        return None, "Download failed. Please check the URLs and try again."
-
-with gr.Blocks() as interface:
-    gr.Markdown("## YouTube Batch Video Downloader")
-    urls_input = gr.Textbox(label="Enter YouTube links (one per line) 🔗", lines=5)
-    download_button = gr.Button("Download Videos")
-    output = gr.File(label="Downloaded Videos", file_count="multiple")
-    message = gr.Markdown()
-
-    download_button.click(
-        fn=app,
-        inputs=urls_input,
-        outputs=[output, message],
-        api_name="download"
-    )
+    extract_button.click(fn=extract_metadata, inputs=youtube_url_input, outputs=output)
 
 if __name__ == "__main__":
-    interface.launch(debug=True)
+    demo.launch()