Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import yt_dlp
 
-def extract_metadata(youtube_url):
+def extract_info(youtube_url):
     ydl_opts = {'quiet': True}
     try:
         with yt_dlp.YoutubeDL(ydl_opts) as ydl:
@@ -17,24 +17,26 @@ def extract_metadata(youtube_url):
                 "설명": info.get('description', 'N/A')[:500] + '...'  # 설명 일부만 표시
             }
             
+            # 최고 품질의 mp4 형식 찾기
+            best_mp4 = next((f for f in info['formats'] if f['ext'] == 'mp4' and f.get('quality') == 1), 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)}"
 
-def check_yt_dlp_version():
-    return f"yt-dlp 버전: {yt_dlp.version.__version__}"
-
 with gr.Blocks() as demo:
-    gr.Markdown("## YouTube 메타데이터 추출기")
-    gr.Markdown(check_yt_dlp_version())
-    
-    with gr.Row():
-        youtube_url_input = gr.Textbox(label="YouTube URL 입력")
-        extract_button = gr.Button("메타데이터 추출")
+    gr.Markdown("## YouTube 메타데이터 및 다운로드 링크 추출기")
+    gr.Markdown("주의: 이 도구를 사용하여 저작권이 있는 콘텐츠를 무단으로 다운로드하는 것은 불법입니다.")
     
-    output = gr.Textbox(label="추출된 메타데이터", lines=10)
+    youtube_url_input = gr.Textbox(label="YouTube URL 입력")
+    extract_button = gr.Button("정보 추출")
+    output = gr.Textbox(label="추출된 정보", lines=10)
     
-    extract_button.click(fn=extract_metadata, inputs=youtube_url_input, outputs=output)
+    extract_button.click(fn=extract_info, inputs=youtube_url_input, outputs=output)
 
 if __name__ == "__main__":
     demo.launch()