Fix HF runtime crash: lazy OCR init + disable SSR

app.py

@@ -1,54 +1,51 @@
+import gradio as gr
+import numpy as np
+from paddleocr import PaddleOCR
+
+ocr = None  # lazy init
+
+def get_ocr():
+    global ocr
+    if ocr is None:
+        ocr = PaddleOCR(use_angle_cls=True, lang="en")
+    return ocr
+
 def run_ocr(img):
     if img is None:
-        return "", 0.0
+        return "(No image)", 0.0
 
     img_np = np.array(img)
 
-    result = ocr.ocr(img_np)  # no cls arg
+    ocr_engine = get_ocr()
+    result = ocr_engine.ocr(img_np)
 
     lines = []
     confs = []
 
-    # result can be:
-    #  - [ [ [box, (text, conf)], ... ] ]  (nested)
-    #  - [ [box, (text, conf)], ... ]      (less nested)
-    #  - other variants depending on version
-    blocks = result
-    if isinstance(result, list) and len(result) == 1 and isinstance(result[0], list):
-        # sometimes it's wrapped one extra level
-        blocks = result[0]
+    blocks = result[0] if isinstance(result, list) and result and isinstance(result[0], list) else result
 
     for item in blocks:
-        if not item:
-            continue
-
-        # item might be: [box, (text, conf)]  OR  [[box], [text, conf]] etc.
-        # Try common case first
         try:
-            maybe = item[1]
-            if isinstance(maybe, (list, tuple)) and len(maybe) >= 2:
-                text, conf = maybe[0], float(maybe[1])
-                if text:
-                    lines.append(str(text))
-                    confs.append(conf)
-                continue
+            text, conf = item[1]
+            lines.append(text)
+            confs.append(float(conf))
         except Exception:
-            pass
-
-        # Fallback: search inside item for a (text, conf) pair
-        found = False
-        for part in item:
-            if isinstance(part, (list, tuple)) and len(part) >= 2:
-                # part could be (text, conf) or [text, conf]
-                if isinstance(part[0], str) and isinstance(part[1], (float, int)):
-                    lines.append(part[0])
-                    confs.append(float(part[1]))
-                    found = True
-                    break
-        if found:
             continue
 
     extracted = "\n".join(lines).strip()
-    avg_conf = float(sum(confs) / len(confs)) if confs else 0.0
+    avg_conf = sum(confs) / len(confs) if confs else 0.0
 
     return extracted if extracted else "(No text detected)", avg_conf
+
+demo = gr.Interface(
+    fn=run_ocr,
+    inputs=gr.Image(type="pil", label="Upload a page photo"),
+    outputs=[
+        gr.Textbox(label="Extracted text", lines=12),
+        gr.Number(label="Average confidence (0–1)")
+    ],
+    title="BookReader × Reachy Mini",
+    description="CPU-based PaddleOCR for physical book pages.",
+)
+
+demo.launch(ssr_mode=False)