58 lines
1.2 KiB
Go
58 lines
1.2 KiB
Go
package handler
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"tencent_ocr/pkg/service"
|
|
)
|
|
|
|
type RateHandler struct {
|
|
geminiService *service.GeminiService
|
|
}
|
|
|
|
func NewRateHandler(geminiService *service.GeminiService) *RateHandler {
|
|
return &RateHandler{
|
|
geminiService: geminiService,
|
|
}
|
|
}
|
|
|
|
type RateRequest struct {
|
|
Text string `json:"text" binding:"required"`
|
|
Criteria string `json:"criteria"`
|
|
WritingRequirement string `json:"writing_requirement"`
|
|
}
|
|
|
|
type RateResponse struct {
|
|
Result string `json:"result"`
|
|
Success bool `json:"success"`
|
|
}
|
|
|
|
func (h *RateHandler) HandleRate(c *gin.Context) {
|
|
var req RateRequest
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
c.JSON(http.StatusBadRequest, RateResponse{
|
|
Success: false,
|
|
Result: "Invalid request format",
|
|
})
|
|
return
|
|
}
|
|
|
|
// Process with Gemini
|
|
criteria := req.Criteria
|
|
if criteria == "" {
|
|
criteria =
|
|
result, err := h.geminiService.ProcessText(c.Request.Context(), req.Text)
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, RateResponse{
|
|
Success: false,
|
|
Result: "Text processing failed",
|
|
})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, RateResponse{
|
|
Success: true,
|
|
Result: result,
|
|
})
|
|
} |