117 lines
2.9 KiB
Go
117 lines
2.9 KiB
Go
package voice
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"log"
|
|
"mime/multipart"
|
|
"net/http"
|
|
"os"
|
|
"path/filepath"
|
|
"time"
|
|
)
|
|
|
|
type GroqTranscriber struct {
|
|
apiKey string
|
|
apiBase string
|
|
httpClient *http.Client
|
|
}
|
|
|
|
type TranscriptionResponse struct {
|
|
Text string `json:"text"`
|
|
Language string `json:"language,omitempty"`
|
|
Duration float64 `json:"duration,omitempty"`
|
|
}
|
|
|
|
func NewGroqTranscriber(apiKey string) *GroqTranscriber {
|
|
apiBase := "https://api.groq.com/openai/v1"
|
|
return &GroqTranscriber{
|
|
apiKey: apiKey,
|
|
apiBase: apiBase,
|
|
httpClient: &http.Client{
|
|
Timeout: 60 * time.Second,
|
|
},
|
|
}
|
|
}
|
|
|
|
func (t *GroqTranscriber) Transcribe(ctx context.Context, audioFilePath string) (*TranscriptionResponse, error) {
|
|
log.Printf("Starting transcription for audio file: %s", audioFilePath)
|
|
|
|
audioFile, err := os.Open(audioFilePath)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to open audio file: %w", err)
|
|
}
|
|
defer audioFile.Close()
|
|
|
|
fileInfo, err := audioFile.Stat()
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to get file info: %w", err)
|
|
}
|
|
|
|
var requestBody bytes.Buffer
|
|
writer := multipart.NewWriter(&requestBody)
|
|
|
|
part, err := writer.CreateFormFile("file", filepath.Base(audioFilePath))
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to create form file: %w", err)
|
|
}
|
|
|
|
if _, err := io.Copy(part, audioFile); err != nil {
|
|
return nil, fmt.Errorf("failed to copy file content: %w", err)
|
|
}
|
|
|
|
if err := writer.WriteField("model", "whisper-large-v3"); err != nil {
|
|
return nil, fmt.Errorf("failed to write model field: %w", err)
|
|
}
|
|
|
|
if err := writer.WriteField("response_format", "json"); err != nil {
|
|
return nil, fmt.Errorf("failed to write response_format field: %w", err)
|
|
}
|
|
|
|
if err := writer.Close(); err != nil {
|
|
return nil, fmt.Errorf("failed to close multipart writer: %w", err)
|
|
}
|
|
|
|
url := t.apiBase + "/audio/transcriptions"
|
|
req, err := http.NewRequestWithContext(ctx, "POST", url, &requestBody)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to create request: %w", err)
|
|
}
|
|
|
|
req.Header.Set("Content-Type", writer.FormDataContentType())
|
|
req.Header.Set("Authorization", "Bearer "+t.apiKey)
|
|
|
|
log.Printf("Sending transcription request to Groq API (file size: %d bytes)", fileInfo.Size())
|
|
|
|
resp, err := t.httpClient.Do(req)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to send request: %w", err)
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
body, err := io.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to read response: %w", err)
|
|
}
|
|
|
|
if resp.StatusCode != http.StatusOK {
|
|
return nil, fmt.Errorf("API error (status %d): %s", resp.StatusCode, string(body))
|
|
}
|
|
|
|
var result TranscriptionResponse
|
|
if err := json.Unmarshal(body, &result); err != nil {
|
|
return nil, fmt.Errorf("failed to unmarshal response: %w", err)
|
|
}
|
|
|
|
log.Printf("Transcription completed successfully (text length: %d chars)", len(result.Text))
|
|
|
|
return &result, nil
|
|
}
|
|
|
|
func (t *GroqTranscriber) IsAvailable() bool {
|
|
return t.apiKey != ""
|
|
}
|