From 4979d4e1e296d0c854553c05ff98d20d9807ee95 Mon Sep 17 00:00:00 2001
From: "huangbin.beal@163.com" <huangbin.beal@bytedance.com>
Date: Tue, 19 Mar 2024 11:53:28 +0800
Subject: [PATCH] fix: ocr cluster

---
 hrp/pkg/uixt/ext.go           | 8 ++------
 hrp/pkg/uixt/service_vedem.go | 2 +-
 2 files changed, 3 insertions(+), 7 deletions(-)

diff --git a/hrp/pkg/uixt/ext.go b/hrp/pkg/uixt/ext.go
index f234cdae..b9ad0041 100644
--- a/hrp/pkg/uixt/ext.go
+++ b/hrp/pkg/uixt/ext.go
@@ -4,9 +4,9 @@ import (
 	"bytes"
 	"fmt"
 	"image"
-	"image/gif"
+	_ "image/gif"
 	"image/jpeg"
-	"image/png"
+	_ "image/png"
 	"math/rand"
 	"mime"
 	"mime/multipart"
@@ -256,10 +256,6 @@ func (dExt *DriverExt) saveScreenShot(raw *bytes.Buffer, fileName string) (strin
 	case "jpeg", "png":
 		jpegOptions := &jpeg.Options{Quality: 95}
 		err = jpeg.Encode(file, img, jpegOptions)
-	case "png_import":
-		err = png.Encode(file, img)
-	case "gif_import":
-		err = gif.Encode(file, img, nil)
 	default:
 		return "", fmt.Errorf("unsupported image format: %s", format)
 	}
diff --git a/hrp/pkg/uixt/service_vedem.go b/hrp/pkg/uixt/service_vedem.go
index e8e1f51e..86d66dd1 100644
--- a/hrp/pkg/uixt/service_vedem.go
+++ b/hrp/pkg/uixt/service_vedem.go
@@ -219,7 +219,7 @@ func (s *veDEMImageService) GetImage(imageBuf *bytes.Buffer, options ...ActionOp
 	}
 
 	// 使用高精度集群
-	bodyWriter.WriteField("ocrCluster", "default_1600")
+	bodyWriter.WriteField("ocrCluster", "highPrecision")
 
 	if actionOptions.ScreenShotWithOCRCluster != "" {
 		bodyWriter.WriteField("ocrCluster", actionOptions.ScreenShotWithOCRCluster)