Update utils/functions.py

utils/functions.py

@@ -6,27 +6,26 @@ import os
 from datetime import datetime
 import logging
 import json
-from PIL import Image  # Make sure you have installed this library: pip install pillow
-import pytesseract  # Make sure you have installed this library: pip install pytesseract
+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
 
-# Configure logging
+# Konfiguracja logowania
 logging.basicConfig(
     filename='app.log',
     level=logging.INFO,
     format='%(asctime)s %(levelname)s:%(message)s'
 )
 
-# Define paths to JSON files
+# Definiowanie ścieżek do plików JSON
 BASE_DIR = os.path.dirname(os.path.abspath(__file__))
 DATA_DIR = os.path.join(BASE_DIR, '..', 'data')
 FAKE_NUMBERS_FILE = os.path.join(DATA_DIR, 'fake_numbers.json')
 HISTORY_FILE = os.path.join(DATA_DIR, 'history.json')
 STATS_FILE = os.path.join(DATA_DIR, 'stats.json')
 
-# Utility functions
-
+# Funkcje pomocnicze
 def load_json(file_path):
-    """Loads data from a JSON file. Returns an empty list or default object if the file does not exist."""
+    """Ładuje dane z pliku JSON. Jeśli plik nie istnieje, zwraca pustą listę lub domyślny obiekt."""
     if not os.path.exists(file_path):
         if file_path.endswith('stats.json'):
             return {"total_analyses": 0, "total_frauds_detected": 0}
@@ -37,43 +36,43 @@ def load_json(file_path):
             data = json.load(file)
             return data
         except json.JSONDecodeError:
-            logging.error(f"Cannot load data from {file_path}. The file is corrupted.")
+            logging.error(f"Nie można załadować danych z {file_path}. Plik jest uszkodzony.")
             if file_path.endswith('stats.json'):
                 return {"total_analyses": 0, "total_frauds_detected": 0}
             return []
 
 def save_json(file_path, data):
-    """Saves data to a JSON file."""
+    """Zapisuje dane do pliku JSON."""
     with open(file_path, 'w', encoding='utf-8') as file:
         json.dump(data, file, ensure_ascii=False, indent=4)
-        logging.info(f"Data has been saved to {file_path}.")
+        logging.info(f"Dane zostały zapisane do {file_path}.")
 
 def add_fake_number(phone_number):
-    """Adds a phone number to the fake_numbers.json file as fake if it does not already exist."""
+    """Dodaje numer telefonu do pliku fake_numbers.json jako fałszywy, jeśli jeszcze go tam nie ma."""
     fake_numbers = load_json(FAKE_NUMBERS_FILE)
     if phone_number not in fake_numbers:
         fake_numbers.append(phone_number)
         save_json(FAKE_NUMBERS_FILE, fake_numbers)
-        logging.info(f"Number {phone_number} has been successfully added to fake_numbers.json.")
+        logging.info(f"Numer {phone_number} został pomyślnie dodany do fake_numbers.json.")
         return True
     else:
-        logging.info(f"Number {phone_number} already exists in fake_numbers.json.")
+        logging.info(f"Numer {phone_number} już istnieje w fake_numbers.json.")
         return False
 
 def is_fake_number(phone_number):
-    """Checks if a given phone number is marked as fake in the fake_numbers.json file."""
+    """Sprawdza, czy dany numer telefonu jest oznaczony jako fałszywy w pliku fake_numbers.json."""
     fake_numbers = load_json(FAKE_NUMBERS_FILE)
     exists = phone_number in fake_numbers
-    logging.info(f"Checking number {phone_number}: {'found' if exists else 'not found'}.")
+    logging.info(f"Sprawdzanie numeru {phone_number}: {'znaleziony' if exists else 'nie znaleziony'}.")
     return exists
 
 def get_fake_numbers():
-    """Retrieves the list of fake numbers from the fake_numbers.json file."""
+    """Pobiera listę fałszywych numerów z pliku fake_numbers.json."""
     fake_numbers = load_json(FAKE_NUMBERS_FILE)
     return fake_numbers
 
 def add_to_history(message, phone_number, analysis, risk, recommendations):
-    """Adds an entry to the analysis history in history.json."""
+    """Dodaje wpis do historii analiz w pliku history.json."""
     history = load_json(HISTORY_FILE)
     history.append({
         "timestamp": datetime.now().isoformat(),
@@ -84,46 +83,52 @@ def add_to_history(message, phone_number, analysis, risk, recommendations):
         "recommendations": recommendations
     })
     save_json(HISTORY_FILE, history)
-    logging.info(f"Added entry to history.json for number {phone_number}.")
+    logging.info(f"Dodano wpis do history.json dla numeru {phone_number}.")
+
+def get_history():
+    """Pobiera historię analiz z pliku history.json jako listę słowników."""
+    history = load_json(HISTORY_FILE)
+    logging.info("Historia analiz została pobrana pomyślnie.")
+    return history
 
 def update_stats(fraud_detected=False):
-    """Updates analysis statistics in stats.json."""
+    """Aktualizuje statystyki analiz w pliku stats.json."""
     stats = load_json(STATS_FILE)
     stats["total_analyses"] += 1
     if fraud_detected:
         stats["total_frauds_detected"] += 1
     save_json(STATS_FILE, stats)
