separate source and content graph

This commit is contained in:
LUIS NOVO 2024-11-10 13:30:03 -03:00
parent b42a95b35f
commit 2e2a4947b3
12 changed files with 167 additions and 56 deletions

View file

@ -4,7 +4,7 @@ import subprocess
from loguru import logger
from open_notebook.graphs.content_processing.state import SourceState
from open_notebook.graphs.content_processing.state import ContentState
def extract_audio_from_video(input_file, output_file, stream_index):
@ -102,7 +102,7 @@ def select_best_audio_stream(streams):
return max(scored_streams, key=lambda x: x[0])[1]
def extract_best_audio_from_video(data: SourceState):
def extract_best_audio_from_video(data: ContentState):
"""
Main function to extract the best audio stream from a video file
"""