Update pages/Analysis.py

pages/Analysis.py

@@ -8,124 +8,175 @@ from utils.functions import (
     add_fake_number,
     is_fake_number,
     update_stats,
-    add_to_history
+    add_to_history,
+    get_history
 )
+import os
+import re  # Dodany import do obsługi wyrażeń regularnych
 
 # 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.""",
+        '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...",
-        'analyze_button': "Analizuj",
+        'phone_input_label': "Numer telefonu nadawcy (w formacie międzynarodowym, 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"
+        'history_header': "Historia Analiz",
+        '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."
     },
     'German': {
         'header': "📊 SMS-Analyse",
-        'content': """Hier befindet sich die Benutzeroberfläche zur Analyse von SMS-Nachrichten. Geben Sie den Text der Nachricht unten ein, um zu überprüfen, ob es sich um einen potenziellen Betrugsversuch handelt.""",
+        '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...",
-        'analyze_button': "Analysieren",
+        'phone_input_label': "Telefonnummer des Absenders (im internationalen Format, 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"
+        '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."
     },
     '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 attempt.""",
+        '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...",
-        'analyze_button': "Analyze",
+        'phone_input_label': "Sender's Phone Number (in international format, 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",
-        'history_header': "Analysis History"
+        'report_button': "Report Number as Fraud",
+        'history_header': "Analysis History",
+        '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."
     }
 }
 
-def main(language):
-    st.header(page_translations[language]['header'])
-    st.write(page_translations[language]['content'])
+def show_analysis(language):
+    translations = page_translations.get(language, page_translations['Polish'])  # Domyślnie Polish
+    st.header(translations['header'])
+    st.write(translations['content'])
 
-    # Input box for SMS content
-    sms_content = st.text_area(
-        label="",
-        placeholder=page_translations[language]['input_placeholder'],
-        height=150
-    )
+    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'])
+        
+        st.markdown("### Dodatkowe Informacje")
+        # Dodane pytania
+        frequency = st.selectbox(
+            "Jak często otrzymujesz wiadomości od tego numeru?",
+            ('Pierwszy raz', 'Sporadycznie', 'Regularnie')
+        )
+        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')
+        )
+        confidence_level = st.slider(
+            "Jak bardzo jesteś pewny, że wiadomość jest autentyczna?",
+            1, 5, 3,
+            help="1: Bardzo mało pewny, 5: Bardzo pewny"
+        )
+        unexpected = st.radio(
+            "Czy otrzymanie tej wiadomości było dla Ciebie niespodziewane?",
+            ('Tak', 'Nie')
+        )
+        additional_notes = st.text_area("Dodatkowe uwagi lub informacje (opcjonalnie)", height=100)
+        
+        submitted = st.form_submit_button(translations['analyze_button'])
 
-    # Analyze button
-    if st.button(page_translations[language]['analyze_button']):
-        if sms_content.strip() == "":
-            if language == 'Polish':
-                st.warning("Proszę wprowadzić treść wiadomości.")
-            elif language == 'German':
-                st.warning("Bitte geben Sie den Nachrichtentext ein.")
-            else:
-                st.warning("Please enter the message content.")
+    if submitted:
+        if not message.strip():
+            st.error(translations['error_message_empty'])
+        elif not phone_number.strip():
+            st.error(translations['error_phone_empty'])
         else:
-            # Pobierz numer telefonu (przykładowy, możesz dodać input)
-            phone_number = st.text_input("Numer telefonu nadawcy:", value="", max_chars=15)
+            # Weryfikacja numeru telefonu
+            country, operator = get_phone_info(phone_number)
+            if not country:
+                st.error(translations['invalid_phone'])
+            else:
+                st.write(f"**Informacje o numerze telefonu:**")
+                st.write(f"- **Kraj:** {country}")
+                st.write(f"- **Operator:** {operator}")
 
