Create modules/ocr.py

modules/ocr.py

@@ -0,0 +1,42 @@
+from paddleocr import PaddleOCR
+import cv2
+import numpy as np
+import os
+
+# تهيئة محرك OCR (يتم تهيئته عند الاستيراد لأداء أفضل)
+ocr_engine = None
+
+def get_ocr_engine(lang='en'):
+    """الحصول على أو إنشاء محرك OCR"""
+    global ocr_engine
+    if ocr_engine is None:
+        try:
+            ocr_engine = PaddleOCR(use_angle_cls=True, lang=lang, show_log=False)
+            print("✅ تم تهيئة محرك PaddleOCR بنجاح")
+        except Exception as e:
+            print(f"❌ فشل في تهيئة PaddleOCR: {e}")
+            raise
+    return ocr_engine
+
+def extract_texts(image_path: str):
+    """استخراج النصوص والمربعات المحيطة بها من الصورة"""
+    try:
+        ocr = get_ocr_engine('en')
+        results = ocr.ocr(image_path, cls=True)
+        
+        texts = []
+        boxes = []
+        
+        if results and results[0] is not None:
+            for line in results[0]:
+                if line and len(line) >= 2:
+                    text = line[1][0]
+                    box = line[0]
+                    texts.append(text)
+                    boxes.append(box)
+        
+        return texts, boxes
+        
+    except Exception as e:
+        print(f"❌ خطأ في استخراج النصوص: {e}")
+        return [], []