Update app.py

app.py

@@ -8,8 +8,8 @@ from collections import defaultdict
 import re
 
 def ocr_with_boxes(image):
-    # แปลง PIL → NumPy และกลับสี RGB→BGR แบบเร็ว
-    img = np.array(image)[:, :, ::-1]
+    # แปลง PIL → RGB → NumPy → BGR สำหรับ OpenCV
+    img = np.array(image.convert('RGB'))[:, :, ::-1].copy()  # BGR uint8
 
     # config ไทย + อังกฤษ
     custom_config = r'-l tha+eng --oem 1 --psm 6'
@@ -43,11 +43,11 @@ def ocr_with_boxes(image):
         cv2.putText(img, " ".join(lines[line_id]), (x_min, y_min-5),
                     cv2.FONT_HERSHEY_SIMPLEX, 0.7, (0, 0, 255), 2, cv2.LINE_AA)
 
-    # รวมข้อความเป็นบรรทัด
+    # รวมข้อความ
     extracted_texts = [" ".join(words) for _, words in sorted(lines.items())]
     final_text = "\n".join(extracted_texts)
 
-    # 🔧 แก้เว้นวรรคภาษาไทย
+    # แก้เว้นวรรคภาษาไทย
     final_text = re.sub(r'([\u0E00-\u0E7F])\s+([\u0E00-\u0E7F])', r'\1\2', final_text)
 
     # แปลงกลับเป็น PIL
@@ -61,7 +61,7 @@ demo = gr.Interface(
     fn=ocr_with_boxes,
     inputs=gr.Image(type="pil", label="อัปโหลดภาพ"),
     outputs=[gr.Image(label="ผลลัพธ์พร้อมกรอบข้อความ"), gr.Textbox(label="ข้อความ OCR")],
-    title="OCR ไทย + อังกฤษ (Tesseract + Faster)",
+    title="OCR ไทย + อังกฤษ (Tesseract + Faster, Fixed OpenCV Error)",
     description="OCR ไทย+อังกฤษ วาดกรอบบรรทัดและแก้เว้นวรรคภาษาไทยอัตโนมัติ"
 )