Update utils/functions.py

utils/functions.py

@@ -1,3 +1,5 @@
+# utils/functions.py
+
 import phonenumbers
 from phonenumbers import geocoder, carrier
 import re
@@ -8,55 +10,68 @@ from datetime import datetime
 import logging
 
 # Konfiguracja logowania
-logging.basicConfig(filename='app.log', level=logging.ERROR, format='%(asctime)s %(levelname)s:%(message)s')
+logging.basicConfig(filename='app.log', level=logging.INFO, format='%(asctime)s %(levelname)s:%(message)s')
+
+# Inicjalizacja ścieżki bazowej
+BASE_DIR = os.path.dirname(os.path.abspath(__file__))
 
 # Ścieżka do pliku JSON przechowującego fałszywe numery
-FAKE_NUMBERS_FILE = 'fake_numbers.json'
+FAKE_NUMBERS_FILE = os.path.join(BASE_DIR, '..', '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)
+        with open(FAKE_NUMBERS_FILE, 'w', encoding='utf-8') as f:
+            json.dump([], f, ensure_ascii=False, indent=4)
+        logging.info(f"Utworzono nowy plik {FAKE_NUMBERS_FILE}.")
     else:
         try:
-            with open(FAKE_NUMBERS_FILE, 'r') as f:
+            with open(FAKE_NUMBERS_FILE, 'r', encoding='utf-8') as f:
                 json.load(f)
+            logging.info(f"Plik {FAKE_NUMBERS_FILE} został załadowany pomyślnie.")
         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)
+            with open(FAKE_NUMBERS_FILE, 'w', encoding='utf-8') as f:
+                json.dump([], f, ensure_ascii=False, indent=4)
+            logging.warning(f"Plik {FAKE_NUMBERS_FILE} był uszkodzony i został zresetowany.")
 
 # Dodanie numeru telefonu do pliku JSON
 def add_fake_number(phone_number):
+    logging.info(f"Próbuję dodać numer: {phone_number}")
     try:
-        with open(FAKE_NUMBERS_FILE, 'r') as f:
+        with open(FAKE_NUMBERS_FILE, 'r', encoding='utf-8') as f:
             fake_numbers = json.load(f)
     except (json.JSONDecodeError, FileNotFoundError):
         fake_numbers = []
-    
+        logging.warning(f"{FAKE_NUMBERS_FILE} jest pusty lub uszkodzony. Inicjalizuję nową listę.")
+
     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)
+            with open(FAKE_NUMBERS_FILE, 'w', encoding='utf-8') as f:
+                json.dump(fake_numbers, f, ensure_ascii=False, indent=4)
+            logging.info(f"Numer {phone_number} został pomyślnie dodany do fake_numbers.json.")
             return True
         except Exception as e:
             logging.error(f"Nie udało się zapisać numeru {phone_number}: {e}")
             return False
     else:
+        logging.info(f"Numer {phone_number} już istnieje w fake_numbers.json.")
         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:
+        with open(FAKE_NUMBERS_FILE, 'r', encoding='utf-8') as f:
             fake_numbers = json.load(f)
-        return any(entry["phone_number"] == phone_number for entry in fake_numbers)
+        exists = any(entry["phone_number"] == phone_number for entry in fake_numbers)
+        logging.info(f"Sprawdzanie numeru {phone_number}: {'znaleziony' if exists else 'nie znaleziony'}.")
+        return exists
     except (json.JSONDecodeError, FileNotFoundError):
+        logging.error(f"Nie udało się załadować {FAKE_NUMBERS_FILE}.")
         return False
 
 # Funkcja do weryfikacji numeru telefonu
@@ -65,8 +80,10 @@ def get_phone_info(phone_number):
         parsed_number = phonenumbers.parse(phone_number, None)
         country = geocoder.description_for_number(parsed_number, 'pl')
         operator = carrier.name_for_number(parsed_number, 'pl')
+        logging.info(f"Numer {phone_number} - Kraj: {country}, Operator: {operator}.")
         return country, operator
-    except phonenumbers.NumberParseException:
+    except phonenumbers.NumberParseException as e:
+        logging.error(f"Nie udało się przetworzyć numeru telefonu {phone_number}: {e}")
         return None, None
 
 # Proste sprawdzenia heurystyczne wiadomości
@@ -94,7 +111,7 @@ def analyze_message(message, phone_number, additional_info, api_key, language):
     if not 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"
     headers = {
         "Authorization": f"Bearer {api_key}"
@@ -180,7 +197,7 @@ Your response should be formatted exactly as specified above, using the <analysi
     }
 
     system_prompt = system_prompts.get(language, system_prompts['English'])  # Default to English if language not found
