Update utils/functions.py

utils/functions.py

@@ -8,10 +8,60 @@ import os
 import json
 from datetime import datetime
 
+# Ścieżka do pliku JSON przechowującego fałszywe numery
+FAKE_NUMBERS_FILE = 'fake_numbers.json'
+
+# Inicjalizacja pliku JSON przechowującego fałszywe numery
+def init_fake_numbers_file():
+    if not os.path.exists(FAKE_NUMBERS_FILE):
+        with open(FAKE_NUMBERS_FILE, 'w') as f:
+            json.dump([], f)
+    else:
+        # Sprawdzenie, czy plik nie jest pusty i zawiera prawidłowy JSON
+        try:
+            with open(FAKE_NUMBERS_FILE, 'r') as f:
+                json.load(f)
+        except json.JSONDecodeError:
+            # Jeśli plik jest uszkodzony lub pusty, zresetuj go do pustej listy
+            with open(FAKE_NUMBERS_FILE, 'w') as f:
+                json.dump([], f)
+
+# Dodanie numeru telefonu do pliku JSON
+def add_fake_number(phone_number):
+    try:
+        with open(FAKE_NUMBERS_FILE, 'r') as f:
+            fake_numbers = json.load(f)
+    except (json.JSONDecodeError, FileNotFoundError):
+        fake_numbers = []
+    
+    if not any(entry["phone_number"] == phone_number for entry in fake_numbers):
+        fake_numbers.append({
+            "phone_number": phone_number,
+            "reported_at": datetime.now().isoformat()
+        })
+        try:
+            with open(FAKE_NUMBERS_FILE, 'w') as f:
+                json.dump(fake_numbers, f, indent=4)
+            return True
+        except Exception as e:
+            return False
+    else:
+        return False  # Numer już istnieje
+
+# Sprawdzenie, czy numer telefonu jest w pliku JSON
+def is_fake_number(phone_number):
+    try:
+        with open(FAKE_NUMBERS_FILE, 'r') as f:
+            fake_numbers = json.load(f)
+        return any(entry["phone_number"] == phone_number for entry in fake_numbers)
+    except (json.JSONDecodeError, FileNotFoundError):
+        return False
+
+# Pobierz klucz API z zmiennej środowiskowej
+API_KEY = os.getenv('SAMBANOVA_API_KEY')
+
+# Funkcja do weryfikacji numeru telefonu
 def get_phone_info(phone_number):
-    """
-    Weryfikuje numer telefonu i zwraca kraj oraz operatora.
-    """
     try:
         parsed_number = phonenumbers.parse(phone_number, None)
         country = geocoder.description_for_number(parsed_number, 'pl')
@@ -20,28 +70,31 @@ def get_phone_info(phone_number):
     except phonenumbers.NumberParseException:
         return None, None
 
+# Proste sprawdzenia heurystyczne wiadomości
 def simple_checks(message):
-    """
-    Wykonuje podstawowe sprawdzenia heurystyczne wiadomości SMS.
-    """
     warnings = []
+    # Słowa kluczowe często używane w oszustwach
     scam_keywords = ['pieniądze', 'przelew', 'hasło', 'kod', 'nagroda', 'wygrana', 'pilne', 'pomoc', 'opłata']
     if any(keyword in message.lower() for keyword in scam_keywords):
         warnings.append("Wiadomość zawiera słowa kluczowe związane z potencjalnym oszustwem.")
+    # Sprawdzenie obecności linków
     if re.search(r'http[s]?://', message):
         warnings.append("Wiadomość zawiera link.")
+    # Sprawdzenie, czy nadawca prosi o poufne informacje
     if re.search(r'\b(podaj|prześlij|udostępnij)\b.*\b(hasło|kod|dane osobowe|numer konta)\b', message.lower()):
         warnings.append("Wiadomość zawiera prośbę o poufne informacje.")
     return warnings
 
