Update app.py

app.py

@@ -6,7 +6,8 @@ from paddleocr import PaddleOCR
 
 ocr = PaddleOCR(
     use_angle_cls=True,
-    lang="en"
+    lang="en",
+    use_gpu=False
 )
 
 
@@ -20,31 +21,37 @@ def extract_descriptions(image: Image.Image):
 
     words = []
 
-    # 🔴 PARSING ROBUSTE PaddleOCR
     for line in result[0]:
-        # Cas 1 : [box, (text, score)]
-        if len(line) >= 2 and isinstance(line[1], (list, tuple)):
-            box = line[0]
-            text = line[1][0]
-            score = line[1][1]
-
-        # Cas 2 : [box, text, score]
-        elif len(line) >= 3:
-            box = line[0]
-            text = line[1]
-            score = line[2]
-
-        else:
+        text, score, box = None, None, None
+
+        # Parsing défensif PaddleOCR
+        if isinstance(line, (list, tuple)):
+            if len(line) >= 2 and isinstance(line[1], (list, tuple)):
+                box = line[0]
+                text = line[1][0]
+                score = line[1][1]
+            elif len(line) >= 3:
+                box = line[0]
+                text = line[1]
+                score = line[2]
+
+        if box is None or text is None:
             continue
 
-        if score < 0.5 or not text.strip():
+        # 🔒 Sécurisation du score
+        try:
+            score = float(score)
+        except:
+            score = 1.0
+
+        if score < 0.5 or not str(text).strip():
             continue
 
         xs = [p[0] for p in box]
         ys = [p[1] for p in box]
 
         words.append({
-            "text": text.strip(),
+            "text": str(text).strip(),
             "x": min(xs),
             "y": min(ys),
             "w": max(xs) - min(xs),
@@ -60,7 +67,7 @@ def extract_descriptions(image: Image.Image):
     if header is None:
         return "❌ Colonne 'Description' non détectée."
 
-    # 2️⃣ Définir la zone de la colonne
+    # 2️⃣ Zone de la colonne
     x_min = header["x"] - 15
     x_max = header["x"] + header["w"] + 380
     y_min = header["y"] + header["h"] + 10
@@ -112,7 +119,7 @@ demo = gr.Interface(
     fn=extract_descriptions,
     inputs=gr.Image(type="pil"),
     outputs=gr.Textbox(lines=20),
-    title="Extraction colonne Description – PaddleOCR (Stable)"
+    title="Extraction colonne Description – PaddleOCR (Production Safe)"
 )
 
 demo.launch(server_name="0.0.0.0", server_port=7860)