Update app.py

app.py

@@ -32,21 +32,15 @@ def extract_info(youtube_url):
             
             metadata_str = "\n".join([f"{k}: {v}" for k, v in metadata.items()])
             
-            video_only_url = best_video_only['url'] if best_video_only else None
-            combined_url = best_combined['url'] if best_combined else None
+            video_only_link = f'<a href="{best_video_only["url"]}" target="_blank">비디오만 다운로드</a>' if best_video_only else ""
+            combined_link = f'<a href="{best_combined["url"]}" target="_blank">비디오+오디오 다운로드</a>' if best_combined else ""
             
-            return (metadata_str, 
-                    video_only_url, combined_url, 
-                    gr.update(visible=bool(video_only_url)), 
-                    gr.update(visible=bool(combined_url)))
-    except Exception as e:
-        return (f"오류 발생: {str(e)}", 
-                None, None, 
-                gr.update(visible=False), 
-                gr.update(visible=False))
+            download_links = f"{video_only_link}<br>{combined_link}" if video_only_link or combined_link else "다운로드 링크를 찾을 수 없습니다."
+            
+            return metadata_str, download_links
 
-def on_download_click(url):
-    return f'<script>window.location.href = "{url}";</script>'
+    except Exception as e:
+        return f"오류 발생: {str(e)}", "다운로드 링크를 찾을 수 없습니다."
 
 with gr.Blocks() as demo:
     gr.Markdown("## YouTube 메타데이터 및 다운로드 링크 추출기")
@@ -55,33 +49,12 @@ with gr.Blocks() as demo:
     youtube_url_input = gr.Textbox(label="YouTube URL 입력")
     extract_button = gr.Button("정보 추출")
     output = gr.Textbox(label="추출된 정보", lines=10)
-    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)
+    download_links = gr.HTML(label="다운로드 링크")
     
     extract_button.click(
         fn=extract_info, 
         inputs=youtube_url_input, 
-        outputs=[output, video_only_button, combined_button, video_only_url, combined_url]
-    )
-    video_only_button.click(
-        fn=on_download_click,
-        inputs=video_only_url,
-        outputs=gr.HTML()
-    )
-    combined_button.click(
-        fn=on_download_click,
-        inputs=combined_url,
-        outputs=gr.HTML()
+        outputs=[output, download_links]
     )
 
 if __name__ == "__main__":