Subimission / submission /tools /tools_video.py
TPey
Add application file
b156b8a
Raw
History Blame Contribute Delete
674 Bytes
from tools.interfaces.interface_video_transcriptor_provider import TranscriptProviderInterface
from tools.interfaces.interface_video_provider import VideoSearchProviderInterface
class VideoTool:
def __init__(self,
search_provider: VideoSearchProviderInterface,
transcript_provider: TranscriptProviderInterface):
self.search_provider = search_provider
self.transcript_provider = transcript_provider
def search(self, query: str) -> str:
return self.search_provider.search(query)
def transcript(self, video_id_or_url: str) -> str:
return self.transcript_provider.get_transcript(video_id_or_url)