import whisper
import os

def getAllWhisperTranscripts():
    directory = os.fsencode("videos")

    for file in os.listdir(directory):
        filename = os.fsdecode(file)
        print(filename)

        model = whisper.load_model("tiny")
        result = model.transcribe("videos/" + filename, fp16=False)
        fname = "whisper/" + filename
        fname = fname.replace(".mp4", ".txt") 
        with open(fname, "w") as f:
            f.write(result["text"])
            print("written to: " + fname)

def getWhisperTranscript(videoId):
    model = whisper.load_model("tiny")
    result = model.transcribe(f"videos/{videoId}.mp4", fp16=False)
    fname = f"whisper/{videoId}.txt"
    with open(fname, "w") as f:
        f.write(result["text"])
        print("written to: " + fname)