Update app.py

app.py

@@ -4,41 +4,35 @@ import os
 import phonenumbers
 from phonenumbers import geocoder, carrier
 import re
-import sqlite3
-
-# Inicjalizacja bazy danych SQLite do przechowywania fałszywych numerów
-def init_db():
-    conn = sqlite3.connect('fake_numbers.db')
-    c = conn.cursor()
-    c.execute('''
-        CREATE TABLE IF NOT EXISTS fake_numbers (
-            id INTEGER PRIMARY KEY AUTOINCREMENT,
-            phone_number TEXT UNIQUE,
-            reported_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
-        )
-    ''')
-    conn.commit()
-    conn.close()
-
-# Dodanie numeru telefonu do bazy fałszywych numerów
-def add_fake_number(phone_number):
-    conn = sqlite3.connect('fake_numbers.db')
-    c = conn.cursor()
-    try:
-        c.execute('INSERT INTO fake_numbers (phone_number) VALUES (?)', (phone_number,))
-        conn.commit()
-    except sqlite3.IntegrityError:
-        pass  # Numer już istnieje
-    conn.close()
+import json
+from datetime import datetime
+
+# Ścieżka do pliku JSON przechowującego fałszywe numery
+FAKE_NUMBERS_FILE = 'fake_numbers.json'
 
-# Sprawdzenie, czy numer telefonu jest w bazie fałszywych numerów
+# Inicjalizacja pliku JSON
+def init_fake_numbers_file():
+    if not os.path.exists(FAKE_NUMBERS_FILE):
+        with open(FAKE_NUMBERS_FILE, 'w') as f:
+            json.dump([], f)
+
+# Dodanie numeru telefonu do pliku JSON
+def add_fake_number(phone_number):
+    with open(FAKE_NUMBERS_FILE, 'r') as f:
+        fake_numbers = json.load(f)
+    if phone_number not in fake_numbers:
+        fake_numbers.append({
+            "phone_number": phone_number,
+            "reported_at": datetime.now().isoformat()
+        })
+        with open(FAKE_NUMBERS_FILE, 'w') as f:
+            json.dump(fake_numbers, f, indent=4)
+
+# Sprawdzenie, czy numer telefonu jest w pliku JSON
 def is_fake_number(phone_number):
-    conn = sqlite3.connect('fake_numbers.db')
-    c = conn.cursor()
-    c.execute('SELECT * FROM fake_numbers WHERE phone_number = ?', (phone_number,))
-    result = c.fetchone()
-    conn.close()
-    return result is not None
+    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)
 
 # Pobierz klucz API z zmiennej środowiskowej
 API_KEY = os.getenv('SAMBANOVA_API_KEY')
@@ -103,7 +97,7 @@ Jesteś zaawansowanym asystentem AI specjalizującym się w identyfikacji fałsz
 </recommendations>
 
 Twoja odpowiedź powinna być sformatowana dokładnie w powyższy sposób, używając znaczników <analysis>, <risk_assessment> i <recommendations>. Upewnij się, że każda sekcja jest wypełniona kompletnie i szczegółowo.
-"""
+    """
 
     user_prompt = f"""Przeanalizuj następującą wiadomość pod kątem potencjalnego oszustwa:
 
@@ -143,7 +137,7 @@ Podaj swoją analizę i wnioski zgodnie z powyższymi wytycznymi."""
 
 # Główny interfejs aplikacji
 def main():
-    st.set_page_config(page_title="Detektor Fałszywych Wiadomości SMS", page_icon="📱")
+    st.set_page_config(page_title="📱 Detektor Fałszywych Wiadomości SMS", page_icon="📱")
     st.title("📱 Detektor Fałszywych Wiadomości SMS")
     st.write("Wprowadź treść wiadomości SMS oraz numer telefonu nadawcy, aby przeprowadzić analizę pod kątem potencjalnego oszustwa.")
 
@@ -200,6 +194,6 @@ def main():
             st.info("📌 Pamiętaj: zawsze zachowuj ostrożność przy otrzymywaniu wiadomości od nieznanych nadawców. Nie udostępniaj poufnych informacji i nie klikaj w podejrzane linki.")
 
 if __name__ == "__main__":
-    init_db()
+    init_fake_numbers_file()
     main()