go-whisper-api/api/garbage.go

package api

import (
    "go-whisper-api/garbage"
    "go-whisper-api/whisper"
)

func applyGarbage(r whisper.TranscriptResult, patterns []string) whisper.TranscriptResult {
    if len(patterns) == 0 {
        return r
    }
    r.Text = garbage.FilterText(r.Text, patterns)
    if len(r.Words) == 0 {
        return r
    }
    gw := make([]garbage.Word, len(r.Words))
    for i, w := range r.Words {
        gw[i] = garbage.Word{Word: w.Word, Start: w.Start, Stop: w.Stop}
    }
    gw = garbage.FilterWords(gw, patterns)
    r.Words = make([]whisper.Word, len(gw))
    for i, w := range gw {
        r.Words[i] = whisper.Word{Word: w.Word, Start: w.Start, Stop: w.Stop}
    }
    return r
}