refactor: TabOcr_B30

ui/extends/ocr/dependencies.py

@@ -0,0 +1,28 @@
+import cv2
+from arcaea_offline_ocr.phash_db import ImagePhashDatabase
+
+
+def getCv2StatModelStatusText(model: cv2.ml.StatModel):
+    if not isinstance(model, cv2.ml.StatModel):
+        return '<font color="red">ERROR</font>'
+
+    varCount = model.getVarCount()
+    if varCount != 81:
+        return f'<font color="darkorange">WARN</font>, varCount {varCount}'
+    else:
+        return f'<font color="green">OK</font>, varCount {varCount}'
+
+
+def getPhashDatabaseStatusText(db: ImagePhashDatabase):
+    if not isinstance(db, ImagePhashDatabase):
+        return '<font color="red">ERROR</font>'
+
+    jacketCount = len(db.jacket_hashes)
+    partnerIconCount = len(db.partner_icon_hashes)
+
+    statusText = f"J{jacketCount} PI{partnerIconCount}"
+
+    if partnerIconCount <= 0:
+        return f'<font color="darkorange">WARN</font>, {statusText}'
+    else:
+        return f'<font color="green">OK</font>, {statusText}'