try fix Summary page

pages/Summary.py

@@ -6,6 +6,8 @@ from dotenv import load_dotenv
 import re
 import subprocess
 import json
+import yt_dlp
+
 
 # Загрузка переменных окружения из .env файла
 load_dotenv()
@@ -45,27 +47,36 @@ def get_transcript(video_id):
 
 def get_transcript_via_ytdlp(video_id):
     try:
-        # Команда для загрузки субтитров через yt-dlp
-        command = [
-            "yt-dlp",
-            "--write-auto-sub",  # Загружаем автоматические субтитры
-            "--sub-lang", "en",  # Используем английские автоматические субтитры
-            "--skip-download",  # Не загружаем само видео
-            f"https://www.youtube.com/watch?v={video_id}",
-            "--output", f"{video_id}.%(ext)s",
-            "--write-sub",  # Загружаем субтитры, если доступны
-            "--sub-format", "json"  # Загружаем субтитры в формате JSON
-        ]
-        
-        subprocess.run(command, check=True)
+        # Настройка параметров для загрузки субтитров
+        ydl_opts = {
+            'writesubtitles': True,
+            'writeautomaticsub': True,  # Загружаем автоматические субтитры
+            'subtitlesformat': 'json',  # Формат субтитров
+            'skip_download': True,  # Не загружаем видео
+            'subtitleslangs': ['en'],  # Используем английские автоматические субтитры
+            'outtmpl': f'{video_id}.%(ext)s'
+        }
+
+        # Использование yt-dlp API для загрузки субтитров
+        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+            info_dict = ydl.extract_info(f"https://www.youtube.com/watch?v={video_id}", download=False)
+            subtitles = info_dict.get('subtitles', {})
+            automatic_captions = info_dict.get('automatic_captions', {})
+
+            # Проверяем наличие автоматических субтитров
+            if 'en' in automatic_captions:
+                caption_url = automatic_captions['en'][0]['url']
+                ydl.download([caption_url])
+
+                with open(f"{video_id}.en.vtt.json", 'r', encoding='utf-8') as file:
+                    data = json.load(file)
+                    transcript = ' '.join([item['text'] for item in data['events']])
+                    return transcript
+            else:
+                raise RuntimeError("Автоматические субтитры не найдены.")
 
-        # Открываем файл субтитров
-        with open(f"{video_id}.en.vtt.json", 'r', encoding='utf-8') as file:
-            data = json.load(file)
-            transcript = ' '.join([item['text'] for item in data['events']])
-            return transcript
     except Exception as e:
-        raise RuntimeError(f"Не удалось получить автоматические субтитры через yt-dlp: {e}")
+        raise RuntimeError(f"Ошибка при загрузке субтитров через yt-dlp: {e}")
 
 def generate_summary_with_claude(transcript, prompt_text):
     try:

requirements.txt

@@ -11,4 +11,5 @@ youtube-transcript-api
 streamlit
 google-api-python-client
 faiss-cpu
-Pillow
+Pillow
+yt-dlp