vrr/open-notebook
2
0
Fork 0
mirror of https://github.com/lfnovo/open-notebook.git synced 2026-04-29 03:50:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

configurable language files

Browse source
This commit is contained in:
LUIS NOVO 2024-10-24 16:45:12 -03:00
parent 6df77c5b92
commit 53d86e2e16
1 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

9
open_notebook/graphs/content_process.py
View file

@ -9,6 +9,8 @@ from typing_extensions import TypedDict
from youtube_transcript_api import YouTubeTranscriptApi # type: ignore
from youtube_transcript_api.formatters import TextFormatter # type: ignore
from open_notebook.config import CONFIG
class SourceState(TypedDict):
content: str
@ -183,9 +185,12 @@ def extract_youtube_transcript(state: SourceState):
Parse the text file and print its content.
"""
transcript = YouTubeTranscriptApi.get_transcript(
_extract_youtube_id(state.get("url")), languages=["pt", "en"]
languages = CONFIG.get("youtube_transcripts", {}).get(
"preferred_languages", ["pt", "en"]
)
video_id = _extract_youtube_id(state.get("url"))
transcript = YouTubeTranscriptApi.get_transcript(video_id, languages=languages)
formatter = TextFormatter()
title = _get_title(state.get("url"))
return {"content": formatter.format_transcript(transcript), "title": title}