Merge pull request #63 from SpenserCai/dev

Dev

intersvc/ocr_predict_inter.go

@@ -0,0 +1,45 @@
+package intersvc
+
+import (
+	"fmt"
+	"reflect"
+
+	webui "github.com/SpenserCai/sd-webui-go"
+	SdApiOperation "github.com/SpenserCai/sd-webui-go/stablediffusion/client/operations"
+)
+
+// API Path: /ocr/predict
+type OcrPredict struct {
+	RequestItem  *OcrPredictRequest
+	ResponseItem *OcrPredictResponse
+	Error        error
+}
+
+func (d *OcrPredict) Action(inter *webui.StableDiffInterface) {
+	RequestData := SdApiOperation.NewPredictOcrPredictPostParams()
+	RequestData.Body = d.RequestItem
+	ResponseData, err := inter.Client.Operations.PredictOcrPredictPost(RequestData)
+	if err != nil {
+		if reflect.TypeOf(err) == reflect.TypeOf(error(nil)) {
+			d.Error = err
+			return
+		}
+		errorValue := reflect.ValueOf(err).Elem().FieldByName("Payload")
+		if !errorValue.IsValid() {
+			d.Error = err
+			return
+		}
+		d.Error = fmt.Errorf("%v", errorValue.Elem())
+		return
+	}
+	ResponseItem, err := ConvertResponse(ResponseData.Payload, &OcrPredictResponse{})
+	if err != nil {
+		d.Error = err
+		return
+	}
+	d.ResponseItem = ResponseItem.(*OcrPredictResponse)
+}
+
+func (d *OcrPredict) GetResponse() *OcrPredictResponse {
+	return d.ResponseItem
+}

intersvc/ocr_predict_model.go

@@ -0,0 +1,17 @@
+/*
+ * @Author: SpenserCai
+ * @Date: 2023-12-18 10:17:08
+ * @version:
+ * @LastEditors: SpenserCai
+ * @LastEditTime: 2023-12-18 10:20:17
+ * @Description: file content
+ */
+package intersvc
+
+import (
+	SdApiModel "github.com/SpenserCai/sd-webui-go/stablediffusion/models"
+)
+
+type OcrPredictRequest = SdApiModel.OcrPredictRequest
+
+type OcrPredictResponse = SdApiModel.OcrPredictResponse