Update pages/Analysis.py

pages/Analysis.py

@@ -1,5 +1,3 @@
-# pages/Analysis.py
-
 import streamlit as st
 from utils.functions import (
     get_phone_info,
@@ -20,7 +18,6 @@ def extract_urls(message):
         r'(https?://[^\s]+)|(www\.[^\s]+)'
     )
     urls = url_pattern.findall(message)
-    # Spłaszcz listę
     urls = [url[0] or url[1] for url in urls]
     return urls
 
@@ -40,8 +37,7 @@ def check_urls_with_phishtank(urls):
                 if in_database and valid:
                     phishing_urls.append(url)
             else:
-                # Obsługa błędów
-                pass
+                st.warning(f"Błąd podczas sprawdzania URL w PhishTank: {response.status_code}")
         except Exception as e:
             st.error(f"Błąd podczas sprawdzania URL w PhishTank: {e}")
     return phishing_urls
@@ -49,7 +45,7 @@ def check_urls_with_phishtank(urls):
 def check_urls_with_safe_browsing(urls):
     api_key = os.getenv('GOOGLE_SAFE_BROWSING_API_KEY')
     if not api_key:
-        return None  # Wskaż, że brak klucza API
+        return None
     unsafe_urls = []
     headers = {'Content-Type': 'application/json'}
     client_body = {
@@ -75,333 +71,329 @@ def check_urls_with_safe_browsing(urls):
             matches = data.get('matches', [])
             unsafe_urls = [match['threat']['url'] for match in matches]
         else:
-            # Obsługa błędów
             st.error(f"Błąd podczas komunikacji z Google Safe Browsing API: {response.status_code}")
     except Exception as e:
         st.error(f"Błąd podczas sprawdzania URL w Google Safe Browsing: {e}")
     return unsafe_urls
 
-def show_analysis(language):
-    try:
-        # Definiowanie tłumaczeń dla tej podstrony
-        page_translations = {
-            'Polish': {
-                'header': "📊 Analiza SMS",
-                'content': "Tutaj znajduje się interfejs do analizy wiadomości SMS. Wprowadź treść wiadomości poniżej, aby sprawdzić, czy jest ona potencjalnym oszustwem.",
-                'input_placeholder': "Wprowadź treść SMS...",
-                'analyze_button': "Analizuj wiadomość",
-                'result_positive': "Ostrzeżenie: Wiadomość może być oszustwem!",
-                'result_negative': "Wiadomość wydaje się być bezpieczna.",
-                'report_button': "Zgłoś numer jako oszustwo",
-                'history_header': "Historia Analiz",
-                'phone_input_label': "Numer telefonu nadawcy (z kodem kierunkowym kraju, np. +48123456789):",
-                'additional_info_header': "Dodatkowe Informacje",
-                'frequency_label': "Jak często otrzymujesz wiadomości od tego numeru?",
-                'frequency_options': ('Pierwszy raz', 'Sporadycznie', 'Regularnie'),
-                'last_received_label': "Kiedy ostatnio otrzymałeś podobną wiadomość?",
-                'last_received_options': ('Dziś', 'W ciągu ostatniego tygodnia', 'W ciągu ostatniego miesiąca', 'Dłużej'),
-                'confidence_level_label': "Jak bardzo jesteś pewny, że wiadomość jest autentyczna?",
-                'confidence_level_help': "1: Bardzo mało pewny, 5: Bardzo pewny",
-                'unexpected_label': "Czy otrzymanie tej wiadomości było dla Ciebie niespodziewane?",
-                'unexpected_options': ('Tak', 'Nie'),
-                'additional_notes_label': "Dodatkowe uwagi lub informacje (opcjonalnie)",
-                'phone_info_header': "Informacje o numerze telefonu",
-                'country_label': "Kraj",
-                'operator_label': "Operator",
-                'unable_to_retrieve': "Nie można uzyskać informacji o numerze telefonu.",
-                'red_flags_found': "Znaleziono następujące potencjalne czerwone flagi:",
-                'remember_caution': "📌 Pamiętaj: Bądź ostrożny z wiadomościami od nieznanych nadawców.",
-                'number_reported': "Numer został zgłoszony jako oszustwo.",
-                'number_already_reported': "Numer został już wcześniej zgłoszony jako oszustwo.",
-                'invalid_number': "Podany numer telefonu jest nieprawidłowy.",
-                'failed_to_parse': "Nie udało się przetworzyć numeru telefonu. Upewnij się, że jest w poprawnym formacie.",
-                'url_analysis_header': "🔗 Analiza URL",
-                'extracted_urls': "Wyodrębnione URL z wiadomości:",
-                'no_urls_found': "Brak URL w wiadomości.",
-                'checking_urls_phishtank': "Sprawdzanie URL w PhishTank...",
-                'phishtank_warning': "Następujące URL są oznaczone jako phishing przez PhishTank:",
-                'no_phishtank_threats': "Brak zagrożeń phishingowych według PhishTank.",
-                'checking_urls_safe_browsing': "Sprawdzanie URL w Google Safe Browsing...",
-                'safe_browsing_warning': "Następujące URL są oznaczone jako niebezpieczne przez Google Safe Browsing:",
-                'no_safe_browsing_threats': "Brak zagrożeń według Google Safe Browsing.",
-                'api_key_missing': "Brak klucza API dla usługi: {}",
-                'analyzing_message': "Analizowanie wiadomości za pomocą modelu AI...",
-                'form_submitted': "Formularz został wysłany.",
-                'starting_phone_analysis': "Rozpoczynam analizę numeru telefonu...",
-                'checking_fake_numbers': "Sprawdzam, czy numer jest w bazie fałszywych numerów...",
-                'performing_simple_checks': "Przeprowadzam proste sprawdzenia heurystyczne...",
-                'starting_url_analysis': "Rozpoczynam analizę URL..."
-            },
-            'German': {
-                'header': "📊 SMS-Analyse",
-                'content': "Hier ist die Schnittstelle zur Analyse von SMS-Nachrichten. Geben Sie unten den Inhalt der Nachricht ein, um zu überprüfen, ob es sich um einen potenziellen Betrug handelt.",
-                'input_placeholder': "Geben Sie den SMS-Inhalt ein...",
-                'analyze_button': "Nachricht analysieren",
-                'result_positive': "Warnung: Die Nachricht könnte ein Betrug sein!",
-                'result_negative': "Die Nachricht scheint sicher zu sein.",
-                'report_button': "Nummer als Betrug melden",
-                'history_header': "Analyseverlauf",
-                'phone_input_label': "Telefonnummer des Absenders (mit Ländervorwahl, z.B. +49123456789):",
-                'additional_info_header': "Zusätzliche Informationen",
-                'frequency_label': "Wie oft erhalten Sie Nachrichten von dieser Nummer?",
-                'frequency_options': ('Zum ersten Mal', 'Gelegentlich', 'Regelmäßig'),
-                'last_received_label': "Wann haben Sie zuletzt eine ähnliche Nachricht erhalten?",
-                'last_received_options': ('Heute', 'In der letzten Woche', 'Im letzten Monat', 'Länger her'),
-                'confidence_level_label': "Wie sicher sind Sie, dass die Nachricht authentisch ist?",
-                'confidence_level_help': "1: Sehr unsicher, 5: Sehr sicher",
-                'unexpected_label': "War der Erhalt dieser Nachricht für Sie unerwartet?",
-                'unexpected_options': ('Ja', 'Nein'),
-                'additional_notes_label': "Zusätzliche Anmerkungen oder Informationen (optional)",
-                'phone_info_header': "Informationen zur Telefonnummer",
-                'country_label': "Land",
-                'operator_label': "Anbieter",
-                'unable_to_retrieve': "Informationen zur Telefonnummer können nicht abgerufen werden.",
-                'red_flags_found': "Folgende potenzielle Warnsignale wurden gefunden:",
-                'remember_caution': "📌 Denken Sie daran: Seien Sie vorsichtig bei Nachrichten von unbekannten Absendern.",
-                'number_reported': "Nummer wurde als Betrug gemeldet.",
-                'number_already_reported': "Nummer wurde bereits als Betrug gemeldet.",
-                'invalid_number': "Die angegebene Telefonnummer ist ungültig.",
-                'failed_to_parse': "Die Telefonnummer konnte nicht verarbeitet werden. Stellen Sie sicher, dass sie im richtigen Format ist.",
-                'url_analysis_header': "🔗 URL-Analyse",
-                'extracted_urls': "Aus der Nachricht extrahierte URLs:",
-                'no_urls_found': "Keine URLs in der Nachricht gefunden.",
-                'checking_urls_phishtank': "Überprüfe URLs mit PhishTank...",
-                'phishtank_warning': "Folgende URLs sind von PhishTank als Phishing markiert:",
-                'no_phishtank_threats': "Keine Phishing-Bedrohungen laut PhishTank.",
-                'checking_urls_safe_browsing': "Überprüfe URLs mit Google Safe Browsing...",
-                'safe_browsing_warning': "Folgende URLs sind von Google Safe Browsing als unsicher markiert:",
-                'no_safe_browsing_threats': "Keine Bedrohungen laut Google Safe Browsing.",
-                'api_key_missing': "API-Schlüssel für Dienst fehlt: {}",
-                'analyzing_message': "Analysiere Nachricht mit AI-Modell...",
-                'form_submitted': "Formular wurde eingereicht.",
-                'starting_phone_analysis': "Beginne Analyse der Telefonnummer...",
-                'checking_fake_numbers': "Überprüfe, ob die Nummer in der Betrugsdatenbank ist...",
-                'performing_simple_checks': "Führe einfache heuristische Prüfungen durch...",
-                'starting_url_analysis': "Beginne URL-Analyse..."
-            },
-            'English': {
-                'header': "📊 SMS Analysis",
-                'content': "Here is the interface for analyzing SMS messages. Enter the content of the message below to check if it is a potential fraud.",
-                'input_placeholder': "Enter SMS content...",
-                'analyze_button': "Analyze Message",
-                'result_positive': "Warning: The message may be a fraud!",
-                'result_negative': "The message appears to be safe.",
-                'report_button': "Report number as fraud",
-                'history_header': "Analysis History",
-                'phone_input_label': "Sender's phone number (with country code, e.g., +123456789):",
-                'additional_info_header': "Additional Information",
-                'frequency_label': "How often do you receive messages from this number?",
-                'frequency_options': ('First time', 'Occasionally', 'Regularly'),
-                'last_received_label': "When did you last receive a similar message?",
-                'last_received_options': ('Today', 'Within the last week', 'Within the last month', 'Longer ago'),
-                'confidence_level_label': "How confident are you that the message is authentic?",
-                'confidence_level_help': "1: Very unsure, 5: Very confident",
-                'unexpected_label': "Was receiving this message unexpected for you?",
-                'unexpected_options': ('Yes', 'No'),
-                'additional_notes_label': "Additional notes or information (optional)",
-                'phone_info_header': "Phone Number Information",
-                'country_label': "Country",
-                'operator_label': "Operator",
-                'unable_to_retrieve': "Unable to retrieve phone number information.",
-                'red_flags_found': "The following potential red flags were found:",
-                'remember_caution': "📌 Remember: Be cautious with messages from unknown senders.",
-                'number_reported': "Number has been reported as fraud.",
-                'number_already_reported': "Number has already been reported as fraud.",
-                'invalid_number': "The provided phone number is invalid.",
-                'failed_to_parse': "Failed to parse the phone number. Ensure it is in the correct format.",
-                'url_analysis_header': "🔗 URL Analysis",
-                'extracted_urls': "Extracted URLs from the message:",
-                'no_urls_found': "No URLs found in the message.",
-                'checking_urls_phishtank': "Checking URLs with PhishTank...",
-                'phishtank_warning': "The following URLs are flagged as phishing by PhishTank:",
-                'no_phishtank_threats': "No phishing threats detected by PhishTank.",
-                'checking_urls_safe_browsing': "Checking URLs with Google Safe Browsing...",
-                'safe_browsing_warning': "The following URLs are flagged as unsafe by Google Safe Browsing:",
-                'no_safe_browsing_threats': "No threats detected by Google Safe Browsing.",
-                'api_key_missing': "API key missing for service: {}",
-                'analyzing_message': "Analyzing message using AI model...",
-                'form_submitted': "Form has been submitted.",
-                'starting_phone_analysis': "Starting phone number analysis...",
-                'checking_fake_numbers': "Checking if the number is in the fraud database...",
-                'performing_simple_checks': "Performing simple heuristic checks...",
-                'starting_url_analysis': "Starting URL analysis..."
-            }
-        }
+def analyze_sms(message, phone_number, additional_info, language):
+    st.subheader(get_translation('sms_analysis_header', language))
+    
+    # Informacje o numerze telefonu
+    country, operator = get_phone_info(phone_number)
+    if country and operator:
+        st.write(get_translation('phone_info', language))
+        st.write(f"- {get_translation('country_label', language)}: {country}")
+        st.write(f"- {get_translation('operator_label', language)}: {operator}")
+    else:
+        st.write(get_translation('unable_to_retrieve', language))
 
-        translations = page_translations.get(language, page_translations['English'])
+    # Sprawdzenie, czy numer jest w bazie fałszywych numerów
+    if is_fake_number(phone_number):
+        st.warning(get_translation('number_reported_warning', language))
+        fraud_detected = True
+    else:
+        fraud_detected = False
 
-        st.header(translations['header'])
-        st.write(translations['content'])
+    # Przeprowadź proste sprawdzenia heurystyczne
+    warnings = simple_checks(message, language)
+    if warnings:
+        st.warning(get_translation('red_flags_found', language))
+        for warn in warnings:
+            st.write(f"- {warn}")
 
-        with st.form("sms_form"):
-            message = st.text_area(
-                label=translations['input_placeholder'],
-                placeholder=translations['input_placeholder'],
-                height=150
-            )
-            phone_number = st.text_input(
-                label=translations['phone_input_label'],
-                value="",
-                max_chars=20
-            )
+    # Analiza URL
+    urls = extract_urls(message)
+    if urls:
+        st.subheader(get_translation('url_analysis_header', language))
+        st.write(get_translation('extracted_urls', language))
+        for url in urls:
+            st.write(f"- {url}")
+
+        # Sprawdzanie URL w PhishTank
+        phishing_urls_phishtank = check_urls_with_phishtank(urls)
+        if phishing_urls_phishtank:
+            st.warning(get_translation('phishtank_warning', language))
+            for url in phishing_urls_phishtank:
+                st.write(f"- {url}")
+        else:
+            st.write(get_translation('no_phishtank_threats', language))
+
+        # Sprawdzanie URL w Google Safe Browsing
+        phishing_urls_safe_browsing = check_urls_with_safe_browsing(urls)
+        if phishing_urls_safe_browsing is None:
+            st.warning(get_translation('api_key_missing', language).format('Google Safe Browsing'))
+        elif phishing_urls_safe_browsing:
+            st.warning(get_translation('safe_browsing_warning', language))
+            for url in phishing_urls_safe_browsing:
+                st.write(f"- {url}")
+        else:
+            st.write(get_translation('no_safe_browsing_threats', language))
+
+    # Analiza wiadomości z użyciem modelu AI
+    api_key = os.getenv('SAMBANOVA_API_KEY')
+    if not api_key:
+        st.error(get_translation('api_key_missing', language).format('SambaNova'))
+    else:
+        with st.spinner(get_translation('analyzing_message', language)):
+            try:
+                analysis_text, risk_text, recommendations_text = analyze_message(
+                    message, phone_number, additional_info, api_key, language
+                )
+                st.subheader(get_translation('message_analysis_header', language))
+                st.markdown(analysis_text)
+                st.subheader(get_translation('risk_assessment_header', language))
+                st.markdown(risk_text)
+                st.subheader(get_translation('user_recommendations_header', language))
+                st.markdown(recommendations_text)
+
+                # Aktualizacja statystyk na podstawie oceny ryzyka
+                try:
+                    match = re.search(r'(\d+)/10', risk_text)
+                    fraud_detected = int(match.group(1)) >= 7 if match else fraud_detected
+                    update_stats(fraud_detected=fraud_detected)
+                except Exception as e:
+                    st.error(f"Błąd podczas aktualizacji statystyk: {e}")
 
-            st.markdown(f"### {translations['additional_info_header']}")
+                # Dodanie analizy do historii
+                add_to_history(message, phone_number, analysis_text, risk_text, recommendations_text)
+            except Exception as e:
+                st.error(f"Błąd podczas analizy wiadomości: {e}")
+
+    st.info(get_translation('remember_caution', language))
+
+    # Zgłoszenie numeru jako oszustwo
+    if fraud_detected:
+        if st.button(get_translation('report_button', language)):
+            try:
+                parsed_number = parse(phone_number, None)
+                if is_valid_number(parsed_number):
+                    success = add_fake_number(phone_number)
+                    if success:
+                        st.success(get_translation('number_reported', language))
+                    else:
+                        st.info(get_translation('number_already_reported', language))
+                else:
+                    st.error(get_translation('invalid_number', language))
+            except NumberParseException:
+                st.error(get_translation('failed_to_parse', language))
+
+def analyze_email(email_content, sender_email, language):
+    st.subheader(get_translation('email_analysis_header', language))
+    st.write(get_translation('email_analysis_in_progress', language))
+    # Tutaj możesz dodać logikę analizy e-maili
+
+def analyze_website(url, language):
+    st.subheader(get_translation('website_analysis_header', language))
+    st.write(get_translation('website_analysis_in_progress', language))
+    # Tutaj możesz dodać logikę analizy stron internetowych
+
+def show_history(language):
+    st.subheader(get_translation('history_header', language))
+    st.write(get_translation('history_in_progress', language))
+    # Tutaj możesz dodać logikę wyświetlania historii analiz
+
+def show_statistics(language):
+    st.subheader(get_translation('statistics_header', language))
+    st.write(get_translation('statistics_in_progress', language))
+    # Tutaj możesz dodać logikę wyświetlania statystyk
+
+def get_translation(key, language):
+    translations = {
+        'Polish': {
+            'sms_analysis_header': "Analiza SMS",
+            'email_analysis_header': "Analiza E-mail",
+            'website_analysis_header': "Analiza Strony Internetowej",
+            'history_header': "Historia Analiz",
+            'statistics_header': "Statystyki",
+            'phone_info': "Informacje o numerze telefonu:",
+            'country_label': "Kraj",
+            'operator_label': "Operator",
+            'unable_to_retrieve': "Nie można uzyskać informacji o numerze telefonu.",
+            'number_reported_warning': "Ostrzeżenie: Ten numer został wcześniej zgłoszony jako oszustwo!",
+            'red_flags_found': "Znaleziono następujące potencjalne czerwone flagi:",
+            'url_analysis_header': "Analiza URL",
+            'extracted_urls': "Wyodrębnione URL z wiadomości:",
+            'phishtank_warning': "Następujące URL są oznaczone jako phishing przez PhishTank:",
+            'no_phishtank_threats': "Brak zagrożeń phishingowych według PhishTank.",
+            'api_key_missing': "Brak klucza API dla usługi: {}.",
+            'safe_browsing_warning': "Następujące URL są oznaczone jako niebezpieczne przez Google Safe Browsing:",
+            'no_safe_browsing_threats': "Brak zagrożeń według Google Safe Browsing.",
+            'analyzing_message': "Analizowanie wiadomości za pomocą modelu AI...",
+            'message_analysis_header': "Analiza treści wiadomości:",
+            'risk_assessment_header': "Ocena ryzyka oszustwa:",
+            'user_recommendations_header': "Zalecenia dla użytkownika:",
+            'remember_caution': "📌 Pamiętaj: Bądź ostrożny z wiadomościami od nieznanych nadawców.",
+            'report_button': "Zgłoś numer jako oszustwo",
+            'number_reported': "Numer został zgłoszony jako oszustwo.",
+            'number_already_reported': "Numer został już wcześniej zgłoszony jako oszustwo.",
+            'invalid_number': "Podany numer telefonu jest nieprawidłowy.",
+            'failed_to_parse': "Nie udało się przetworzyć numeru telefonu. Upewnij się, że jest w poprawnym formacie.",
+            'email_analysis_in_progress': "Funkcja analizy e-maili jest w trakcie implementacji.",
+            'website_analysis_in_progress': "Funkcja analizy stron internetowych jest w trakcie implementacji.",
+            'history_in_progress': "Funkcja wyświetlania historii analiz jest w trakcie implementacji.",
+            'statistics_in_progress': "Funkcja wyświetlania statystyk jest w trakcie implementacji.",
+            'analysis_center_title': "Centrum Analizy Zagrożeń",
+            'select_analysis_type': "Wybierz typ analizy:",
+            'additional_info_header': "Dodatkowe Informacje",
+            'frequency_label': "Jak często otrzymujesz wiadomości od tego numeru?",
+            'frequency_options': ('Pierwszy raz', 'Sporadycznie', 'Regularnie'),
+            'last_received_label': "Kiedy ostatnio otrzymałeś podobną wiadomość?",
+            'last_received_options': ('Dziś', 'W ciągu ostatniego tygodnia', 'W ciągu ostatniego miesiąca', 'Dłużej'),
+            'confidence_level_label': "Jak bardzo jesteś pewny, że wiadomość jest autentyczna?",
+            'confidence_level_help': "1: Bardzo mało pewny, 5: Bardzo pewny",
+            'unexpected_label': "Czy otrzymanie tej wiadomości było dla Ciebie niespodziewane?",
+            'unexpected_options': ('Tak', 'Nie'),
+            'additional_notes_label': "Dodatkowe uwagi lub informacje (opcjonalnie)",
+            'analyze_button': "Analizuj",
+            'enter_sms_content': "Wprowadź treść SMS",
+            'sender_phone_number': "Numer telefonu nadawcy (z kodem kierunkowym kraju, np. +48123456789)",
+            'enter_email_content': "Wprowadź treść e-maila",
+            'sender_email_address': "Adres e-mail nadawcy",
+            'enter_website_url': "Wprowadź adres URL strony do analizy"
+        },
+        'English': {
+            'sms_analysis_header': "SMS Analysis",
+            'email_analysis_header': "Email Analysis",
+            'website_analysis_header': "Website Analysis",
+            'history_header': "Analysis History",
+            'statistics_header': "Statistics",
+            'phone_info': "Phone number information:",
+            'country_label': "Country",
+            'operator_label': "Operator",
+            'unable_to_retrieve': "Unable to retrieve phone number information.",
+            'number_reported_warning': "Warning: This number has been previously reported as fraud!",
+            'red_flags_found': "The following potential red flags were found:",
+            'url_analysis_header': "URL Analysis",
+            'extracted_urls': "Extracted URLs from the message:",
+            'phishtank_warning': "The following URLs are flagged as phishing by PhishTank:",
+            'no_phishtank_threats': "No phishing threats detected by PhishTank.",
+            'api_key_missing': "API key missing for service: {}.",
+            'safe_browsing_warning': "The following URLs are flagged as unsafe by Google Safe Browsing:",
+            'no_safe_browsing_threats': "No threats detected by Google Safe Browsing.",
+            'analyzing_message': "Analyzing message using AI model...",
+            'message_analysis_header': "Message Content Analysis:",
+            'risk_assessment_header': "Fraud Risk Assessment:",
+            'user_recommendations_header': "User Recommendations:",
+            'remember_caution': "📌 Remember: Be cautious with messages from unknown senders.",
+            'report_button': "Report number as fraud",
+            'number_reported': "Number has been reported as fraud.",
+            'number_already_reported': "Number has already been reported as fraud.",
+            'invalid_number': "The provided phone number is invalid.",
+            'failed_to_parse': "Failed to parse the phone number. Ensure it is in the correct format.",
+            'email_analysis_in_progress': "Email analysis function is under development.",
+            'website_analysis_in_progress': "Website analysis function is under development.",
+            'history_in_progress': "Analysis history function is under development.",
+            'statistics_in_progress': "Statistics display function is under development.",
+            'analysis_center_title': "Threat Analysis Center",
+            'select_analysis_type': "Select analysis type:",
+            'additional_info_header': "Additional Information",
+            'frequency_label': "How often do you receive messages from this number?",
+            'frequency_options': ('First time', 'Occasionally', 'Regularly'),
+            'last_received_label': "When did you last receive a similar message?",
+            'last_received_options': ('Today', 'Within the last week', 'Within the last month', 'Longer ago'),
+            'confidence_level_label': "How confident are you that the message is authentic?",
+            'confidence_level_help': "1: Very unsure, 5: Very confident",
+            'unexpected_label': "Was receiving this message unexpected for you?",
+            'unexpected_options': ('Yes', 'No'),
+            'additional_notes_label': "Additional notes or information (optional)",
+            'analyze_button': "Analyze",
+            'enter_sms_content': "Enter SMS content",
+            'sender_phone_number': "Sender's phone number (with country code, e.g., +123456789)",
+            'enter_email_content': "Enter email content",
+            'sender_email_address': "Sender's email address",
+            'enter_website_url': "Enter website URL to analyze"
+        },
+        'German': {
+            # Dodaj tłumaczenia dla języka niemieckiego
+            # Add German translations here
+        }
+    }
+    return translations.get(language, translations['English']).get(key, key)
 
-            # Dodane pytania
+def show_analysis(language):
+    st.title(get_translation('analysis_center_title', language))
+    
+    analysis_type = st.sidebar.radio(
+        get_translation('select_analysis_type', language),
+        (
+            get_translation('sms_analysis_header', language),
+            get_translation('email_analysis_header', language),
+            get_translation('website_analysis_header', language),
+            get_translation('history_header', language),
+            get_translation('statistics_header', language)
+        )
+    )
+
+    if analysis_type == get_translation('sms_analysis_header', language):
+        with st.form("sms_form"):
+            message = st.text_area(get_translation('enter_sms_content', language), height=150)
+            phone_number = st.text_input(get_translation('sender_phone_number', language))
+            
+            st.subheader(get_translation('additional_info_header', language))
             frequency = st.selectbox(
-                label=translations['frequency_label'],
-                options=translations['frequency_options']
+                get_translation('frequency_label', language),
+                get_translation('frequency_options', language)
             )
             last_received = st.selectbox(
-                label=translations['last_received_label'],
-                options=translations['last_received_options']
+                get_translation('last_received_label', language),
+                get_translation('last_received_options', language)
             )
             confidence_level = st.slider(
-                label=translations['confidence_level_label'],
-                min_value=1,
-                max_value=5,
-                value=3,
-                help=translations['confidence_level_help']
+                get_translation('confidence_level_label', language),
+                1, 5, 3,
+                help=get_translation('confidence_level_help', language)
             )
             unexpected = st.radio(
-                label=translations['unexpected_label'],
-                options=translations['unexpected_options']
-            )
-            additional_notes = st.text_area(
-                label=translations['additional_notes_label'],
-                height=100
+                get_translation('unexpected_label', language),
+                get_translation('unexpected_options', language)
             )
+            additional_notes = st.text_area(get_translation('additional_notes_label', language), height=100)
 
-            submitted = st.form_submit_button(translations['analyze_button'])
+            submitted = st.form_submit_button(get_translation('analyze_button', language))
 
         if submitted:
-            st.write(translations['form_submitted'])  # Diagnostyka
             if not message.strip():
-                st.error(translations['input_placeholder'])
+                st.error(get_translation('enter_sms_content', language))
             elif not phone_number.strip():
-                st.error(translations['phone_input_label'])
+                st.error(get_translation('sender_phone_number', language))
             else:
-                # Informacje o numerze telefonu
-                st.write(translations['starting_phone_analysis'])  # Diagnostyka
-                country, operator = get_phone_info(phone_number)
-                if country and operator:
-                    st.subheader(translations['phone_info_header'])
-                    st.write(f"- **{translations['country_label']}:** {country}")
-                    st.write(f"- **{translations['operator_label']}:** {operator}")
-                else:
-                    st.write(translations['unable_to_retrieve'])
-
-                # Sprawdzenie, czy numer jest w bazie fałszywych numerów
-                st.write(translations['checking_fake_numbers'])  # Diagnostyka
-                if is_fake_number(phone_number):
-                    st.warning(translations['result_positive'])
-                    fraud_detected = True
-                else:
-                    st.info(translations['result_negative'])
-                    fraud_detected = False
-
-                # Przeprowadź proste sprawdzenia heurystyczne
-                st.write(translations['performing_simple_checks'])  # Diagnostyka
-                warnings = simple_checks(message, language)
-                if warnings:
-                    st.warning(translations['red_flags_found'])
-                    for warn in warnings:
-                        st.write(f"- {warn}")
-
-                # Zbierz dodatkowe informacje
                 additional_info = f"""
-**{translations['frequency_label']}** {frequency}
-
-**{translations['last_received_label']}** {last_received}
-
-**{translations['confidence_level_label']}** {confidence_level}/5
-
-**{translations['unexpected_label']}** {unexpected}
-
-**{translations['additional_notes_label']}** {additional_notes}
+**{get_translation('frequency_label', language)}**: {frequency}
+**{get_translation('last_received_label', language)}**: {last_received}
+**{get_translation('confidence_level_label', language)}**: {confidence_level}/5
+**{get_translation('unexpected_label', language)}**: {unexpected}
+**{get_translation('additional_notes_label', language)}**: {additional_notes}
                 """
+                analyze_sms(message, phone_number, additional_info, language)
 
-                # Analiza URL
-                st.write(translations['starting_url_analysis'])  # Diagnostyka
-                urls = extract_urls(message)
-                if urls:
-                    st.subheader(translations['url_analysis_header'])
-                    st.write(translations['extracted_urls'])
-                    for url in urls:
-                        st.write(f"- {url}")
-
-                    # Sprawdzanie URL w PhishTank
-                    st.write(translations['checking_urls_phishtank'])
-                    phishing_urls_phishtank = check_urls_with_phishtank(urls)
-                    if phishing_urls_phishtank:
-                        st.warning(translations['phishtank_warning'])
-                        for url in phishing_urls_phishtank:
-                            st.write(f"- {url}")
-                    else:
-                        st.write(translations['no_phishtank_threats'])
-
-                    # Sprawdzanie URL w Google Safe Browsing
-                    st.write(translations['checking_urls_safe_browsing'])
-                    phishing_urls_safe_browsing = check_urls_with_safe_browsing(urls)
-                    if phishing_urls_safe_browsing is None:
-                        st.warning(translations['api_key_missing'].format('Google Safe Browsing'))
-                    elif phishing_urls_safe_browsing:
-                        st.warning(translations['safe_browsing_warning'])
-                        for url in phishing_urls_safe_browsing:
-                            st.write(f"- {url}")
-                    else:
-                        st.write(translations['no_safe_browsing_threats'])
-                else:
-                    st.write(translations['no_urls_found'])
+    elif analysis_type == get_translation('email_analysis_header', language):
+        with st.form("email_form"):
+            email_content = st.text_area(get_translation('enter_email_content', language), height=150)
+            sender_email = st.text_input(get_translation('sender_email_address', language))
+            submitted = st.form_submit_button(get_translation('analyze_button', language))
 
-                # Pobierz klucz API z zmiennych środowiskowych
-                api_key = os.getenv('SAMBANOVA_API_KEY')
-                if not api_key:
-                    st.error("API key is missing. Please set the SAMBANOVA_API_KEY environment variable.")
-                else:
-                    with st.spinner(translations['analyzing_message']):
-                        # Upewnij się, że funkcja analyze_message jest zaimplementowana
-                        try:
-                            analysis_text, risk_text, recommendations_text = analyze_message(
-                                message, phone_number, additional_info, api_key, language
-                            )
-                            st.subheader("📑 " + translations['header'])
-                            st.markdown(analysis_text)
-                            st.subheader("⚠️ " + translations['red_flags_found'])
-                            st.markdown(risk_text)
-                            st.subheader("✅ " + translations['remember_caution'])
-                            st.markdown(recommendations_text)
+        if submitted:
+            if not email_content.strip():
+                st.error(get_translation('enter_email_content', language))
+            elif not sender_email.strip():
+                st.error(get_translation('sender_email_address', language))
+            else:
+                analyze_email(email_content, sender_email, language)
 
-                            # Aktualizacja statystyk na podstawie oceny ryzyka
-                            try:
-                                match = re.search(r'(\d+)/10', risk_text)
-                                fraud_detected = int(match.group(1)) >= 7 if match else False
-                                update_stats(fraud_detected=fraud_detected)
-                            except Exception as e:
-                                st.error(f"Błąd podczas aktualizacji statystyk: {e}")
-                                fraud_detected = False
+    elif analysis_type == get_translation('website_analysis_header', language):
+        with st.form("website_form"):
+            url = st.text_input(get_translation('enter_website_url', language))
+            submitted = st.form_submit_button(get_translation('analyze_button', language))
 
-                            # Dodanie analizy do historii
-                            add_to_history(message, phone_number, analysis_text, risk_text, recommendations_text)
-                        except Exception as e:
-                            st.error(f"Błąd podczas analizy wiadomości: {e}")
+        if submitted:
+            if not url.strip():
+                st.error(get_translation('enter_website_url', language))
+            else:
+                analyze_website(url, language)
 
-                    st.info(translations['remember_caution'])
+    elif analysis_type == get_translation('history_header', language):
+        show_history(language)
 
-                # Zgłoszenie numeru jako oszustwo
-                if fraud_detected:
-                    if st.button(translations['report_button']):
-                        try:
-                            parsed_number = parse(phone_number, None)
-                            if is_valid_number(parsed_number):
-                                success = add_fake_number(phone_number)
-                                if success:
-                                    st.success(translations['number_reported'])
-                                else:
-                                    st.info(translations['number_already_reported'])
-                            else:
-                                st.error(translations['invalid_number'])
-                        except NumberParseException:
-                            st.error(translations['failed_to_parse'])
-    except Exception as e:
-        st.error(f"Wystąpił błąd: {e}")
+    elif analysis_type == get_translation('statistics_header', language):
+        show_statistics(language)