from API import aud2text, Summarize
from moviepy.editor import *

class Vid2SumModel:

    def predict(self, vid_file):
        video = VideoFileClip(vid_file)
        audio_file = "converted_audio.mp3"
        video.audio.write_audiofile(audio_file)

        text = self.aud2text(audio_file)
        sum_text = self.summarize(text["text"])
        return sum_text[0]["summary_text"]