Update app.py

app.py

@@ -1,19 +1,14 @@
 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 = {
         '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:
@@ -27,69 +22,41 @@ def extract_info(youtube_url):
                 "길이 (초)": 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)
             
-            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, 
-                        video_b64, audio_b64, combined_b64,
-                        gr.update(visible=True), gr.update(visible=True), gr.update(visible=True if combined_b64 else False))
+            if best_mp4:
+                download_url = best_mp4['url']
+                metadata_str = "\n".join([f"{k}: {v}" for k, v in metadata.items()])
+                return metadata_str, download_url, gr.update(visible=True)
             else:
-                return ("적절한 형식을 찾을 수 없습니다.", None, None, None,
-                        gr.update(visible=False), gr.update(visible=False), gr.update(visible=False))
+                return "적절한 형식을 찾을 수 없습니다.", None, gr.update(visible=False)
     except Exception as e:
-        return (f"오류 발생: {str(e)}", None, None, None,
-                gr.update(visible=False), gr.update(visible=False), gr.update(visible=False))
+        return f"오류 발생: {str(e)}", None, gr.update(visible=False)
+
+def on_download_click(url):
+    return f'<script>window.location.href = "{url}";</script>'
 
 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)
-    video_button = gr.Button("비디오 다운로드 (오디오 없음)", visible=False)
-    audio_button = gr.Button("오디오 다운로드", visible=False)
-    combined_button = gr.Button("비디오+오디오 다운로드", visible=False)
-    
-    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>
-        '''
+    download_button = gr.Button("다운로드", visible=False)
     
     extract_button.click(
         fn=extract_info, 
         inputs=youtube_url_input, 
-        outputs=[output, video_button, audio_button, combined_button, video_button, audio_button, combined_button]
+        outputs=[output, download_button, download_button]
+    )
+    download_button.click(
+        fn=on_download_click,
+        inputs=download_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()