Update utils/functions.py

utils/functions.py

@@ -7,6 +7,10 @@ import requests
 import os
 import json
 from datetime import datetime
+import logging
+
+# Konfiguracja logowania
+logging.basicConfig(filename='app.log', level=logging.ERROR, format='%(asctime)s %(levelname)s:%(message)s')
 
 # Ścieżka do pliku JSON przechowującego fałszywe numery
 FAKE_NUMBERS_FILE = 'fake_numbers.json'
@@ -44,6 +48,7 @@ def add_fake_number(phone_number):
                 json.dump(fake_numbers, f, indent=4)
             return True
         except Exception as e:
+            logging.error(f"Nie udało się zapisać numeru {phone_number}: {e}")
             return False
     else:
         return False  # Numer już istnieje
@@ -86,13 +91,14 @@ def simple_checks(message):
     return warnings
 
 # 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:
+def analyze_message(message, phone_number, additional_info, api_key):
+    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}"
+        "Authorization": f"Bearer {api_key}"
     }
     # Rozbudowany system prompt z głębszym procesem myślenia
     system_prompt = """
@@ -124,7 +130,10 @@ Twoja odpowiedź powinna być sformatowana dokładnie w powyższy sposób, używ
     user_prompt = f"""Przeanalizuj następującą wiadomość pod kątem potencjalnego oszustwa:
 
 Wiadomość: "{message}"
-Numer telefonu nadawcy: {phone_number}"
+Numer telefonu nadawcy: "{phone_number}"
+
+Dodatkowe informacje:
+{additional_info}
 
 Podaj swoją analizę i wnioski zgodnie z powyższymi wytycznymi."""
 
@@ -156,8 +165,10 @@ Podaj swoją analizę i wnioski zgodnie z powyższymi wytycznymi."""
 
             return analysis_text, risk_text, recommendations_text
         else:
+            logging.error(f"Błąd API: {response.status_code} - {response.text}")
             return f"Błąd API: {response.status_code} - {response.text}", "Błąd analizy.", "Błąd analizy."
     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."
 
 # Inicjalizacja pliku statystyk
@@ -167,6 +178,16 @@ def init_stats_file():
         with open(stats_file, 'w') as f:
             json.dump({"total_analyses": 0, "total_frauds_detected": 0}, f)
 
+# Pobranie statystyk
+def get_stats():
+    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}
+
 # Aktualizacja statystyk analizy
 def update_stats(fraud_detected=False):
     stats_file = 'stats.json'