Update youtube_extractor.py

youtube_extractor.py

@@ -6,9 +6,6 @@ class YouTubeExtractor:
             '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',
         }
 
     def extract_info(self, youtube_url):
@@ -26,23 +23,42 @@ class YouTubeExtractor:
                 
                 formats = info.get('formats', [])
                 
-                best_video_only = next((f for f in formats if f['ext'] == 'mp4' and f.get('vcodec') != 'none' and f.get('acodec') == 'none'), None)
-                combined = next((f for f in formats if f['ext'] == 'mp4' and f.get('vcodec') != 'none' and f.get('acodec') != 'none'), None)
+                # 비디오 전용 (가장 높은 해상도)
+                best_video_only = max(
+                    (f for f in formats if f.get('vcodec') != 'none' and f.get('acodec') == 'none'),
+                    key=lambda x: x.get('height', 0),
+                    default=None
+                )
                 
-                return metadata, best_video_only, combined
+                # 오디오 전용 (가장 높은 비트레이트)
+                best_audio_only = max(
+                    (f for f in formats if f.get('acodec') != 'none' and f.get('vcodec') == 'none'),
+                    key=lambda x: x.get('abr', 0),
+                    default=None
+                )
+                
+                # 비디오+오디오 결합 (가장 높은 해상도)
+                best_combined = max(
+                    (f for f in formats if f.get('vcodec') != 'none' and f.get('acodec') != 'none'),
+                    key=lambda x: x.get('height', 0),
+                    default=None
+                )
+                
+                return metadata, best_video_only, best_audio_only, best_combined
 
         except Exception as e:
-            return None, None, None
+            return None, None, None, None
 
-    def format_output(self, metadata, best_video_only, combined):
+    def format_output(self, metadata, best_video_only, best_audio_only, best_combined):
         if not metadata:
             return "오류 발생: 메타데이터를 추출할 수 없습니다.", "다운로드 링크를 찾을 수 없습니다."
 
         metadata_str = "\n".join([f"{k}: {v}" for k, v in metadata.items()])
         
         video_only_link = f'<a href="{best_video_only["url"]}" target="_blank">비디오만 다운로드</a>' if best_video_only else ""
-        combined_link = f'<a href="{combined["url"]}" target="_blank">비디오+오디오 다운로드</a>' if combined else ""
+        audio_only_link = f'<a href="{best_audio_only["url"]}" target="_blank">오디오만 다운로드</a>' if best_audio_only else ""
+        combined_link = f'<a href="{best_combined["url"]}" target="_blank">비디오+오디오 다운로드</a>' if best_combined else ""
         
-        download_links = f"{video_only_link}<br>{combined_link}" if video_only_link or combined_link else "다운로드 링크를 찾을 수 없습니다."
+        download_links = f"{video_only_link}<br>{audio_only_link}<br>{combined_link}" if any([video_only_link, audio_only_link, combined_link]) else "다운로드 링크를 찾을 수 없습니다."
         
         return metadata_str, download_links