-    logging.info(f"Statistics have been updated: Analyses {stats['total_analyses']}, Frauds {stats['total_frauds_detected']}.")
+    logging.info(f"Statystyki zostały zaktualizowane: Analiz {stats['total_analyses']}, Oszustw {stats['total_frauds_detected']}.")
 
 def get_stats():
-    """Retrieves analysis statistics from stats.json."""
+    """Pobiera statystyki analiz z pliku stats.json."""
     stats = load_json(STATS_FILE)
-    logging.info("Statistics have been successfully retrieved.")
+    logging.info("Statystyki zostały pobrane pomyślnie.")
     return stats
 
 def get_statistics():
-    """Returns detailed analysis statistics from stats.json."""
+    """Zwraca szczegółowe statystyki analiz z pliku stats.json."""
     stats = load_json(STATS_FILE)
     return stats
 
 def get_phone_info(phone_number):
-    """Verifies the phone number and returns information about the country and operator."""
+    """Weryfikuje numer telefonu i zwraca informacje o kraju i operatorze."""
     try:
         parsed_number = phonenumbers.parse(phone_number, None)
-        country = geocoder.description_for_number(parsed_number, 'pl')  # Change to 'pl' for Polish
-        operator = carrier.name_for_number(parsed_number, 'pl')  # Change to 'pl' for Polish
+        country = geocoder.description_for_number(parsed_number, 'pl')  # Zmiana na 'pl' dla polskiego
+        operator = carrier.name_for_number(parsed_number, 'pl')  # Zmiana na 'pl' dla polskiego
         if not country:
-            country = "Unknown"
+            country = "Nieznany"
         if not operator:
-            operator = "Unknown"
-        logging.info(f"Number {phone_number} - Country: {country}, Operator: {operator}.")
+            operator = "Nieznany"
+        logging.info(f"Numer {phone_number} - Kraj: {country}, Operator: {operator}.")
         return country, operator
     except phonenumbers.NumberParseException as e:
-        logging.error(f"Could not process phone number {phone_number}: {e}")
-        return "Unknown", "Unknown"
+        logging.error(f"Nie udało się przetworzyć numeru telefonu {phone_number}: {e}")
+        return "Nieznany", "Nieznany"
 
 def simple_checks(message, language):
-    """Performs simple heuristic checks on SMS messages."""
+    """Przeprowadza proste sprawdzenia heurystyczne wiadomości SMS."""
     warnings = []
     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'],
@@ -135,20 +140,20 @@ def simple_checks(message, language):
     message_lower = message.lower()
 
     if any(keyword.lower() in message_lower for keyword in selected_keywords):
-        warnings.append("The message contains keywords associated with potential fraud.")
+        warnings.append("Wiadomość zawiera słowa kluczowe związane z potencjalnym oszustwem.")
     if re.search(r'http[s]?://', message):
-        warnings.append("The message contains a link.")
+        warnings.append("Wiadomość zawiera link.")
     if re.search(r'\b(podaj|prześlij|udostępnij|sende|übermittle|teile|send|provide|share)\b.*\b(hasło|kod|dane osobowe|numer konta|Passwort|Code|persönliche Daten|Kontonummer|password|code|personal information|account number)\b', message_lower):
-        warnings.append("The message contains a request for confidential information.")
+        warnings.append("Wiadomość zawiera prośbę o poufne informacje.")
     return warnings
 
 def analyze_message(message, phone_number, additional_info, api_key, language):
-    """Analyzes the SMS message using the SambaNova API."""
+    """Analizuje wiadomość SMS za pomocą API SambaNova."""
     if not api_key:
-        logging.error("Missing API key.")
-        return "Missing API key.", "Missing API key.", "Missing API key."
+        logging.error("Brak klucza API.")
+        return "Brak klucza API.", "Brak klucza API.", "Brak klucza API."
 
-    url = "https://api.sambanova.ai/v1/chat/completions"  # Make sure this is the correct API URL
+    url = "https://api.sambanova.ai/v1/chat/completions"  # Upewnij się, że to poprawny URL
     headers = {
         "Authorization": f"Bearer {api_key}",
         "Content-Type": "application/json"
@@ -280,10 +285,12 @@ You are an advanced AI assistant specializing in identifying fake SMS messages.
 - Do you have access to other tools that can help you assess this message?
 - What actions can you take to enhance your security in the future?
 </recommendations>
+
+Your response should be formatted exactly as specified above, using the <analysis>, <risk_assessment>, and <recommendations> tags. Ensure that each section is thoroughly and comprehensively filled out.
         """
     }
 
-    system_prompt = system_prompts.get(language, system_prompts['English'])  # Default to English if language is unsupported
+    system_prompt = system_prompts.get(language, system_prompts['English'])  # Domyślnie angielski, jeśli język nie jest obsługiwany
 
     user_prompt = f"""Analyze the following message for potential fraud:
 
@@ -296,7 +303,7 @@ Additional Information:
 Provide your analysis and conclusions following the guidelines above."""
 
     payload = {
-        "model": "Meta-Llama-3.1-8B-Instruct",  # Ensure this is the correct API model
+        "model": "Meta-Llama-3.1-8B-Instruct",  # Upewnij się, że to poprawny model API
         "messages": [
             {"role": "system", "content": system_prompt},
             {"role": "user", "content": user_prompt}