Create modules/vision.py

modules/vision.py

@@ -0,0 +1,23 @@
+import cv2
+import easyocr
+import numpy as np
+
+# Inicializa o leitor apenas uma vez (é pesado)
+reader = easyocr.Reader(['en'], gpu=False) # GPU=False para garantir compatibilidade no HF CPU
+
+def analyze_image_text(uploaded_image):
+    """Lê texto de IHMs ou placas de identificação."""
+    try:
+        # Converte o arquivo do streamlit para formato OpenCV
+        file_bytes = np.asarray(bytearray(uploaded_image.read()), dtype=np.uint8)
+        img = cv2.imdecode(file_bytes, 1)
+        
+        # Realiza o OCR
+        results = reader.readtext(img, detail=0)
+        
+        if not results:
+            return "Nenhum texto legível detectado."
+        
+        return "\n".join(results)
+    except Exception as e:
+        return f"Erro no processamento óptico: {e}"