grail/gf-lt
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Enha: remove special tokens from whisper resp

Browse Source
This commit is contained in:
Grail Finder
2025-05-18 15:28:34 +03:00
parent 441225ede8
commit 2e5755c28a
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
extra/stt.go
View File

@@ -9,11 +9,14 @@ import (
"log/slog" "log/slog"
"mime/multipart" "mime/multipart"
"net/http" "net/http"
"regexp"
"strings" "strings"
"github.com/gordonklaus/portaudio" "github.com/gordonklaus/portaudio"
) )
var specialRE = regexp.MustCompile(`\[.*?\]`)
type STT interface { type STT interface {
StartRecording() error StartRecording() error
StopRecording() (string, error) StopRecording() (string, error)
@@ -99,7 +102,9 @@ func (stt *WhisperSTT) StopRecording() (string, error) {
stt.logger.Error("fn: StopRecording", "error", err) stt.logger.Error("fn: StopRecording", "error", err)
return "", err return "", err
} }
return strings.TrimRight(string(responseTextBytes), "\n"), nil resptext := strings.TrimRight(string(responseTextBytes), "\n")
// in case there are special tokens like [_BEG_]
return specialRE.ReplaceAllString(resptext, ""), nil
} }
func (stt *WhisperSTT) writeWavHeader(w io.Writer, dataSize int) { func (stt *WhisperSTT) writeWavHeader(w io.Writer, dataSize int) {