paperless-gpt/ocr.go

package main

import (
	"context"
	"fmt"
	"os"
	"strings"
)

// ProcessDocumentOCR processes a document through OCR and returns the combined text
func (app *App) ProcessDocumentOCR(ctx context.Context, documentID int) (string, error) {
	docLogger := documentLogger(documentID)
	docLogger.Info("Starting OCR processing")

	imagePaths, err := app.Client.DownloadDocumentAsImages(ctx, documentID, limitOcrPages)
	defer func() {
		for _, imagePath := range imagePaths {
			if err := os.Remove(imagePath); err != nil {
				docLogger.WithError(err).WithField("image_path", imagePath).Warn("Failed to remove temporary image file")
			}
		}
	}()
	if err != nil {
		return "", fmt.Errorf("error downloading document images for document %d: %w", documentID, err)
	}

	docLogger.WithField("page_count", len(imagePaths)).Debug("Downloaded document images")

	var ocrTexts []string
	for i, imagePath := range imagePaths {
		pageLogger := docLogger.WithField("page", i+1)
		pageLogger.Debug("Processing page")

		imageContent, err := os.ReadFile(imagePath)
		if err != nil {
			return "", fmt.Errorf("error reading image file for document %d, page %d: %w", documentID, i+1, err)
		}

		ocrText, err := app.doOCRViaLLM(ctx, imageContent, pageLogger)
		if err != nil {
			return "", fmt.Errorf("error performing OCR for document %d, page %d: %w", documentID, i+1, err)
		}
		pageLogger.Debug("OCR completed for page")

		ocrTexts = append(ocrTexts, ocrText)
	}

	docLogger.Info("OCR processing completed successfully")
	return strings.Join(ocrTexts, "\n\n"), nil
}