Update app.py

app.py

@@ -24,11 +24,9 @@ def extract_info(youtube_url):
             
             formats = info.get('formats', [])
             
-            # 최고 품질의 mp4 (비디오만)
             best_video_only = max((f for f in formats if f['ext'] == 'mp4' and f.get('vcodec') != 'none' and f.get('acodec') == 'none'), 
                                    key=lambda x: x.get('height', 0), default=None)
             
-            # 최고 품질의 통합 스트림 (비디오+오디오)
             best_combined = max((f for f in formats if f['ext'] == 'mp4' and f.get('vcodec') != 'none' and f.get('acodec') != 'none'), 
                                  key=lambda x: x.get('height', 0), default=None)
             
@@ -60,19 +58,29 @@ with gr.Blocks() as demo:
     video_only_button = gr.Button("비디오만 다운로드", visible=False)
     combined_button = gr.Button("비디오+오디오 다운로드", visible=False)
     
+    video_only_url = gr.State()
+    combined_url = gr.State()
+    
+    def update_urls(metadata, video_url, combined_url, video_visibility, combined_visibility):
+        return (metadata, 
+                gr.update(visible=video_visibility), 
+                gr.update(visible=combined_visibility), 
+                video_url, 
+                combined_url)
+    
     extract_button.click(
         fn=extract_info, 
         inputs=youtube_url_input, 
-        outputs=[output, video_only_button, combined_button, video_only_button, combined_button]
+        outputs=[output, video_only_button, combined_button, video_only_url, combined_url]
     )
     video_only_button.click(
         fn=on_download_click,
-        inputs=video_only_button,
+        inputs=video_only_url,
         outputs=gr.HTML()
     )
     combined_button.click(
         fn=on_download_click,
-        inputs=combined_button,
+        inputs=combined_url,
         outputs=gr.HTML()
     )