...

LunaTranslator/LunaTranslator/ocrengines/googlecloudvision.py

@@ -0,0 +1,47 @@
+from ocrengines.baseocrclass import baseocr
+import base64
+
+
+class OCR(baseocr):
+
+    def ocr(self, imgfile):
+        # https://github.com/dmotz/thing-translator/blob/d1fec3f38d24e973af49766669f9ee00bd9e98a8/src/effects/snap.js
+        # https://cloud.google.com/vision/docs/ocr?hl=zh-cn
+        # https://cloud.google.com/vision/docs/reference/rest/v1/AnnotateImageResponse#EntityAnnotation
+        self.checkempty(["googlecloudvision"])
+        ocr_url = (
+            "https://vision.googleapis.com/v1/images:annotate?key=" + self.config["key"]
+        )
+        with open(imgfile, "rb") as f:
+            data = f.read()
+            encodestr = str(base64.b64encode(data), "utf-8")
+        data = {
+            "requests": [
+                {
+                    "image": {"content": encodestr},
+                    "features": [{"type": "TEXT_DETECTION"}],
+                }
+            ]
+        }
+        response = self.session.post(ocr_url, json=data)
+        try:
+            boxs = []
+            texts = []
+            for anno in response.json()["responses"][0]["textAnnotations"]:
+
+                texts.append(anno["description"])
+                boxs.append(
+                    [
+                        anno["boundingPoly"]["vertices"][0]["x"],
+                        anno["boundingPoly"]["vertices"][0]["y"],
+                        anno["boundingPoly"]["vertices"][1]["x"],
+                        anno["boundingPoly"]["vertices"][1]["y"],
+                        anno["boundingPoly"]["vertices"][2]["x"],
+                        anno["boundingPoly"]["vertices"][2]["y"],
+                        anno["boundingPoly"]["vertices"][3]["x"],
+                        anno["boundingPoly"]["vertices"][3]["y"],
+                    ]
+                )
+            return self.common_solve_text_orientation(boxs, texts)
+        except:
+            raise Exception(response.text)

LunaTranslator/files/defaultconfig/config.json

@@ -787,6 +787,10 @@
             "use": false,
             "name": "docsumo"
         },
+        "googlecloudvision": {
+            "use": false,
+            "name": "googlecloudvision"
+        },
         "youdaocr": {
             "use": false,
             "name": "有道OCR"

LunaTranslator/files/defaultconfig/ocrsetting.json

@@ -212,6 +212,11 @@
             }
         }
     },
+	"googlecloudvision": {
+		"args": {
+			"key": ""
+		}
+	},
     "mangaocr": {
         "args": {
             "项目仓库": "https://github.com/kha-white/manga-ocr",