mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-07-15 09:28:27 +08:00
55af1d70f3
Ports remaining Go parser wiring and PDF backends, adds tenant-aware VLM dispatch, aligns post-processing with Python, and adds end-to-end pipeline coverage with a generated six-page PDF.
47 lines
1.1 KiB
Go
47 lines
1.1 KiB
Go
//go:build cgo
|
|
|
|
package component
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/base64"
|
|
"fmt"
|
|
"image/png"
|
|
|
|
deepdocpdf "ragflow/internal/deepdoc/parser/pdf"
|
|
)
|
|
|
|
const pdfVisionZoom = 3.0
|
|
|
|
func defaultRenderPDFVisionPages(binary []byte) ([]pdfVisionPage, error) {
|
|
engine, err := deepdocpdf.NewEngine(binary)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer engine.Close()
|
|
|
|
pageCount, err := engine.PageCount()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
pages := make([]pdfVisionPage, 0, pageCount)
|
|
for pageIdx := 0; pageIdx < pageCount; pageIdx++ {
|
|
img, err := deepdocpdf.RenderPageToImage(engine, pageIdx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("page %d: %w", pageIdx+1, err)
|
|
}
|
|
var buf bytes.Buffer
|
|
if err := png.Encode(&buf, img); err != nil {
|
|
return nil, fmt.Errorf("page %d encode png: %w", pageIdx+1, err)
|
|
}
|
|
bounds := img.Bounds()
|
|
pages = append(pages, pdfVisionPage{
|
|
PageNumber: pageIdx + 1,
|
|
WidthPts: float64(bounds.Dx()) / pdfVisionZoom,
|
|
HeightPts: float64(bounds.Dy()) / pdfVisionZoom,
|
|
ImageURL: "data:image/png;base64," + base64.StdEncoding.EncodeToString(buf.Bytes()),
|
|
})
|
|
}
|
|
return pages, nil
|
|
}
|