Update app/services/classifier.py

app/services/classifier.py

@@ -22,7 +22,7 @@ LLM_MAX_CHARS = 3000
 
 _nsfw_runtime: dict[str, Any] | None = None
 _nsfw_error: str | None = None
-_ocr_reader: Any | None = None
+_ocr_reader: dict[str, Any] | None = None
 _ocr_error: str | None = None
 
 logger = logging.getLogger(__name__)
@@ -189,14 +189,27 @@ def classify_game_with_ocr_llm(file_path: str, filename: str, suspected_game: bo
 
 
 def extract_ocr_text_and_urls(file_path: str) -> tuple[str, list[str]]:
-    reader = _load_ocr_reader()
-    if reader is None:
+    runtime = _load_ocr_reader()
+    if runtime is None:
         logger.warning("ocr unavailable: reader is None")
         return "", []
 
     try:
-        results = reader.readtext(file_path, detail=0, paragraph=True)
-        ocr_text = "\n".join(str(x) for x in results).strip()
+        backend = runtime["backend"]
+        reader = runtime["reader"]
+        if backend == "easyocr":
+            results = reader.readtext(file_path, detail=0, paragraph=True)
+            ocr_text = "\n".join(str(x) for x in results).strip()
+        elif backend == "rapidocr":
+            results, _elapsed = reader(file_path)
+            if not results:
+                ocr_text = ""
+            else:
+                # RapidOCR result item: [box, text, score]
+                ocr_text = "\n".join(str(item[1]) for item in results if len(item) > 1).strip()
+        else:
+            logger.warning("ocr unknown backend=%s", backend)
+            return "", []
     except Exception:
         logger.exception("ocr read failed for file=%s", file_path)
         return "", []
@@ -234,15 +247,32 @@ def _load_ocr_reader() -> Any | None:
     if _ocr_error is not None:
         return None
 
+    easyocr_error: str | None = None
     try:
         import easyocr
 
-        _ocr_reader = easyocr.Reader(["en"], gpu=False)
-        logger.info("ocr reader initialized")
+        _ocr_reader = {
+            "backend": "easyocr",
+            "reader": easyocr.Reader(["en"], gpu=False),
+        }
+        logger.info("ocr reader initialized backend=easyocr")
+        return _ocr_reader
+    except Exception as exc:
+        easyocr_error = str(exc)
+        logger.warning("ocr easyocr init failed: %s", easyocr_error)
+
+    try:
+        from rapidocr_onnxruntime import RapidOCR
+
+        _ocr_reader = {
+            "backend": "rapidocr",
+            "reader": RapidOCR(),
+        }
+        logger.info("ocr reader initialized backend=rapidocr")
         return _ocr_reader
     except Exception as exc:
-        _ocr_error = str(exc)
-        logger.warning("ocr reader init failed: %s", _ocr_error)
+        _ocr_error = f"easyocr={easyocr_error}; rapidocr={exc}"
+        logger.warning("ocr reader init failed all backends: %s", _ocr_error)
         return None