You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
61 lines
1.2 KiB
61 lines
1.2 KiB
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"time"
|
|
|
|
// Package imports
|
|
whisper "github.com/ggerganov/whisper.cpp/bindings/go/pkg/whisper"
|
|
wav "github.com/go-audio/wav"
|
|
)
|
|
|
|
func Process(model whisper.Model, path string) error {
|
|
var data []float32
|
|
|
|
// Create processing context
|
|
context, err := model.NewContext()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Open the file
|
|
fh, err := os.Open(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer fh.Close()
|
|
|
|
// Decode the WAV file
|
|
dec := wav.NewDecoder(fh)
|
|
if buf, err := dec.FullPCMBuffer(); err != nil {
|
|
return err
|
|
} else if dec.SampleRate != whisper.SampleRate {
|
|
return fmt.Errorf("unsupported sample rate: %d", dec.SampleRate)
|
|
} else if dec.NumChans != 1 {
|
|
return fmt.Errorf("unsupported number of channels: %d", dec.NumChans)
|
|
} else {
|
|
data = buf.AsFloat32Buffer().Data
|
|
}
|
|
|
|
// Process the data
|
|
if err := context.Process(data); err != nil {
|
|
return err
|
|
}
|
|
|
|
// Print out the results
|
|
for {
|
|
segment, err := context.NextSegment()
|
|
if err == io.EOF {
|
|
break
|
|
} else if err != nil {
|
|
return err
|
|
}
|
|
fmt.Printf("[%6s->%6s] %s\n", segment.Start.Truncate(time.Millisecond), segment.End.Truncate(time.Millisecond), segment.Text)
|
|
}
|
|
|
|
// Return success
|
|
return nil
|
|
}
|