Update utils/functions.py

utils/functions.py

@@ -12,12 +12,6 @@ from sqlalchemy import create_engine, Column, String, Integer, DateTime
 from sqlalchemy.ext.declarative import declarative_base
 from sqlalchemy.orm import sessionmaker
 
-from dotenv import load_dotenv
-
-# Załaduj zmienne środowiskowe z pliku .env (jeśli używasz)
-# W Streamlit Spaces Secrets są automatycznie dostępne
-load_dotenv()
-
 # Konfiguracja logowania
 logging.basicConfig(
     filename='app.log',
@@ -55,20 +49,15 @@ class Stats(Base):
     total_analyses = Column(Integer, default=0)
     total_frauds_detected = Column(Integer, default=0)
 
-# Tworzenie tabel w bazie danych
+# Tworzenie tabel w bazie danych (jeśli jeszcze nie istnieją)
 Base.metadata.create_all(bind=engine)
 
 # Funkcje pomocnicze
 
-def get_db():
-    db = SessionLocal()
-    try:
-        yield db
-    finally:
-        db.close()
-
-# Dodanie numeru telefonu do bazy danych
 def add_fake_number(phone_number):
+    """
+    Dodaje numer telefonu do bazy danych jako fałszywy, jeśli jeszcze go tam nie ma.
+    """
     db = SessionLocal()
     try:
         existing_number = db.query(FakeNumber).filter(FakeNumber.phone_number == phone_number).first()
@@ -89,8 +78,10 @@ def add_fake_number(phone_number):
     finally:
         db.close()
 
-# Sprawdzenie, czy numer telefonu jest w bazie danych
 def is_fake_number(phone_number):
+    """
+    Sprawdza, czy dany numer telefonu jest oznaczony jako fałszywy w bazie danych.
+    """
     db = SessionLocal()
     try:
         exists = db.query(FakeNumber).filter(FakeNumber.phone_number == phone_number).first() is not None
@@ -102,8 +93,89 @@ def is_fake_number(phone_number):
     finally:
         db.close()
 
-# Funkcja do weryfikacji numeru telefonu
+def add_to_history(message, phone_number, analysis, risk, recommendations):
+    """
+    Dodaje wpis do historii analiz.
+    """
+    db = SessionLocal()
+    try:
+        new_entry = AnalysisHistory(
+            message=message,
+            phone_number=phone_number,
+            analysis=analysis,
+            risk_assessment=risk,
+            recommendations=recommendations
+        )
+        db.add(new_entry)
+        db.commit()
+        logging.info(f"Dodano wpis do historii analiz dla numeru {phone_number}.")
+    except Exception as e:
+        db.rollback()
+        logging.error(f"Nie udało się zapisać historii analiz dla numeru {phone_number}: {e}")
+    finally:
+        db.close()
+
+def update_stats(fraud_detected=False):
+    """
+    Aktualizuje statystyki analiz w bazie danych.
+    """
+    db = SessionLocal()
+    try:
+        stats = db.query(Stats).first()
+        if not stats:
+            stats = Stats()
+            db.add(stats)
+        
+        stats.total_analyses += 1
+        if fraud_detected:
+            stats.total_frauds_detected += 1
+
+        db.commit()
+        logging.info(f"Statystyki zostały zaktualizowane: Analiz {stats.total_analyses}, Oszustw {stats.total_frauds_detected}.")
+    except Exception as e:
+        db.rollback()
+        logging.error(f"Nie udało się zaktualizować statystyk: {e}")
+    finally:
+        db.close()
+
+def get_stats():
+    """
+    Pobiera statystyki analiz z bazy danych.
+    """
+    db = SessionLocal()
+    try:
+        stats = db.query(Stats).first()
+        if stats:
+            logging.info("Statystyki zostały pobrane pomyślnie.")
+            return {"total_analyses": stats.total_analyses, "total_frauds_detected": stats.total_frauds_detected}
+        else:
+            logging.info("Brak statystyk w bazie danych.")
+            return {"total_analyses": 0, "total_frauds_detected": 0}
+    except Exception as e:
+        logging.error(f"Nie udało się pobrać statystyk: {e}")
+        return {"total_analyses": 0, "total_frauds_detected": 0}
+    finally:
+        db.close()
+
+def get_history():
+    """
+    Pobiera historię analiz z bazy danych.
+    """
+    db = SessionLocal()
+    try:
+        history = db.query(AnalysisHistory).order_by(AnalysisHistory.timestamp.desc()).all()
+        logging.info("Historia analiz została pobrana pomyślnie.")
+        return history
+    except Exception as e:
+        logging.error(f"Nie udało się pobrać historii analiz: {e}")
+        return []
+    finally:
+        db.close()
+
 def get_phone_info(phone_number):
+    """
+    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')
@@ -114,8 +186,10 @@ def get_phone_info(phone_number):
         logging.error(f"Nie udało się przetworzyć numeru telefonu {phone_number}: {e}")
         return None, None
 
-# Proste sprawdzenia heurystyczne wiadomości
 def simple_checks(message, language):
+    """
+    Przeprowadza proste sprawdzenia heurystyczne wiadomości SMS.
+    """
     warnings = []
     # Baza słów kluczowych (polski, niemiecki, angielski)
     scam_keywords = {
@@ -134,8 +208,10 @@ def simple_checks(message, language):
         warnings.append("Wiadomość zawiera prośbę o poufne informacje.")
     return warnings
 
-# Funkcja do analizy wiadomości za pomocą API SambaNova
 def analyze_message(message, phone_number, additional_info, api_key, language):
+    """
+    Analizuje wiadomość SMS za pomocą API SambaNova.
+    """
     if not api_key:
         logging.error("Brak klucza API.")
         return "Brak klucza API.", "Brak klucza API.", "Brak klucza API."
@@ -269,90 +345,3 @@ Provide your analysis and conclusions following the guidelines above."""
     except Exception as e:
         logging.error(f"API Connection Error: {e}")
         return f"API Connection Error: {e}", "Analysis Error.", "Analysis Error."
-
-# Inicjalizacja pliku statystyk
-def init_stats_file():
-    db = SessionLocal()
-    try:
-        stats = db.query(Stats).first()
-        if not stats:
-            new_stats = Stats()
-            db.add(new_stats)
-            db.commit()
-            logging.info("Utworzono nowe statystyki.")
-    except Exception as e:
-        db.rollback()
-        logging.error(f"Nie udało się zainicjalizować statystyk: {e}")
-    finally:
-        db.close()
-
-# Pobranie statystyk
-def get_stats():
-    db = SessionLocal()
-    try:
-        stats = db.query(Stats).first()
-        if stats:
-            logging.info("Statystyki zostały pobrane pomyślnie.")
-            return {"total_analyses": stats.total_analyses, "total_frauds_detected": stats.total_frauds_detected}
-        else:
-            logging.info("Brak statystyk w bazie danych.")
-            return {"total_analyses": 0, "total_frauds_detected": 0}
-    except Exception as e:
-        logging.error(f"Nie udało się pobrać statystyk: {e}")
-        return {"total_analyses": 0, "total_frauds_detected": 0}
-    finally:
-        db.close()
-
-# Aktualizacja statystyk analizy
-def update_stats(fraud_detected=False):
-    db = SessionLocal()
-    try:
-        stats = db.query(Stats).first()
-        if not stats:
-            stats = Stats()
-            db.add(stats)
-        
-        stats.total_analyses += 1
-        if fraud_detected:
-            stats.total_frauds_detected += 1
-
-        db.commit()
-        logging.info(f"Statystyki zostały zaktualizowane: Analiz {stats.total_analyses}, Oszustw {stats.total_frauds_detected}.")
-    except Exception as e:
-        db.rollback()
-        logging.error(f"Nie udało się zaktualizować statystyk: {e}")
-    finally:
-        db.close()
-
-# Dodanie wpisu do historii analiz
-def add_to_history(message, phone_number, analysis, risk, recommendations):
-    db = SessionLocal()
-    try:
-        new_entry = AnalysisHistory(
-            message=message,
-            phone_number=phone_number,
-            analysis=analysis,
-            risk_assessment=risk,
-            recommendations=recommendations
-        )
-        db.add(new_entry)
-        db.commit()
-        logging.info(f"Dodano wpis do historii analiz dla numeru {phone_number}.")
-    except Exception as e:
-        db.rollback()
-        logging.error(f"Nie udało się zapisać historii analiz dla numeru {phone_number}: {e}")
-    finally:
-        db.close()
-
-# Pobranie historii analiz
-def get_history():
-    db = SessionLocal()
-    try:
-        history = db.query(AnalysisHistory).order_by(AnalysisHistory.timestamp.desc()).all()
-        logging.info("Historia analiz została pobrana pomyślnie.")
-        return history
-    except Exception as e:
-        logging.error(f"Nie udało się pobrać historii analiz: {e}")
-        return []
-    finally:
-        db.close()