Update utils/functions.py

utils/functions.py

@@ -6,8 +6,8 @@ import os
 from datetime import datetime
 import logging
 import json
-
-import pycountry  # Upewnij się, że zainstalowałeś tę bibliotekę: pip install pycountry
+from PIL import Image  # Upewnij się, że zainstalowałeś tę bibliotekę: pip install pillow
+import pytesseract  # Upewnij się, że zainstalowałeś tę bibliotekę: pip install pytesseract
 
 # Konfiguracja logowania
 logging.basicConfig(
@@ -370,6 +370,18 @@ def analyze_url(url):
 
     return phishing_urls  # Zwraca listę zagrożonych URL
 
+def extract_text_from_image(image_file):
+    """
+    Ekstrahuje tekst z przesłanego obrazu przy użyciu pytesseract.
+    """
+    try:
+        image = Image.open(image_file)
+        text = pytesseract.image_to_string(image)
+        return text
+    except Exception as e:
+        logging.error(f"Błąd podczas ekstrakcji tekstu z obrazu: {e}")
+        return "Błąd podczas ekstrakcji tekstu."
+        
 def get_email_info(email):
     """
     Sprawdza informacje o nadawcy e-maila (np. domena, organizacja, kraj).