package formula

import (
	"gitea.com/bitwsd/document_ai/pkg/common"
	"github.com/gin-gonic/gin"
)

func SetupRouter(engine *gin.RouterGroup) {
	endpoint := NewFormulaEndpoint()
	formulaRouter := engine.Group("/formula", common.GetAuthMiddleware())
	{
		formulaRouter.POST("/recognition", endpoint.CreateTask)
		formulaRouter.POST("/ai_enhance", endpoint.AIEnhanceRecognition)
		formulaRouter.GET("/recognition/:task_no", endpoint.GetTaskStatus)
	}
}