-def analyze_message(message, phone_number, additional_info, api_key):
-    """
-    Analizuje wiadomość SMS za pomocą API SambaNova.
-    """
+# Funkcja do analizy wiadomości za pomocą API SambaNova z głębszym procesem myślenia
+def analyze_message(message, phone_number):
+    if not API_KEY:
+        return "Brak klucza API.", "Brak klucza API.", "Brak klucza API."
+    
     url = "https://api.sambanova.ai/v1/chat/completions"
     headers = {
-        "Authorization": f"Bearer {api_key}"
+        "Authorization": f"Bearer {API_KEY}"
     }
+    # Rozbudowany system prompt z głębszym procesem myślenia
     system_prompt = """
 Jesteś zaawansowanym asystentem AI specjalizującym się w identyfikacji fałszywych wiadomości SMS. Twoim zadaniem jest przeprowadzenie szczegółowej analizy wiadomości, wykorzystując głęboki proces myślenia i dostarczając kompleksową ocenę. Twoja odpowiedź powinna być podzielona na trzy sekcje:
 
@@ -73,9 +126,6 @@ Twoja odpowiedź powinna być sformatowana dokładnie w powyższy sposób, używ
 Wiadomość: "{message}"
 Numer telefonu nadawcy: {phone_number}"
 
-Dodatkowe informacje:
-{additional_info}
-
 Podaj swoją analizę i wnioski zgodnie z powyższymi wytycznymi."""
 
     payload = {
@@ -95,6 +145,7 @@ Podaj swoją analizę i wnioski zgodnie z powyższymi wytycznymi."""
         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)
@@ -109,19 +160,15 @@ Podaj swoją analizę i wnioski zgodnie z powyższymi wytycznymi."""
     except Exception as e:
         return f"Błąd połączenia z API: {e}", "Błąd analizy.", "Błąd analizy."
 
+# Inicjalizacja pliku statystyk
 def init_stats_file():
-    """
-    Inicjalizuje plik statystyk, jeśli nie istnieje.
-    """
     stats_file = 'stats.json'
     if not os.path.exists(stats_file):
         with open(stats_file, 'w') as f:
             json.dump({"total_analyses": 0, "total_frauds_detected": 0}, f)
 
+# Aktualizacja statystyk analizy
 def update_stats(fraud_detected=False):
-    """
-    Aktualizuje statystyki analizy.
-    """
     stats_file = 'stats.json'
     try:
         with open(stats_file, 'r') as f:
@@ -136,31 +183,15 @@ def update_stats(fraud_detected=False):
     with open(stats_file, 'w') as f:
         json.dump(stats, f, indent=4)
 
-def get_stats():
-    """
-    Pobiera aktualne statystyki analizy.
-    """
-    stats_file = 'stats.json'
-    try:
-        with open(stats_file, 'r') as f:
-            stats = json.load(f)
-        return stats
-    except (json.JSONDecodeError, FileNotFoundError):
-        return {"total_analyses": 0, "total_frauds_detected": 0}
-
+# Inicjalizacja pliku historii analiz
 def init_history_file():
-    """
-    Inicjalizuje plik historii analiz, jeśli nie istnieje.
-    """
     history_file = 'history.json'
     if not os.path.exists(history_file):
         with open(history_file, 'w') as f:
             json.dump([], f)
 
+# Dodanie wpisu do historii analiz
 def add_to_history(message, phone_number, analysis, risk, recommendations):
-    """
-    Dodaje wpis do historii analiz.
-    """
     history_file = 'history.json'
     try:
         with open(history_file, 'r') as f:
@@ -180,10 +211,8 @@ def add_to_history(message, phone_number, analysis, risk, recommendations):
     with open(history_file, 'w') as f:
         json.dump(history, f, indent=4)
 
+# Pobranie historii analiz
 def get_history():
-    """
-    Pobiera historię analiz.
-    """
     history_file = 'history.json'
     try:
         with open(history_file, 'r') as f: