Update pages/Analysis.py

pages/Analysis.py

@@ -8,8 +8,7 @@ from utils.functions import (
     add_fake_number,
     is_fake_number,
     update_stats,
-    add_to_history,
-    get_history
+    add_to_history
 )
 import os
 import re  # Dodany import do obsługi wyrażeń regularnych
@@ -18,173 +17,284 @@ import re  # Dodany import do obsługi wyrażeń regularnych
 page_translations = {
     'Polish': {
         'header': "📊 Analiza SMS",
-        'content': """Tutaj znajduje się interfejs do analizy wiadomości SMS. Wprowadź treść wiadomości oraz numer telefonu nadawcy, aby sprawdzić, czy jest ona potencjalnym oszustwem.""",
-        'input_placeholder': "Wprowadź treść SMS...",
-        'phone_input_label': "Numer telefonu nadawcy (w formacie międzynarodowym, np. +48123123123)",
+        'content': "Tutaj znajduje się interfejs do analizy wiadomości SMS. Wprowadź treść wiadomości oraz numer telefonu nadawcy, aby sprawdzić, czy jest ona potencjalnym oszustwem.",
+        'sms_placeholder': "Wprowadź treść SMS...",
+        'phone_placeholder': "Numer telefonu nadawcy (np. +48123123123)",
         '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",
+        'form_label_message': "Treść wiadomości SMS",
+        'form_label_phone': "Numer telefonu nadawcy",
+        'form_label_frequency': "Jak często otrzymujesz wiadomości od tego numeru?",
+        'form_label_last_received': "Kiedy ostatnio otrzymałeś podobną wiadomość?",
+        'form_label_confidence': "Jak bardzo jesteś pewny, że wiadomość jest autentyczna?",
+        'form_label_unexpected': "Czy otrzymanie tej wiadomości było dla Ciebie niespodziewane?",
+        'form_label_notes': "Dodatkowe uwagi lub informacje (opcjonalnie)",
+        'form_frequency_options': ['Pierwszy raz', 'Sporadycznie', 'Regularnie'],
+        'form_last_received_options': ['Dziś', 'W ciągu ostatniego tygodnia', 'W ciągu ostatniego miesiąca', 'Dłużej'],
+        'form_unexpected_options': ['Tak', 'Nie'],
         'no_history': "Brak historii analiz.",
-        'invalid_api_key': "Brak klucza API. Ustaw zmienną środowiskową SAMBANOVA_API_KEY w ustawieniach aplikacji.",
         'info_reminder': "📌 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.",
-        'error_message_empty': "Proszę wprowadzić treść wiadomości.",
-        'error_phone_empty': "Proszę wprowadzić numer telefonu nadawcy.",
-        'invalid_phone': "Nieprawidłowy numer telefonu.",
-        'already_reported': "Numer został już zgłoszony jako oszustwo."
+        'error_no_api_key': "Brak klucza API. Ustaw zmienną środowiskową SAMBANOVA_API_KEY w ustawieniach aplikacji.",
+        'error_invalid_phone': "Nieprawidłowy numer telefonu.",
+        'warning_no_message': "Proszę wprowadzić treść wiadomości.",
+        'warning_no_phone': "Proszę wprowadzić numer telefonu nadawcy.",
+        'success_reported': "Numer został zgłoszony jako oszustwo.",
+        'info_already_reported': "Numer został już zgłoszony jako oszustwo."
     },
     'German': {
         'header': "📊 SMS-Analyse",
-        'content': """Hier befindet sich die Benutzeroberfläche zur Analyse von SMS-Nachrichten. Geben Sie den Text der Nachricht sowie die Telefonnummer des Absenders ein, um zu überprüfen, ob es sich um einen potenziellen Betrugsversuch handelt.""",
-        'input_placeholder': "Geben Sie den SMS-Inhalt ein...",
-        'phone_input_label': "Telefonnummer des Absenders (im internationalen Format, z.B. +49123123123)",
+        'content': "Hier befindet sich die Benutzeroberfläche zur Analyse von SMS-Nachrichten. Geben Sie den Text der Nachricht sowie die Telefonnummer des Absenders ein, um zu überprüfen, ob es sich um einen potenziellen Betrugsversuch handelt.",
+        'sms_placeholder': "Geben Sie den SMS-Inhalt ein...",
+        'phone_placeholder': "Telefonnummer des Absenders (z.B. +49123123123)",
         '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",
-        'no_history': "Keine Analysehistorie vorhanden.",
-        'invalid_api_key': "Fehlender API-Schlüssel. Setzen Sie die Umgebungsvariable SAMBANOVA_API_KEY in den App-Einstellungen.",
-        'info_reminder': "📌 Erinnerung: Seien Sie immer vorsichtig beim Empfang von Nachrichten von unbekannten Absendern. Teilen Sie keine vertraulichen Informationen und klicken Sie nicht auf verdächtige Links.",
-        'error_message_empty': "Bitte geben Sie den Nachrichtentext ein.",
-        'error_phone_empty': "Bitte geben Sie die Telefonnummer des Absenders ein.",
-        'invalid_phone': "Ungültige Telefonnummer.",
-        'already_reported': "Nummer wurde bereits als Betrug gemeldet."
+        'form_label_message': "SMS-Nachricht Inhalt",
+        'form_label_phone': "Telefonnummer des Absenders",
+        'form_label_frequency': "Wie häufig erhalten Sie Nachrichten von dieser Nummer?",
+        'form_label_last_received': "Wann haben Sie zuletzt eine ähnliche Nachricht erhalten?",
+        'form_label_confidence': "Wie sicher sind Sie, dass die Nachricht authentisch ist?",
+        'form_label_unexpected': "War das Erhalten dieser Nachricht für Sie unerwartet?",
+        'form_label_notes': "Zusätzliche Anmerkungen oder Informationen (optional)",
+        'form_frequency_options': ['Zum ersten Mal', 'Gelegentlich', 'Regelmäßig'],
+        'form_last_received_options': ['Heute', 'In der letzten Woche', 'In den letzten Monaten', 'Länger'],
+        'form_unexpected_options': ['Ja', 'Nein'],
+        'no_history': "Keine Analyseverlauf vorhanden.",
+        'info_reminder': "📌 Erinnerung: Seien Sie immer vorsichtig beim Erhalt von Nachrichten von unbekannten Absendern. Geben Sie keine vertraulichen Informationen preis und klicken Sie nicht auf verdächtige Links.",
+        'error_no_api_key': "Kein API-Schlüssel vorhanden. Setzen Sie die Umgebungsvariable SAMBANOVA_API_KEY in den Anwendungseinstellungen.",
+        'error_invalid_phone': "Ungültige Telefonnummer.",
+        'warning_no_message': "Bitte geben Sie den Nachrichtentext ein.",
+        'warning_no_phone': "Bitte geben Sie die Telefonnummer des Absenders ein.",
+        'success_reported': "Nummer wurde als Betrug gemeldet.",
+        'info_already_reported': "Nummer wurde bereits als Betrug gemeldet."
     },
     'English': {
         'header': "📊 SMS Analysis",
-        'content': """Here is the interface for analyzing SMS messages. Enter the message content and the sender's phone number to check if it is a potential fraud attempt.""",
-        'input_placeholder': "Enter SMS content...",
-        'phone_input_label': "Sender's Phone Number (in international format, e.g., +123123123123)",
+        'content': "Here is the interface for analyzing SMS messages. Enter the message content and the sender's phone number to check if it is a potential fraud attempt.",
+        'sms_placeholder': "Enter SMS content...",
+        'phone_placeholder': "Sender's phone number (e.g., +123123123123)",
         '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",
+        'report_button': "Report number as fraud",
         'history_header': "Analysis History",
+        'form_label_message': "SMS Message Content",
+        'form_label_phone': "Sender's Phone Number",
+        'form_label_frequency': "How often do you receive messages from this number?",
+        'form_label_last_received': "When did you last receive a similar message?",
+        'form_label_confidence': "How confident are you that the message is authentic?",
+        'form_label_unexpected': "Was receiving this message unexpected for you?",
+        'form_label_notes': "Additional notes or information (optional)",
+        'form_frequency_options': ['First time', 'Occasionally', 'Regularly'],
+        'form_last_received_options': ['Today', 'Within the last week', 'Within the last month', 'Longer'],
+        'form_unexpected_options': ['Yes', 'No'],
         'no_history': "No analysis history available.",
-        'invalid_api_key': "Missing API key. Set the SAMBANOVA_API_KEY environment variable in the app settings.",
-        'info_reminder': "📌 Reminder: Always exercise caution when receiving messages from unknown senders. Do not share confidential information and avoid clicking on suspicious links.",
-        'error_message_empty': "Please enter the message content.",
-        'error_phone_empty': "Please enter the sender's phone number.",
-        'invalid_phone': "Invalid phone number.",
-        'already_reported': "Number has already been reported as fraud."
+        'info_reminder': "📌 Remember: always exercise caution when receiving messages from unknown senders. Do not share confidential information and do not click on suspicious links.",
+        'error_no_api_key': "No API key available. Set the SAMBANOVA_API_KEY environment variable in the application settings.",
+        'error_invalid_phone': "Invalid phone number.",
+        'warning_no_message': "Please enter the message content.",
+        'warning_no_phone': "Please enter the sender's phone number.",
+        'success_reported': "Number reported as fraud.",
+        'info_already_reported': "Number has already been reported as fraud."
     }
 }
 
 def show_analysis(language):
-    translations = page_translations.get(language, page_translations['Polish'])  # Domyślnie Polish
+    translations = page_translations[language]
+    
     st.header(translations['header'])
     st.write(translations['content'])
 
     with st.form("sms_form"):
-        message = st.text_area("Treść wiadomości SMS", placeholder=translations['input_placeholder'], height=150)
-        phone_number = st.text_input("Numer telefonu nadawcy", placeholder=translations['phone_input_label'])
+        message = st.text_area(translations['form_label_message'], placeholder=translations['sms_placeholder'], height=150)
+        phone_number = st.text_input(translations['form_label_phone'], placeholder=translations['phone_placeholder'], max_chars=15)
+        
+        st.markdown("### " + "Dodatkowe Informacje" if language == 'Polish' else 
+                    "### " + "Zusätzliche Informationen" if language == 'German' else 
+                    "### " + "Additional Information")
         
-        st.markdown("### Dodatkowe Informacje")
-        # Dodane pytania
         frequency = st.selectbox(
-            "Jak często otrzymujesz wiadomości od tego numeru?",
-            ('Pierwszy raz', 'Sporadycznie', 'Regularnie')
+            translations['form_label_frequency'],
+            translations['form_frequency_options']
         )
         last_received = st.selectbox(
-            "Kiedy ostatnio otrzymałeś podobną wiadomość?",
-            ('Dziś', 'W ciągu ostatniego tygodnia', 'W ciągu ostatniego miesiąca', 'Dłużej')
+            translations['form_label_last_received'],
+            translations['form_last_received_options']
         )
         confidence_level = st.slider(
-            "Jak bardzo jesteś pewny, że wiadomość jest autentyczna?",
+            translations['form_label_confidence'],
             1, 5, 3,
-            help="1: Bardzo mało pewny, 5: Bardzo pewny"
+            help="1: Very Unconfident, 5: Very Confident" if language == 'English' else
+                 "1: Bardzo mało pewny, 5: Bardzo pewny" if language == 'Polish' else
+                 "1: Sehr unsicher, 5: Sehr sicher"
         )
         unexpected = st.radio(
-            "Czy otrzymanie tej wiadomości było dla Ciebie niespodziewane?",
-            ('Tak', 'Nie')
+            translations['form_label_unexpected'],
+            translations['form_unexpected_options']
         )
-        additional_notes = st.text_area("Dodatkowe uwagi lub informacje (opcjonalnie)", height=100)
+        additional_notes = st.text_area(translations['form_label_notes'], height=100)
         
         submitted = st.form_submit_button(translations['analyze_button'])
 
     if submitted:
         if not message.strip():
-            st.error(translations['error_message_empty'])
+            st.error(translations['warning_no_message'])
         elif not phone_number.strip():
-            st.error(translations['error_phone_empty'])
+            st.error(translations['warning_no_phone'])
         else:
-            # Weryfikacja numeru telefonu
+            # Informacje o numerze telefonu
             country, operator = get_phone_info(phone_number)
-            if not country:
-                st.error(translations['invalid_phone'])
+            if country and operator:
+                if language == 'Polish':
+                    st.write(f"**Informacje o numerze telefonu:**")
+                    st.write(f"- **Kraj:** {country}")
+                    st.write(f"- **Operator:** {operator}")
+                elif language == 'German':
+                    st.write(f"**Informationen zur Telefonnummer:**")
+                    st.write(f"- **Land:** {country}")
+                    st.write(f"- **Anbieter:** {operator}")
+                else:
+                    st.write(f"**Phone Number Information:**")
+                    st.write(f"- **Country:** {country}")
+                    st.write(f"- **Carrier:** {operator}")
             else:
-                st.write(f"**Informacje o numerze telefonu:**")
-                st.write(f"- **Kraj:** {country}")
-                st.write(f"- **Operator:** {operator}")
-
-                # Sprawdzenie, czy numer jest w bazie fałszywych numerów
-                if is_fake_number(phone_number):
-                    st.warning(translations['already_reported'])
+                if language == 'Polish':
+                    st.write("Nie udało się uzyskać informacji o numerze telefonu.")
+                elif language == 'German':
+                    st.write("Informationen zur Telefonnummer konnten nicht abgerufen werden.")
+                else:
+                    st.write("Failed to retrieve phone number information.")
+    
+            # Sprawdzenie, czy numer jest w bazie fałszywych numerów
+            if is_fake_number(phone_number):
+                if language == 'Polish':
+                    st.warning("Numer telefonu znajduje się w bazie znanych fałszywych numerów.")
+                elif language == 'German':
+                    st.warning("Die Telefonnummer befindet sich in der Datenbank bekannter betrügerischer Nummern.")
                 else:
+                    st.warning("The phone number is in the database of known fake numbers.")
+            else:
+                if language == 'Polish':
                     st.info("Numer telefonu nie znajduje się w bazie fałszywych numerów.")
-
-                # Przeprowadź proste sprawdzenia heurystyczne
-                warnings = simple_checks(message)
-                if warnings:
+                elif language == 'German':
+                    st.info("Die Telefonnummer befindet sich nicht in der Datenbank bekannter betrügerischer Nummern.")
+                else:
+                    st.info("The phone number is not in the database of known fake numbers.")
+    
+            # Przeprowadź proste sprawdzenia heurystyczne
+            warnings = simple_checks(message)
+            if warnings:
+                if language == 'Polish':
                     st.warning("Znaleziono następujące potencjalne czerwone flagi:")
-                    for warn in warnings:
-                        st.write(f"- {warn}")
-
-                # Zbierz dodatkowe informacje
-                additional_info = f"""
-**Jak często otrzymujesz wiadomości od tego numeru?** {frequency}
-**Kiedy ostatnio otrzymałeś podobną wiadomość?** {last_received}
-**Jak bardzo jesteś pewny, że wiadomość jest autentyczna?** {confidence_level}/5
-**Czy otrzymanie tej wiadomości było dla Ciebie niespodziewane?** {unexpected}
-**Dodatkowe uwagi lub informacje:** {additional_notes}
-                """
-
-                # Pobierz klucz API
-                api_key = os.getenv('SAMBANOVA_API_KEY')
-                if not api_key:
-                    st.error(translations['invalid_api_key'])
+                elif language == 'German':
+                    st.warning("Folgende potenzielle Warnsignale wurden gefunden:")
                 else:
-                    # Analiza za pomocą modelu AI
-                    with st.spinner("Analizuję wiadomość przy użyciu modelu AI..."):
-                        analysis_text, risk_text, recommendations_text = analyze_message(message, phone_number, additional_info, api_key)
-                        st.subheader("📄 Analiza treści wiadomości:")
+                    st.warning("The following potential red flags were found:")
+                for warn in warnings:
+                    st.write(f"- {warn}")
+    
+            # Zbierz dodatkowe informacje
+            additional_info = f"""
+**How often do you receive messages from this number?** {frequency}
+**When did you last receive a similar message?** {last_received}
+**How confident are you that the message is authentic?** {confidence_level}/5
+**Was receiving this message unexpected for you?** {unexpected}
+**Additional notes or information:** {additional_notes}
+            """
+    
+            # Pobierz klucz API
+            api_key = os.getenv('SAMBANOVA_API_KEY')
+            if not api_key:
+                st.error(translations['error_no_api_key'])
+            else:
+                # Analiza za pomocą modelu AI
+                with st.spinner(translations['analyze_button'] + "..."):
+                    analysis_text, risk_text, recommendations_text = analyze_message(message, phone_number, additional_info, api_key)
+                    
+                    if language == 'Polish':
+                        st.subheader("📑 Analiza treści wiadomości:")
                         st.markdown(analysis_text)
-                        st.subheader("⚖️ Ocena ryzyka oszustwa:")
+    
+                        st.subheader("⚠️ Ocena ryzyka oszustwa:")
                         st.markdown(risk_text)