-    
+
     user_prompt = f"""Analyze the following message for potential fraud:
 
 Message: "{message}"
@@ -227,52 +244,62 @@ Provide your analysis and conclusions following the guidelines above."""
 
 # Inicjalizacja pliku statystyk
 def init_stats_file():
-    stats_file = 'stats.json'
+    stats_file = os.path.join(BASE_DIR, '..', '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)
+        with open(stats_file, 'w', encoding='utf-8') as f:
+            json.dump({"total_analyses": 0, "total_frauds_detected": 0}, f, ensure_ascii=False, indent=4)
+        logging.info(f"Utworzono nowy plik statystyk {stats_file}.")
 
 # Pobranie statystyk
 def get_stats():
-    stats_file = 'stats.json'
+    stats_file = os.path.join(BASE_DIR, '..', 'stats.json')
     try:
-        with open(stats_file, 'r') as f:
+        with open(stats_file, 'r', encoding='utf-8') as f:
             stats = json.load(f)
+        logging.info("Statystyki zostały pobrane pomyślnie.")
         return stats
-    except (json.JSONDecodeError, FileNotFoundError):
+    except (json.JSONDecodeError, FileNotFoundError) as e:
+        logging.error(f"Nie udało się załadować statystyk: {e}")
         return {"total_analyses": 0, "total_frauds_detected": 0}
 
 # Aktualizacja statystyk analizy
 def update_stats(fraud_detected=False):
-    stats_file = 'stats.json'
+    stats_file = os.path.join(BASE_DIR, '..', 'stats.json')
     try:
-        with open(stats_file, 'r') as f:
+        with open(stats_file, 'r', encoding='utf-8') as f:
             stats = json.load(f)
     except (json.JSONDecodeError, FileNotFoundError):
         stats = {"total_analyses": 0, "total_frauds_detected": 0}
+        logging.warning(f"{stats_file} jest pusty lub uszkodzony. Inicjalizuję nową listę.")
 
     stats["total_analyses"] += 1
     if fraud_detected:
         stats["total_frauds_detected"] += 1
 
-    with open(stats_file, 'w') as f:
-        json.dump(stats, f, indent=4)
+    try:
+        with open(stats_file, 'w', encoding='utf-8') as f:
+            json.dump(stats, f, ensure_ascii=False, indent=4)
+        logging.info(f"Statystyki zostały zaktualizowane: {stats}.")
+    except Exception as e:
+        logging.error(f"Nie udało się zaktualizować statystyk: {e}")
 
 # Inicjalizacja pliku historii analiz
 def init_history_file():
-    history_file = 'history.json'
+    history_file = os.path.join(BASE_DIR, '..', 'history.json')
     if not os.path.exists(history_file):
-        with open(history_file, 'w') as f:
-            json.dump([], f)
+        with open(history_file, 'w', encoding='utf-8') as f:
+            json.dump([], f, ensure_ascii=False, indent=4)
+        logging.info(f"Utworzono nowy plik historii analiz {history_file}.")
 
 # Dodanie wpisu do historii analiz
 def add_to_history(message, phone_number, analysis, risk, recommendations):
-    history_file = 'history.json'
+    history_file = os.path.join(BASE_DIR, '..', 'history.json')
     try:
-        with open(history_file, 'r') as f:
+        with open(history_file, 'r', encoding='utf-8') as f:
             history = json.load(f)
     except (json.JSONDecodeError, FileNotFoundError):
         history = []
+        logging.warning(f"{history_file} jest pusty lub uszkodzony. Inicjalizuję nową listę.")
 
     history.append({
         "timestamp": datetime.now().isoformat(),
@@ -283,16 +310,21 @@ def add_to_history(message, phone_number, analysis, risk, recommendations):
         "recommendations": recommendations
     })
 
-    with open(history_file, 'w') as f:
-        json.dump(history, f, indent=4)
+    try:
+        with open(history_file, 'w', encoding='utf-8') as f:
+            json.dump(history, f, ensure_ascii=False, indent=4)
+        logging.info(f"Dodano wpis do historii analiz dla numeru {phone_number}.")
+    except Exception as e:
+        logging.error(f"Nie udało się zapisać historii analiz dla numeru {phone_number}: {e}")
 
 # Pobranie historii analiz
 def get_history():
-    history_file = 'history.json'
+    history_file = os.path.join(BASE_DIR, '..', 'history.json')
     try:
-        with open(history_file, 'r') as f:
+        with open(history_file, 'r', encoding='utf-8') as f:
             history = json.load(f)
+        logging.info("Historia analiz została pobrana pomyślnie.")
         return history
-    except (json.JSONDecodeError, FileNotFoundError):
+    except (json.JSONDecodeError, FileNotFoundError) as e:
+        logging.error(f"Nie udało się załadować historii analiz: {e}")
         return []
-