diff --git a/main.go b/main.go index b98484a..11b4c05 100755 --- a/main.go +++ b/main.go @@ -158,6 +158,9 @@ func main() { &models.UmindEventoLog{}, // API Keys de /api/v2 (token + IP obligatoria + scopes) &models.ApiKey{}, + // Integraciones: OCR y transcripción de audio (servicios propios) + &models.OcrConfig{}, + &models.WhisperAsrConfig{}, } for _, m := range modelosBase { if err := app.Http.Database.DB.AutoMigrate(m); err != nil { diff --git a/migrations/migrate.go b/migrations/migrate.go index cb6864b..a23132d 100755 --- a/migrations/migrate.go +++ b/migrations/migrate.go @@ -129,6 +129,9 @@ func Migrate() { &models.UmindEventoLog{}, // API Keys de /api/v2 (token + IP obligatoria + scopes) &models.ApiKey{}, + // Integraciones: OCR y transcripción de audio (servicios propios) + &models.OcrConfig{}, + &models.WhisperAsrConfig{}, } for _, m := range modelosPrincipales { if err := app.Http.Database.DB.Migrator().AutoMigrate(m); err != nil { @@ -571,6 +574,8 @@ func SeedIntegraciones() { {"Hostinger", "Panel de VPS, dominios, hosting y DNS de Hostinger", "/app/hostinger"}, {"Cloudflare", "Gestión de zonas, DNS, SSL y firewall en Cloudflare", "/app/cloudflare"}, {"VCard API", "Integración con el sistema VCard externo (Laravel + Sanctum): usuarios, membresías, vcards, pagos y más", "/app/vcard-api"}, + {"OCR", "Extracción de texto de imágenes (comprobantes, capturas) vía servicio OCR propio", "/app/ocr"}, + {"Whisper ASR", "Transcripción de audio vía servicio propio de reconocimiento de voz (self-hosted)", "/app/whisper-asr"}, } var insertados []models.Submodules diff --git a/pkg/models/ocr_config.go b/pkg/models/ocr_config.go new file mode 100644 index 0000000..029ecee --- /dev/null +++ b/pkg/models/ocr_config.go @@ -0,0 +1,41 @@ +package models + +import ( + "github.com/sujit-baniya/fiber-boilerplate/app" + "gorm.io/gorm" +) + +// OcrConfig almacena la conexión al servicio propio de OCR (extracción de +// texto de imágenes, ej. comprobantes de pago). Solo un registro activo a +// la vez, mismo patrón que HostingerConfig/WebSmsConfig. +type OcrConfig struct { + gorm.Model + BaseURL string `json:"base_url" gorm:"column:base_url;type:text;not null"` // ej: https://ocr.u-s.app/extract + Token string `json:"token" gorm:"column:token;type:text;not null"` // Bearer token + Notas string `json:"notas" gorm:"column:notas;type:text"` + Activo bool `json:"activo" gorm:"column:activo;default:true"` +} + +func (OcrConfig) TableName() string { return "ocr_config" } + +func GetOcrConfig() (*OcrConfig, error) { + var item OcrConfig + if err := app.Http.Database.DB.Where("activo = ?", true).Order("id DESC").First(&item).Error; err != nil { + return nil, err + } + return &item, nil +} + +func SaveOcrConfig(s OcrConfig) error { + app.Http.Database.DB.Model(&OcrConfig{}).Where("activo = ?", true).Update("activo", false) + s.Activo = true + if s.ID > 0 { + return app.Http.Database.DB.Model(&s).Updates(map[string]interface{}{ + "base_url": s.BaseURL, + "token": s.Token, + "notas": s.Notas, + "activo": true, + }).Error + } + return app.Http.Database.DB.Create(&s).Error +} diff --git a/pkg/models/whisper_asr_config.go b/pkg/models/whisper_asr_config.go new file mode 100644 index 0000000..8e85ae4 --- /dev/null +++ b/pkg/models/whisper_asr_config.go @@ -0,0 +1,44 @@ +package models + +import ( + "github.com/sujit-baniya/fiber-boilerplate/app" + "gorm.io/gorm" +) + +// WhisperAsrConfig almacena la conexión al servicio propio de transcripción +// de audio (whisper-asr-webservice self-hosted, autenticado con Basic Auth) +// — distinto del Whisper de OpenAI que ya se configura vía AiConfig con +// modulo "whisper" para el bot de Telegram. Solo un registro activo a la vez. +type WhisperAsrConfig struct { + gorm.Model + BaseURL string `json:"base_url" gorm:"column:base_url;type:text;not null"` // ej: https://whisper.u-s.app/asr + Username string `json:"username" gorm:"column:username;size:255;not null"` + Password string `json:"password" gorm:"column:password;type:text;not null"` + Notas string `json:"notas" gorm:"column:notas;type:text"` + Activo bool `json:"activo" gorm:"column:activo;default:true"` +} + +func (WhisperAsrConfig) TableName() string { return "whisper_asr_config" } + +func GetWhisperAsrConfig() (*WhisperAsrConfig, error) { + var item WhisperAsrConfig + if err := app.Http.Database.DB.Where("activo = ?", true).Order("id DESC").First(&item).Error; err != nil { + return nil, err + } + return &item, nil +} + +func SaveWhisperAsrConfig(s WhisperAsrConfig) error { + app.Http.Database.DB.Model(&WhisperAsrConfig{}).Where("activo = ?", true).Update("activo", false) + s.Activo = true + if s.ID > 0 { + return app.Http.Database.DB.Model(&s).Updates(map[string]interface{}{ + "base_url": s.BaseURL, + "username": s.Username, + "password": s.Password, + "notas": s.Notas, + "activo": true, + }).Error + } + return app.Http.Database.DB.Create(&s).Error +} diff --git a/pkg/services/ocr_service.go b/pkg/services/ocr_service.go new file mode 100644 index 0000000..1bf7ceb --- /dev/null +++ b/pkg/services/ocr_service.go @@ -0,0 +1,67 @@ +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 +} diff --git a/pkg/services/whisper_asr_service.go b/pkg/services/whisper_asr_service.go new file mode 100644 index 0000000..0379652 --- /dev/null +++ b/pkg/services/whisper_asr_service.go @@ -0,0 +1,67 @@ +package services + +import ( + "bytes" + "encoding/json" + "fmt" + "io" + "mime/multipart" + "net/http" + "time" + + "github.com/sujit-baniya/fiber-boilerplate/pkg/models" +) + +var whisperAsrHTTPClient = &http.Client{Timeout: 120 * time.Second} // audio largo puede tardar + +// TranscribirAudioSelfHosted manda un archivo de audio al servicio propio de +// transcripción (whisper-asr-webservice, Basic Auth) y devuelve el texto. +func TranscribirAudioSelfHosted(audioBytes []byte, filename string) (string, error) { + cfg, err := models.GetWhisperAsrConfig() + if err != nil { + return "", fmt.Errorf("el servicio de transcripción no está configurado (Integraciones → Whisper ASR)") + } + if filename == "" { + filename = "audio.wav" + } + + var buf bytes.Buffer + w := multipart.NewWriter(&buf) + fw, err := w.CreateFormFile("audio_file", filename) + if err != nil { + return "", err + } + if _, err := fw.Write(audioBytes); err != nil { + return "", err + } + _ = w.WriteField("response_format", "json") + if err := w.Close(); err != nil { + return "", err + } + + req, err := http.NewRequest(http.MethodPost, cfg.BaseURL, &buf) + if err != nil { + return "", err + } + req.Header.Set("Content-Type", w.FormDataContentType()) + req.SetBasicAuth(cfg.Username, cfg.Password) + + resp, err := whisperAsrHTTPClient.Do(req) + if err != nil { + return "", fmt.Errorf("no se pudo contactar el servicio de transcripción: %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 transcripción respondió %d: %s", resp.StatusCode, string(raw)) + } + + var out struct { + Text string `json:"text"` + } + if err := json.Unmarshal(raw, &out); err != nil { + return "", fmt.Errorf("respuesta inesperada del servicio de transcripción: %s", string(raw)) + } + return out.Text, nil +} diff --git a/resources/views/ocr_config.html b/resources/views/ocr_config.html new file mode 100644 index 0000000..d3835a2 --- /dev/null +++ b/resources/views/ocr_config.html @@ -0,0 +1,107 @@ +
+ +
+
+

OCR

+

Extracción de texto de imágenes vía servicio propio

+
+
+ + +
+

Configuración

+
+
+
+ + +
+
+ +
+ + +
+
+
+ + +
+
+

+

Configuración guardada

+
+ +
+
+
+ + +
+

Probar extracción

+
+ + +

+
+
+ +
+
+
+ +
+ + + + diff --git a/resources/views/whisper_asr_config.html b/resources/views/whisper_asr_config.html new file mode 100644 index 0000000..7bb96e3 --- /dev/null +++ b/resources/views/whisper_asr_config.html @@ -0,0 +1,111 @@ +
+ +
+
+

Whisper ASR

+

Transcripción de audio vía servicio propio (self-hosted)

+
+
+ + +
+

Configuración

+
+
+
+ + +
+
+ + +
+
+ +
+ + +
+
+
+ + +
+
+

+

Configuración guardada

+
+ +
+
+
+ + +
+

Probar transcripción

+
+ + +

+
+
+ +
+
+
+ +
+ + + + diff --git a/rest/controllers/ocr_controller.go b/rest/controllers/ocr_controller.go new file mode 100644 index 0000000..5267972 --- /dev/null +++ b/rest/controllers/ocr_controller.go @@ -0,0 +1,84 @@ +package controllers + +import ( + "io" + + "github.com/gofiber/fiber/v2" + "github.com/sujit-baniya/fiber-boilerplate/pkg/models" + "github.com/sujit-baniya/fiber-boilerplate/pkg/services" +) + +func OcrConfigPage(c *fiber.Ctx) error { + cfg, _ := models.GetOcrConfig() + return c.Render("ocr_config", fiber.Map{ + "user": c.Locals("user"), + "modules": c.Locals("userModules"), + "cfg": cfg, + }, "layouts/main") +} + +func GetOcrConfigHandler(c *fiber.Ctx) error { + cfg, err := models.GetOcrConfig() + if err != nil { + return c.JSON(fiber.Map{"data": nil}) + } + return c.JSON(fiber.Map{"data": cfg}) +} + +func SaveOcrConfigHandler(c *fiber.Ctx) error { + type body struct { + ID uint `json:"id"` + BaseURL string `json:"base_url"` + Token string `json:"token"` + Notas string `json:"notas"` + } + var b body + if err := c.BodyParser(&b); err != nil { + return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "body inválido"}) + } + if b.BaseURL == "" { + return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "base_url es requerida"}) + } + if b.Token == "" { + return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "token es requerido"}) + } + + cfg := models.OcrConfig{BaseURL: b.BaseURL, Token: b.Token, Notas: b.Notas} + cfg.ID = b.ID + if err := models.SaveOcrConfig(cfg); err != nil { + return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{"error": err.Error()}) + } + return c.JSON(fiber.Map{"message": "Configuración guardada"}) +} + +// TestOcrConfigHandler recibe una imagen de prueba y devuelve el texto que +// extrae el servicio configurado — confirma que la URL/token funcionan de +// verdad, no solo que se guardaron. +func TestOcrConfigHandler(c *fiber.Ctx) error { + fh, err := c.FormFile("imagen") + if err != nil { + return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "imagen requerida"}) + } + if fh.Size > 8*1024*1024 { + return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "máximo 8MB"}) + } + f, err := fh.Open() + if err != nil { + return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{"error": "no se pudo leer el archivo"}) + } + defer f.Close() + data, err := io.ReadAll(f) + if err != nil { + return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{"error": "no se pudo leer el archivo"}) + } + + mimeType := fh.Header.Get("Content-Type") + if mimeType == "" { + mimeType = "image/png" + } + texto, err := services.ExtraerTextoOCR(data, mimeType) + if err != nil { + return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": err.Error()}) + } + return c.JSON(fiber.Map{"text": texto}) +} diff --git a/rest/controllers/whisper_asr_controller.go b/rest/controllers/whisper_asr_controller.go new file mode 100644 index 0000000..201cf1d --- /dev/null +++ b/rest/controllers/whisper_asr_controller.go @@ -0,0 +1,80 @@ +package controllers + +import ( + "io" + + "github.com/gofiber/fiber/v2" + "github.com/sujit-baniya/fiber-boilerplate/pkg/models" + "github.com/sujit-baniya/fiber-boilerplate/pkg/services" +) + +func WhisperAsrConfigPage(c *fiber.Ctx) error { + cfg, _ := models.GetWhisperAsrConfig() + return c.Render("whisper_asr_config", fiber.Map{ + "user": c.Locals("user"), + "modules": c.Locals("userModules"), + "cfg": cfg, + }, "layouts/main") +} + +func GetWhisperAsrConfigHandler(c *fiber.Ctx) error { + cfg, err := models.GetWhisperAsrConfig() + if err != nil { + return c.JSON(fiber.Map{"data": nil}) + } + return c.JSON(fiber.Map{"data": cfg}) +} + +func SaveWhisperAsrConfigHandler(c *fiber.Ctx) error { + type body struct { + ID uint `json:"id"` + BaseURL string `json:"base_url"` + Username string `json:"username"` + Password string `json:"password"` + Notas string `json:"notas"` + } + var b body + if err := c.BodyParser(&b); err != nil { + return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "body inválido"}) + } + if b.BaseURL == "" { + return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "base_url es requerida"}) + } + if b.Username == "" || b.Password == "" { + return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "usuario y contraseña son requeridos"}) + } + + cfg := models.WhisperAsrConfig{BaseURL: b.BaseURL, Username: b.Username, Password: b.Password, Notas: b.Notas} + cfg.ID = b.ID + if err := models.SaveWhisperAsrConfig(cfg); err != nil { + return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{"error": err.Error()}) + } + return c.JSON(fiber.Map{"message": "Configuración guardada"}) +} + +// TestWhisperAsrConfigHandler recibe un audio de prueba y devuelve la +// transcripción — confirma que la URL/credenciales funcionan de verdad. +func TestWhisperAsrConfigHandler(c *fiber.Ctx) error { + fh, err := c.FormFile("audio") + if err != nil { + return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "audio requerido"}) + } + if fh.Size > 20*1024*1024 { + return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "máximo 20MB"}) + } + f, err := fh.Open() + if err != nil { + return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{"error": "no se pudo leer el archivo"}) + } + defer f.Close() + data, err := io.ReadAll(f) + if err != nil { + return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{"error": "no se pudo leer el archivo"}) + } + + texto, err := services.TranscribirAudioSelfHosted(data, fh.Filename) + if err != nil { + return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": err.Error()}) + } + return c.JSON(fiber.Map{"text": texto}) +} diff --git a/rest/routes/user.go b/rest/routes/user.go index ab472ed..d10d136 100755 --- a/rest/routes/user.go +++ b/rest/routes/user.go @@ -619,6 +619,18 @@ func UserRoutes(app fiber.Router) { protected.Get("/websms/logs", controllers.GetWebSmsLogs) protected.Get("/websms/webhook-logs", controllers.GetWebSmsWebhookLogs) + // ─── OCR (servicio propio) ───────────────────────────────────────────────── + protected.Get("/ocr", middlewares.MenuMiddleware, controllers.OcrConfigPage) + protected.Get("/ocr/config", controllers.GetOcrConfigHandler) + protected.Post("/ocr/save", controllers.SaveOcrConfigHandler) + protected.Post("/ocr/test", controllers.TestOcrConfigHandler) + + // ─── Whisper ASR (transcripción, servicio propio) ───────────────────────── + protected.Get("/whisper-asr", middlewares.MenuMiddleware, controllers.WhisperAsrConfigPage) + protected.Get("/whisper-asr/config", controllers.GetWhisperAsrConfigHandler) + protected.Post("/whisper-asr/save", controllers.SaveWhisperAsrConfigHandler) + protected.Post("/whisper-asr/test", controllers.TestWhisperAsrConfigHandler) + // ─── Planes dLocal (gestión desde panel protegido) ──────────────────────── protected.Get("/dlocal/planes", apiControllers.SeePlanes) protected.Post("/dlocal/planes", apiControllers.CreatePlan)