+    
                         st.subheader("✅ Zalecenia dla użytkownika:")
                         st.markdown(recommendations_text)
-
-                        # Aktualizacja statystyk na podstawie oceny ryzyka
-                        try:
-                            # Zakładając, że risk_text zawiera np. "Ryzyko: 7/10"
-                            # Możesz dostosować parsowanie w zależności od formatu
-                            match = re.search(r'(\d+)/10', risk_text)
-                            if match:
-                                risk_score = int(match.group(1))
-                                fraud_detected = risk_score >= 7  # Przykładowy próg
-                                update_stats(fraud_detected=fraud_detected)
-                            else:
-                                fraud_detected = False
-                        except:
+                    elif language == 'German':
+                        st.subheader("📑 Analyse des Nachrichtentextes:")
+                        st.markdown(analysis_text)
+    
+                        st.subheader("⚠️ Bewertung des Betrugsrisikos:")
+                        st.markdown(risk_text)
+    
+                        st.subheader("✅ Empfehlungen für den Benutzer:")
+                        st.markdown(recommendations_text)
+                    else:
+                        st.subheader("📑 Message Content Analysis:")
+                        st.markdown(analysis_text)
+    
+                        st.subheader("⚠️ Fraud Risk Assessment:")
+                        st.markdown(risk_text)
+    
+                        st.subheader("✅ Recommendations for the User:")
+                        st.markdown(recommendations_text)
+    
+                    # Aktualizacja statystyk na podstawie oceny ryzyka
+                    try:
+                        # Zakładając, że risk_text zawiera np. "Ryzyko: 7/10" lub podobne w innych językach
+                        # Możesz dostosować parsowanie w zależności od formatu
+                        match = re.search(r'(\d+)/10', risk_text)
+                        if match:
+                            risk_score = int(match.group(1))
+                            fraud_detected = risk_score >= 7  # Przykładowy próg
+                            update_stats(fraud_detected=fraud_detected)
+                        else:
                             fraud_detected = False
-
-                        # Dodanie analizy do historii
-                        add_to_history(message, phone_number, analysis_text, risk_text, recommendations_text)
-
+                    except:
+                        fraud_detected = False
+    
+                    # Dodanie analizy do historii
+                    add_to_history(message, phone_number, analysis_text, risk_text, recommendations_text)
+    
+                # Zgłoś numer jako oszustwo, jeśli fraud_detected
+                if fraud_detected:
+                    if st.button(translations['report_button']):
+                        success = add_fake_number(phone_number)
+                        if success:
+                            if language == 'Polish':
+                                st.success(translations['success_reported'])
+                            elif language == 'German':
+                                st.success(translations['success_reported'])
+                            else:
+                                st.success(translations['success_reported'])
+                        else:
+                            if language == 'Polish':
+                                st.info(translations['info_already_reported'])
+                            elif language == 'German':
+                                st.info(translations['info_already_reported'])
+                            else:
+                                st.info(translations['info_already_reported'])
+    
+                if language == 'Polish':
+                    st.info(translations['info_reminder'])
+                elif language == 'German':
+                    st.info(translations['info_reminder'])
+                else:
                     st.info(translations['info_reminder'])
+    
+    # Funkcja do sprawdzenia, czy numer jest fałszywy (dodatkowa funkcja)
+    def is_fake_number(phone_number):
+        try:
+            with open('fake_numbers.json', 'r') as f:
+                fake_numbers = json.load(f)
+            return any(entry["phone_number"] == phone_number for entry in fake_numbers)
+        except (json.JSONDecodeError, FileNotFoundError):
+            return False
 
-    # Wyświetlanie historii analiz
-    st.subheader(translations['history_header'])
-    history = get_history()
-    if history:
-        for entry in reversed(history[-5:]):  # Wyświetl ostatnie 5 analiz
-            with st.expander(f"{entry['timestamp']} - {entry['phone_number']}"):
-                st.write(f"**Wiadomość:** {entry['message']}")
-                st.write(f"**Analiza:** {entry['analysis']}")
-                st.write(f"**Ocena Ryzyka:** {entry['risk_assessment']}")
-                st.write(f"**Zalecenia:** {entry['recommendations']}")
-    else:
-        st.info(translations['no_history'])
-        
+    # Nie dodawaj "if __name__ == '__main__':" w podstronach
+