Update pages/SearchNumber.py

pages/SearchNumber.py

@@ -1,71 +1,63 @@
+# pages/SearchNumber.py
+
 import streamlit as st
-import json
-from utils.functions import is_fake_number, get_history
+from utils.functions import (
+    get_phone_info,
+    is_fake_number
+)
+import os
 
-# Function to display information about the searched number
 def show_number_search(language):
-    # Define translations for this page
+    # Definiowanie tłumaczeń dla tej podstrony
     page_translations = {
         'Polish': {
-            'header': "🔍 Wyszukiwanie Numeru",
-            'content': """Tutaj możesz sprawdzić, czy dany numer telefonu został zgłoszony jako oszustwo oraz zobaczyć wszystkie przypisane do niego wiadomości.""",
-            'search_placeholder': "Wprowadź numer telefonu (np. +48123456789)...",
+            'header': "🔍 Wyszukaj Numer",
+            'content': "Wprowadź numer telefonu, aby sprawdzić, czy jest zgłoszony jako oszustwo.",
+            'input_placeholder': "Wprowadź numer telefonu...",
             'search_button': "Wyszukaj",
-            'not_found': "Numer nie został zgłoszony jako oszustwo.",
-            'found': "Numer został zgłoszony jako oszustwo.",
-            'message_history': "Historia wiadomości powiązanych z tym numerem:",
-            'no_messages': "Brak powiązanych wiadomości dla tego numeru.",
+            'result_found': "Numer jest zgłoszony jako oszustwo.",
+            'result_not_found': "Numer nie jest zgłoszony jako oszustwo."
         },
         'German': {
             'header': "🔍 Nummernsuche",
-            'content': """Hier können Sie überprüfen, ob eine Telefonnummer als Betrug gemeldet wurde und alle damit verbundenen Nachrichten einsehen.""",
-            'search_placeholder': "Geben Sie die Telefonnummer ein (z. B. +49123456789)...",
+            'content': "Geben Sie eine Telefonnummer ein, um zu überprüfen, ob sie als Betrug gemeldet wurde.",
+            'input_placeholder': "Geben Sie eine Telefonnummer ein...",
             'search_button': "Suchen",
-            'not_found': "Die Nummer wurde nicht als Betrug gemeldet.",
-            'found': "Die Nummer wurde als Betrug gemeldet.",
-            'message_history': "Nachrichtenverlauf zu dieser Nummer:",
-            'no_messages': "Es gibt keine Nachrichten zu dieser Nummer.",
+            'result_found': "Die Nummer wurde als Betrug gemeldet.",
+            'result_not_found': "Die Nummer wurde nicht als Betrug gemeldet."
         },
         'English': {
-            'header': "🔍 Number Search",
-            'content': """Here you can check if a phone number has been reported as fraud and view all messages associated with it.""",
-            'search_placeholder': "Enter phone number (e.g. +123456789)...",
+            'header': "🔍 Search Number",
+            'content': "Enter a phone number to check if it has been reported as fraud.",
+            'input_placeholder': "Enter phone number...",
             'search_button': "Search",
-            'not_found': "The number has not been reported as fraud.",
-            'found': "The number has been reported as fraud.",
-            'message_history': "Message history associated with this number:",
-            'no_messages': "No messages associated with this number.",
+            'result_found': "The number has been reported as fraud.",
+            'result_not_found': "The number has not been reported as fraud."
         }
     }
 
     st.header(page_translations[language]['header'])
     st.write(page_translations[language]['content'])
 
-    # Input for searching a phone number
-    phone_number = st.text_input(page_translations[language]['search_placeholder'], max_chars=15)
+    phone_number = st.text_input(
+        label=page_translations[language]['content'],
+        placeholder=page_translations[language]['input_placeholder']
+    )
 
-    # Search button
     if st.button(page_translations[language]['search_button']):
-        if phone_number:
-            # Check if the number is in the fraud database
-            if is_fake_number(phone_number):
-                st.success(page_translations[language]['found'])
-
-                # Get message history
-                history = get_history()
-                associated_messages = [entry for entry in history if entry['phone_number'] == phone_number]
+        if not phone_number.strip():
+            st.error(page_translations[language]['input_placeholder'])
+        else:
+            country, operator = get_phone_info(phone_number)
+            if country and operator:
+                st.write("**Phone Number Information:**")
+                st.write(f"- **Country:** {country}")
+                st.write(f"- **Operator:** {operator}")
+            else:
+                st.write("Unable to retrieve phone number information.")
 
-                if associated_messages:
-                    st.subheader(page_translations[language]['message_history'])
-                    for entry in associated_messages:
-                        st.markdown(f"**Data:** {entry['timestamp']}")
-                        st.markdown(f"**Treść wiadomości:** {entry['message']}")
-                        st.markdown(f"**Ocena Ryzyka:** {entry['risk_assessment']}")
-                        st.markdown(f"**Rekomendacje:** {entry['recommendations']}")
-                        st.markdown("---")
-                else:
-                    st.info(page_translations[language]['no_messages'])
+            if is_fake_number(phone_number):
+                st.warning(page_translations[language]['result_found'])
             else:
-                st.error(page_translations[language]['not_found'])
-        else:
-            st.warning("Please enter a valid phone number.")
+                st.success(page_translations[language]['result_not_found'])
+