Update app.py

app.py

@@ -4,6 +4,41 @@ 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()
+
+# Sprawdzenie, czy numer telefonu jest w bazie fałszywych numerów
+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
 
 # Pobierz klucz API z zmiennej środowiskowej
 API_KEY = os.getenv('SAMBANOVA_API_KEY')
@@ -131,6 +166,12 @@ def main():
                 else:
                     st.write("Nie udało się uzyskać informacji o numerze telefonu.")
 
+                # Sprawdzenie, czy numer jest w bazie fałszywych numerów
+                if is_fake_number(phone_number):
+                    st.warning("Numer telefonu znajduje się w bazie znanych fałszywych numerów.")
+                else:
+                    st.info("Numer telefonu nie znajduje się w bazie fałszywych numerów.")
+
             # Przeprowadź proste sprawdzenia heurystyczne
             warnings = simple_checks(message)
             if warnings:
@@ -148,8 +189,17 @@ def main():
                 st.subheader("✅ Zalecenia dla użytkownika:")
                 st.markdown(recommendations_text)
 
+            # Opcja dodania numeru do bazy fałszywych numerów
+            if is_fake_number(phone_number):
+                st.success("Numer telefonu został już zgłoszony jako fałszywy.")
+            else:
+                if st.button("Zgłoś numer jako fałszywy"):
+                    add_fake_number(phone_number)
+                    st.success("Numer telefonu został zgłoszony jako fałszywy.")
+
             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()
     main()