-            if phone_number:
-                # Weryfikacja numeru telefonu
-                country, operator = get_phone_info(phone_number)
-                if not country:
-                    if language == 'Polish':
-                        st.error("Nieprawidłowy numer telefonu.")
-                    elif language == 'German':
-                        st.error("Ungültige Telefonnummer.")
-                    else:
-                        st.error("Invalid phone number.")
+                # Sprawdzenie, czy numer jest w bazie fałszywych numerów
+                if is_fake_number(phone_number):
+                    st.warning(translations['already_reported'])
                 else:
-                    # Proste sprawdzenia heurystyczne
-                    warnings = simple_checks(sms_content)
-                    for warning in warnings:
-                        st.warning(warning)
-                    
-                    # Analiza wiadomości
-                    additional_info = f"Kraj: {country}, Operator: {operator}"
-                    analysis, risk, recommendations = analyze_message(sms_content, phone_number, additional_info, os.getenv('SAMBANOVA_API_KEY'))
+                    st.info("Numer telefonu nie znajduje się w bazie fałszywych numerów.")
 
-                    st.subheader("📄 Analiza Treści")
-                    st.write(analysis)
+                # Przeprowadź proste sprawdzenia heurystyczne
+                warnings = simple_checks(message)
+                if warnings:
+                    st.warning("Znaleziono następujące potencjalne czerwone flagi:")
+                    for warn in warnings:
+                        st.write(f"- {warn}")
 
-                    st.subheader("⚖️ Ocena Ryzyka")
-                    st.write(risk)
+                # 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}
+                """
 
-                    st.subheader("✅ Zalecenia")
-                    st.write(recommendations)
+                # Pobierz klucz API
+                api_key = os.getenv('SAMBANOVA_API_KEY')
+                if not api_key:
+                    st.error(translations['invalid_api_key'])
+                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.markdown(analysis_text)
+                        st.subheader("⚖️ Ocena ryzyka oszustwa:")
+                        st.markdown(risk_text)
+                        st.subheader("✅ Zalecenia dla użytkownika:")
+                        st.markdown(recommendations_text)
 
-                    # Aktualizacja statystyk
-                    fraud_detected = "Wiadomość może być oszustwem" in analysis or "Betrug" in analysis or "fraud" in analysis
-                    update_stats(fraud_detected=fraud_detected)
+                        # 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:
+                            fraud_detected = False
 
-                    # Dodanie do historii
-                    add_to_history(sms_content, phone_number, analysis, risk, recommendations)
+                        # Dodanie analizy do historii
+                        add_to_history(message, phone_number, analysis_text, risk_text, recommendations_text)
 
-                    # Zgłoś numer jako oszustwo
-                    if fraud_detected:
-                        if st.button(page_translations[language]['report_button']):
-                            success = add_fake_number(phone_number)
-                            if success:
-                                if language == 'Polish':
-                                    st.success("Numer został zgłoszony jako oszustwo.")
-                                elif language == 'German':
-                                    st.success("Nummer wurde als Betrug gemeldet.")
-                                else:
-                                    st.success("Number reported as fraud.")
-                            else:
-                                if language == 'Polish':
-                                    st.info("Numer został już zgłoszony jako oszustwo.")
-                                elif language == 'German':
-                                    st.info("Nummer wurde bereits als Betrug gemeldet.")
-                                else:
-                                    st.info("Number has already been reported as fraud.")
+                    st.info(translations['info_reminder'])
 
     # Wyświetlanie historii analiz
-    st.subheader(page_translations[language]['history_header'])
+    st.subheader(translations['history_header'])
     history = get_history()
     if history:
         for entry in reversed(history[-5:]):  # Wyświetl ostatnie 5 analiz
@@ -135,11 +186,5 @@ def main(language):
                 st.write(f"**Ocena Ryzyka:** {entry['risk_assessment']}")
                 st.write(f"**Zalecenia:** {entry['recommendations']}")
     else:
-        if language == 'Polish':
-            st.info("Brak historii analiz.")
-        elif language == 'German':
-            st.info("Keine Analysehistorie vorhanden.")
-        else:
-            st.info("No analysis history available.")
-
-# Nie dodawaj "if __name__ == '__main__':" w podstronach
+        st.info(translations['no_history'])
+