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)