mirror of
https://github.com/xzeldon/whisper-api-server.git
synced 2024-12-28 01:45:46 +00:00
Transcribe using file buffer straight from form
This commit is contained in:
parent
dde206facd
commit
c4daf3ec71
3
.gitignore
vendored
3
.gitignore
vendored
@ -1,5 +1,6 @@
|
|||||||
Whisper.dll
|
Whisper.dll
|
||||||
ggml-medium.bin
|
ggml-*
|
||||||
|
*.exe
|
||||||
|
|
||||||
whisper-api-server.exe
|
whisper-api-server.exe
|
||||||
|
|
||||||
|
@ -1,6 +1,7 @@
|
|||||||
package api
|
package api
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"io"
|
||||||
"net/http"
|
"net/http"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
@ -11,7 +12,7 @@ type TranscribeResponse struct {
|
|||||||
Text string `json:"text"`
|
Text string `json:"text"`
|
||||||
}
|
}
|
||||||
|
|
||||||
func Transcribe(c echo.Context, whisperState *WhisperState) error {
|
func TranscribeFromFile(c echo.Context, whisperState *WhisperState) error {
|
||||||
audioPath, err := saveFormFile("file", c)
|
audioPath, err := saveFormFile("file", c)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.Logger().Errorf("Error reading file: %s", err)
|
c.Logger().Errorf("Error reading file: %s", err)
|
||||||
@ -26,6 +27,11 @@ func Transcribe(c echo.Context, whisperState *WhisperState) error {
|
|||||||
|
|
||||||
err = whisperState.context.RunFull(whisperState.params, buffer)
|
err = whisperState.context.RunFull(whisperState.params, buffer)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
c.Logger().Errorf("Error processing audio: %s", err)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
result, err := getResult(whisperState.context)
|
result, err := getResult(whisperState.context)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.Logger().Error(err)
|
c.Logger().Error(err)
|
||||||
@ -43,3 +49,59 @@ func Transcribe(c echo.Context, whisperState *WhisperState) error {
|
|||||||
|
|
||||||
return c.JSON(http.StatusOK, response)
|
return c.JSON(http.StatusOK, response)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func Transcribe(c echo.Context, whisperState *WhisperState) error {
|
||||||
|
// Get the file header
|
||||||
|
fileHeader, err := c.FormFile("file")
|
||||||
|
if err != nil {
|
||||||
|
c.Logger().Errorf("Error retrieving the file: %s", err)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Open the file
|
||||||
|
file, err := fileHeader.Open()
|
||||||
|
if err != nil {
|
||||||
|
c.Logger().Errorf("Error opening the file: %s", err)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer file.Close()
|
||||||
|
|
||||||
|
// Read the file into a buffer
|
||||||
|
buffer, err := io.ReadAll(file)
|
||||||
|
if err != nil {
|
||||||
|
c.Logger().Errorf("Error reading the file into buffer: %s", err)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
whisperState.mutex.Lock()
|
||||||
|
defer whisperState.mutex.Unlock()
|
||||||
|
|
||||||
|
bufferSpecial, err := whisperState.media.LoadAudioFileData(&buffer, true)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
c.Logger().Errorf("Error loading audio file data: %s", err)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
err = whisperState.context.RunStreamed(whisperState.params, bufferSpecial)
|
||||||
|
if err != nil {
|
||||||
|
c.Logger().Errorf("Error processing audio: %s", err)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
result, err := getResult(whisperState.context)
|
||||||
|
if err != nil {
|
||||||
|
c.Logger().Error(err)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(result) == 0 {
|
||||||
|
return c.JSON(http.StatusInternalServerError, map[string]string{"error": "Internal server error"})
|
||||||
|
}
|
||||||
|
|
||||||
|
response := TranscribeResponse{
|
||||||
|
Text: strings.TrimLeft(result, " "),
|
||||||
|
}
|
||||||
|
|
||||||
|
return c.JSON(http.StatusOK, response)
|
||||||
|
}
|
||||||
|
@ -146,9 +146,9 @@ func (this *FullParams) TestDefaultsOK() bool {
|
|||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
if this.cStruct.Language != English {
|
// if this.cStruct.Language != English {
|
||||||
return false
|
// return false
|
||||||
}
|
// }
|
||||||
|
|
||||||
// Todo ... why do these not line up as expected.. is our struct out of alignment ?
|
// Todo ... why do these not line up as expected.. is our struct out of alignment ?
|
||||||
/*
|
/*
|
||||||
@ -214,6 +214,7 @@ func NewFullParams(cstruct *_FullParams) *FullParams {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func _newFullParams_cStruct() *_FullParams {
|
func _newFullParams_cStruct() *_FullParams {
|
||||||
|
|
||||||
return &_FullParams{
|
return &_FullParams{
|
||||||
|
|
||||||
strategy: 0,
|
strategy: 0,
|
||||||
|
@ -207,6 +207,7 @@ func (context *IContext) FullDefaultParams(strategy eSamplingStrategy) (*FullPar
|
|||||||
return nil, errors.New("FullDefaultParams did not return params")
|
return nil, errors.New("FullDefaultParams did not return params")
|
||||||
}
|
}
|
||||||
ParamObj := NewFullParams(params)
|
ParamObj := NewFullParams(params)
|
||||||
|
// ParamObj.SetLanguage(Polish)
|
||||||
|
|
||||||
if ParamObj.TestDefaultsOK() {
|
if ParamObj.TestDefaultsOK() {
|
||||||
return ParamObj, nil
|
return ParamObj, nil
|
||||||
|
Loading…
Reference in New Issue
Block a user