Update app.py

app.py

@@ -22,18 +22,30 @@ 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)
             
-            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, gr.update(visible=False)
+            # 최고 품질의 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)
+            
+            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
+            
+            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, gr.update(visible=False)
+        return (f"오류 발생: {str(e)}", 
+                None, None, 
+                gr.update(visible=False), 
+                gr.update(visible=False))
 
 def on_download_click(url):
     return f'<script>window.location.href = "{url}";</script>'
@@ -45,16 +57,22 @@ with gr.Blocks() as demo:
     youtube_url_input = gr.Textbox(label="YouTube URL 입력")
     extract_button = gr.Button("정보 추출")
     output = gr.Textbox(label="추출된 정보", lines=10)
-    download_button = gr.Button("다운로드", visible=False)
+    video_only_button = gr.Button("비디오만 다운로드", visible=False)
+    combined_button = gr.Button("비디오+오디오 다운로드", visible=False)
     
     extract_button.click(
         fn=extract_info, 
         inputs=youtube_url_input, 
-        outputs=[output, download_button, download_button]
+        outputs=[output, video_only_button, combined_button, video_only_button, combined_button]
+    )
+    video_only_button.click(
+        fn=on_download_click,
+        inputs=video_only_button,
+        outputs=gr.HTML()
     )
-    download_button.click(
+    combined_button.click(
         fn=on_download_click,
-        inputs=download_button,
+        inputs=combined_button,
         outputs=gr.HTML()
     )