Update pages/Analysis.py

pages/Analysis.py

@@ -2,12 +2,13 @@
 
 import streamlit as st
 from utils.functions import (
-    get_phone_info, 
-    simple_checks, 
-    analyze_message, 
-    update_stats, 
-    add_to_history, 
-    is_fake_number
+    get_phone_info,
+    simple_checks,
+    analyze_message,
+    update_stats,
+    add_to_history,
+    is_fake_number,
+    add_fake_number  # Upewnij się, że ta funkcja jest zaimplementowana
 )
 import os
 import re
@@ -35,7 +36,17 @@ def show_analysis(language):
             '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)"
+            '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."
         },
         'German': {
             'header': "📊 SMS-Analyse",
@@ -56,7 +67,17 @@ def show_analysis(language):
             'confidence_level_help': "1: Sehr unsicher, 5: Sehr sicher",
             'unexpected_label': "War der Empfang dieser Nachricht für Sie unerwartet?",
             'unexpected_options': ('Ja', 'Nein'),
-            'additional_notes_label': "Zusätzliche Bemerkungen oder Informationen (optional)"
+            'additional_notes_label': "Zusätzliche Bemerkungen 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 rote Flaggen wurden gefunden:",
+            'remember_caution': "📌 Denken Sie daran: Seien Sie vorsichtig mit 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. Bitte stellen Sie sicher, dass sie im richtigen Format ist."
         },
         'English': {
             'header': "📊 SMS Analysis",
@@ -77,7 +98,17 @@ def show_analysis(language):
             '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)"
+            '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: Stay 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. Please ensure it is in the correct format."
         }
     }
 
@@ -88,14 +119,14 @@ def show_analysis(language):
 
     with st.form("sms_form"):
         message = st.text_area(
-            label=translations['content'],  # Non-empty label for accessibility
+            label=translations['input_placeholder'],
             placeholder=translations['input_placeholder'], 
             height=150
         )
         phone_number = st.text_input(
             label=translations['phone_input_label'],
             value="", 
-            max_chars=20  # Zwiększenie limitu znaków dla numerów międzynarodowych
+            max_chars=20
         )
         
         st.markdown(f"### {translations['additional_info_header']}")
@@ -136,11 +167,11 @@ def show_analysis(language):
             # Informacje o numerze telefonu
             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}")
+                st.subheader(translations['phone_info_header'])
+                st.write(f"- **{translations['country_label']}:** {country}")
+                st.write(f"- **{translations['operator_label']}:** {operator}")
             else:
-                st.write("Unable to retrieve phone number information.")
+                st.write(translations['unable_to_retrieve'])
 
             # Sprawdzenie, czy numer jest w bazie fałszywych numerów
             if is_fake_number(phone_number):
@@ -151,41 +182,26 @@ def show_analysis(language):
                 fraud_detected = False
 
             # Przeprowadź proste sprawdzenia heurystyczne
-            warnings = simple_checks(message, language)  # Przekazujemy argument language
+            warnings = simple_checks(message, language)
             if warnings:
-                if language == 'Polish':
-                    st.warning("Znaleziono następujące potencjalne czerwone flagi:")
-                elif language == 'German':
-                    st.warning("Folgende potenzielle rote Flaggen wurden gefunden:")
-                else:
-                    st.warning("The following potential red flags were found:")
+                st.warning(translations['red_flags_found'])
                 for warn in warnings:
                     st.write(f"- {warn}")
 
             # Zbierz dodatkowe informacje
             additional_info = f"""
-**{ 'Jak często otrzymujesz wiadomości od tego numeru?' if language == 'Polish' else 
-   'Wie oft erhalten Sie Nachrichten von dieser Nummer?' if language == 'German' else 
-   'How often do you receive messages from this number?' }** {frequency}
+**{translations['frequency_label']}** {frequency}
 
-**{ 'Kiedy ostatnio otrzymałeś podobną wiadomość?' if language == 'Polish' else 
-   'Wann haben Sie zuletzt eine ähnliche Nachricht erhalten?' if language == 'German' else 
-   'When did you last receive a similar message?' }** {last_received}
+**{translations['last_received_label']}** {last_received}
 
-**{ 'Jak bardzo jesteś pewny, że wiadomość jest autentyczna?' if language == 'Polish' else 
-   'Wie sicher sind Sie, dass die Nachricht authentisch ist?' if language == 'German' else 
-   'How confident are you that the message is authentic?' }** {confidence_level}/5
+**{translations['confidence_level_label']}** {confidence_level}/5
 
-**{ 'Czy otrzymanie tej wiadomości było dla Ciebie niespodziewane?' if language == 'Polish' else 
-   'War der Empfang dieser Nachricht für Sie unerwartet?' if language == 'German' else 
-   'Was receiving this message unexpected for you?' }** {unexpected}
+**{translations['unexpected_label']}** {unexpected}
 
-**{ 'Dodatkowe uwagi lub informacje:' if language == 'Polish' else 
-   'Zusätzliche Bemerkungen oder Informationen:' if language == 'German' else 
-   'Additional notes or information:' }** {additional_notes}
+**{translations['additional_notes_label']}** {additional_notes}
             """
 
-            # Pobierz klucz API z Secret
+            # 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.")
@@ -212,7 +228,7 @@ def show_analysis(language):
                     # Dodanie analizy do historii
                     add_to_history(message, phone_number, analysis_text, risk_text, recommendations_text)
 
-                st.info("📌 Remember: Stay cautious with messages from unknown senders.")
+                st.info(translations['remember_caution'])
 
             # Zgłoszenie numeru jako oszustwo
             if fraud_detected:
@@ -222,11 +238,11 @@ def show_analysis(language):
                         if is_valid_number(parsed_number):
                             success = add_fake_number(phone_number)
                             if success:
-                                st.success("Number has been reported as fraud.")
+                                st.success(translations['number_reported'])
                             else:
-                                st.info("Number has already been reported as fraud.")
+                                st.info(translations['number_already_reported'])
                         else:
-                            st.error("The provided phone number is invalid.")
+                            st.error(translations['invalid_number'])
                     except NumberParseException:
-                        st.error("Failed to parse the phone number. Please ensure it is in the correct format.")
+                        st.error(translations['failed_to_parse'])