Update utils/functions.py

utils/functions.py

@@ -1,5 +1,3 @@
-# utils/functions.py
-
 import phonenumbers
 from phonenumbers import geocoder, carrier
 import re
@@ -146,7 +144,6 @@ def simple_checks(message, language):
     Przeprowadza proste sprawdzenia heurystyczne wiadomości SMS.
     """
     warnings = []
-    # Baza słów kluczowych (polski, niemiecki, angielski)
     scam_keywords = {
         'Polish': ['pieniądze', 'przelew', 'hasło', 'kod', 'nagroda', 'wygrana', 'pilne', 'pomoc', 'opłata', 'bank', 'karta', 'konto', 'logowanie', 'transakcja', 'weryfikacja', 'dane osobowe', 'szybka płatność', 'blokada konta', 'powiadomienie'],
         'German': ['Geld', 'Überweisung', 'Passwort', 'Code', 'Preis', 'Gewinn', 'dringend', 'Hilfe', 'Gebühr', 'Bank', 'Karte', 'Konto', 'Anmeldung', 'Transaktion', 'Verifizierung', 'persönliche Daten', 'schnelle Zahlung', 'Kontosperrung', 'Benachrichtigung'],
@@ -154,7 +151,6 @@ def simple_checks(message, language):
     }
 
     selected_keywords = scam_keywords.get(language, scam_keywords['English'])
-
     message_lower = message.lower()
 
     if any(keyword.lower() in message_lower for keyword in selected_keywords):
@@ -286,7 +282,6 @@ Provide your analysis and conclusions following the guidelines above."""
         if response.status_code == 200:
             data = response.json()
             ai_response = data['choices'][0]['message']['content']
-            # Parsowanie odpowiedzi
             analysis = re.search(r'<analysis>(.*?)</analysis>', ai_response, re.DOTALL)
             risk_assessment = re.search(r'<risk_assessment>(.*?)</risk_assessment>', ai_response, re.DOTALL)
             recommendations = re.search(r'<recommendations>(.*?)</recommendations>', ai_response, re.DOTALL)
@@ -302,3 +297,15 @@ Provide your analysis and conclusions following the guidelines above."""
     except Exception as e:
         logging.error(f"Błąd połączenia z API: {e}")
         return f"Błąd połączenia z API: {e}", "Błąd analizy.", "Błąd analizy."
+
+def get_email_info(email):
+    """
+    Sprawdza informacje o nadawcy e-maila (np. domena, organizacja, kraj).
+    """
+    domain = email.split('@')[-1]  # Prosta ekstrakcja domeny
+    # Możesz dodać więcej logiki do weryfikacji domeny
+    return {
+        "domain": domain,
+        "organization": "Nieznana organizacja",  # Możesz dodać logikę, aby zidentyfikować organizację
+        "country": "Nieznany kraj"  # Możesz dodać logikę, aby zidentyfikować kraj
+    }