Files
soft_usite/pkg/services/ocr_service.go
T
Lizandro GuarnizoandClaude Sonnet 5 d493d6dee6 feat: agrega integraciones OCR y Whisper ASR (servicios propios)
Nuevos submódulos en Integraciones para conectar el servicio propio de
OCR (extracción de texto de imágenes) y el de transcripción de audio
self-hosted (whisper-asr-webservice, Basic Auth), con panel de
configuración y prueba en vivo, siguiendo el patrón ya usado por
WebSMS/Hostinger.

Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>
2026-08-13 10:09:40 -05:00

68 lines
1.8 KiB
Go

package services
import (
"bytes"
"encoding/base64"
"encoding/json"
"fmt"
"io"
"net/http"
"time"
"github.com/sujit-baniya/fiber-boilerplate/pkg/models"
)
var ocrHTTPClient = &http.Client{Timeout: 30 * time.Second}
// ExtraerTextoOCR manda una imagen al servicio propio de OCR y devuelve el
// texto extraído. mimeType ej: "image/png", "image/jpeg".
func ExtraerTextoOCR(imagenBytes []byte, mimeType string) (string, error) {
cfg, err := models.GetOcrConfig()
if err != nil {
return "", fmt.Errorf("el servicio de OCR no está configurado (Integraciones → OCR)")
}
body, err := json.Marshal(map[string]string{
"image_base64": base64.StdEncoding.EncodeToString(imagenBytes),
"mime_type": mimeType,
})
if err != nil {
return "", err
}
req, err := http.NewRequest(http.MethodPost, cfg.BaseURL, bytes.NewReader(body))
if err != nil {
return "", err
}
req.Header.Set("Authorization", "Bearer "+cfg.Token)
req.Header.Set("Content-Type", "application/json")
resp, err := ocrHTTPClient.Do(req)
if err != nil {
return "", fmt.Errorf("no se pudo contactar el servicio de OCR: %w", err)
}
defer resp.Body.Close()
raw, _ := io.ReadAll(io.LimitReader(resp.Body, 2*1024*1024))
if resp.StatusCode < 200 || resp.StatusCode >= 300 {
return "", fmt.Errorf("el servicio de OCR respondió %d: %s", resp.StatusCode, string(raw))
}
var out struct {
Success bool `json:"success"`
Text string `json:"text"`
Error string `json:"error"`
}
if err := json.Unmarshal(raw, &out); err != nil {
return "", fmt.Errorf("respuesta inesperada del servicio de OCR: %s", string(raw))
}
if !out.Success {
msg := out.Error
if msg == "" {
msg = "el servicio de OCR no pudo procesar la imagen"
}
return "", fmt.Errorf("%s", msg)
}
return out.Text, nil
}