Update app.py

app.py

@@ -1,49 +1,64 @@
 import gradio as gr
 import yt_dlp
+import os
+import tempfile
+import base64
+
+def download_and_merge(youtube_url):
+    with tempfile.TemporaryDirectory() as temp_dir:
+        ydl_opts = {
+            'format': 'bestvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]/best',
+            'outtmpl': os.path.join(temp_dir, '%(title)s.%(ext)s'),
+            'merge_output_format': 'mp4',
+        }
+        try:
+            with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+                info = ydl.extract_info(youtube_url, download=True)
+                filename = ydl.prepare_filename(info)
+                final_filename = os.path.splitext(filename)[0] + '.mp4'
+                
+                if os.path.exists(final_filename):
+                    with open(final_filename, 'rb') as f:
+                        video_data = f.read()
+                    b64 = base64.b64encode(video_data).decode()
+                    return info['title'], b64
+                else:
+                    return info['title'], None
+        except Exception as e:
+            return str(e), None
 
 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)}"
+    title, video_data = download_and_merge(youtube_url)
+    if video_data:
+        download_link = f'data:video/mp4;base64,{video_data}'
+        return (f"제목: {title}\n\n다운로드가 준비되었습니다. '다운로드' 버튼을 클릭하세요.", 
+                download_link, 
+                gr.update(visible=True))
+    else:
+        return f"오류 발생: {title}", None, gr.update(visible=False)
 
 with gr.Blocks() as demo:
-    gr.Markdown("## YouTube 메타데이터 및 다운로드 링크 추출기")
+    gr.Markdown("## YouTube 다운로더")
     gr.Markdown("주의: 이 도구를 사용하여 저작권이 있는 콘텐츠를 무단으로 다운로드하는 것은 불법입니다.")
     
     youtube_url_input = gr.Textbox(label="YouTube URL 입력")
-    extract_button = gr.Button("정보 추출")
-    output = gr.Textbox(label="추출된 정보", lines=10)
+    extract_button = gr.Button("다운로드 준비")
+    output = gr.Textbox(label="상태", lines=5)
+    download_button = gr.Button("다운로드", visible=False)
+    
+    def on_download_click(url):
+        return f'<script>var link = document.createElement("a"); link.href = "{url}"; link.download = "video.mp4"; document.body.appendChild(link); link.click(); document.body.removeChild(link);</script>'
     
-    extract_button.click(fn=extract_info, inputs=youtube_url_input, outputs=output)
+    extract_button.click(
+        fn=extract_info, 
+        inputs=youtube_url_input, 
+        outputs=[output, download_button, download_button]
+    )
+    download_button.click(
+        fn=on_download_click,
+        inputs=download_button,
+        outputs=gr.HTML()
+    )
 
 if __name__ == "__main__":
     demo.launch()