Update app.py

app.py

@@ -1,5 +1,13 @@
 import gradio as gr
 import yt_dlp
+import base64
+import requests
+
+def download_and_encode(url):
+    response = requests.get(url)
+    if response.status_code == 200:
+        return base64.b64encode(response.content).decode('utf-8')
+    return None
 
 def extract_info(youtube_url):
     ydl_opts = {
@@ -21,24 +29,25 @@ def extract_info(youtube_url):
             
             formats = info.get('formats', [])
             
-            # 최고 품질의 비디오 (오디오 없음)
             best_video = max((f for f in formats if f['vcodec'] != 'none' and f['acodec'] == 'none'), 
                               key=lambda x: x.get('height', 0), default=None)
             
-            # 최고 품질의 오디오
             best_audio = max((f for f in formats if f['acodec'] != 'none' and f['vcodec'] == 'none'), 
                               key=lambda x: x.get('abr', 0), default=None)
             
-            # 비디오+오디오 통합 스트림 (일반적으로 더 낮은 품질)
             best_combined = max((f for f in formats if f['vcodec'] != 'none' and f['acodec'] != 'none'), 
                                  key=lambda x: x.get('height', 0), default=None)
 
             metadata_str = "\n".join([f"{k}: {v}" for k, v in metadata.items()])
             
             if best_video and best_audio:
+                video_b64 = download_and_encode(best_video['url'])
+                audio_b64 = download_and_encode(best_audio['url'])
+                combined_b64 = download_and_encode(best_combined['url']) if best_combined else None
+                
                 return (metadata_str, 
-                        best_video['url'], best_audio['url'], best_combined['url'] if best_combined else None,
-                        gr.update(visible=True), gr.update(visible=True), gr.update(visible=True if best_combined else False))
+                        video_b64, audio_b64, combined_b64,
+                        gr.update(visible=True), gr.update(visible=True), gr.update(visible=True if combined_b64 else False))
             else:
                 return ("적절한 형식을 찾을 수 없습니다.", None, None, None,
                         gr.update(visible=False), gr.update(visible=False), gr.update(visible=False))
@@ -47,7 +56,7 @@ def extract_info(youtube_url):
                 gr.update(visible=False), gr.update(visible=False), gr.update(visible=False))
 
 with gr.Blocks() as demo:
-    gr.Markdown("## YouTube 메타데이터 및 다운로드 URL 추출기")
+    gr.Markdown("## YouTube 메타데이터 및 다운로드")
     gr.Markdown("주의: 이 도구를 사용하여 저작권이 있는 콘텐츠를 무단으로 다운로드하는 것은 불법입니다.")
     
     youtube_url_input = gr.Textbox(label="YouTube URL 입력")
@@ -57,17 +66,30 @@ with gr.Blocks() as demo:
     audio_button = gr.Button("오디오 다운로드", visible=False)
     combined_button = gr.Button("비디오+오디오 다운로드", visible=False)
     
-    def on_download_click(url):
-        return f'<script>window.location.href = "{url}";</script>'
+    def on_download_click(b64_data, file_type):
+        return f'''
+        <script>
+        var binary = atob('{b64_data}');
+        var array = new Uint8Array(binary.length);
+        for (var i = 0; i < binary.length; i++) {{
+            array[i] = binary.charCodeAt(i);
+        }}
+        var blob = new Blob([array], {{type: 'video/mp4'}});
+        var link = document.createElement('a');
+        link.href = window.URL.createObjectURL(blob);
+        link.download = "download.{file_type}";
+        link.click();
+        </script>
+        '''
     
     extract_button.click(
         fn=extract_info, 
         inputs=youtube_url_input, 
         outputs=[output, video_button, audio_button, combined_button, video_button, audio_button, combined_button]
     )
-    video_button.click(fn=on_download_click, inputs=video_button, outputs=gr.HTML())
-    audio_button.click(fn=on_download_click, inputs=audio_button, outputs=gr.HTML())
-    combined_button.click(fn=on_download_click, inputs=combined_button, outputs=gr.HTML())
+    video_button.click(fn=lambda x: on_download_click(x, 'mp4'), inputs=video_button, outputs=gr.HTML())
+    audio_button.click(fn=lambda x: on_download_click(x, 'm4a'), inputs=audio_button, outputs=gr.HTML())
+    combined_button.click(fn=lambda x: on_download_click(x, 'mp4'), inputs=combined_button, outputs=gr.HTML())
 
 if __name__ == "__main__":
     demo.launch()