Update pages/Analysis.py

pages/Analysis.py

@@ -2,12 +2,13 @@
 
 import streamlit as st
 from utils.functions import (
+    analyze_message,
     get_phone_info,
     simple_checks,
-    analyze_message,
+    add_fake_number,
+    is_fake_number,
     update_stats,
-    add_to_history,
-    is_fake_number  # Dodany import
+    add_to_history
 )
 import os
 import re  # Dodany import do obsługi wyrażeń regularnych
@@ -103,74 +104,77 @@ def show_analysis(language):
             'remember_note': "📌 Remember: Always exercise caution when receiving messages from unknown senders. Do not share confidential information and do not click on suspicious links."
         }
     }
-    
+
     def show_analysis(language):
-        st.header(page_translations[language]['header'])
-        st.write(page_translations[language]['content'])
-    
+        # Pobierz tłumaczenia dla wybranego języka
+        pt = page_translations.get(language, page_translations['Polish'])
+
+        st.header(pt['header'])
+        st.write(pt['content'])
+
         with st.form("sms_form"):
             message = st.text_area(
-                page_translations[language]['message_label'],
-                placeholder=page_translations[language]['message_label'],
+                pt['message_label'],
+                placeholder=pt['message_label'],
                 height=150
             )
             phone_number = st.text_input(
-                page_translations[language]['phone_label'],
+                pt['phone_label'],
                 placeholder="+48123123123"
             )
-    
-            st.markdown("### " + " ".join(page_translations[language]['frequency_label'].split()))
+
+            st.markdown("### " + pt['frequency_label'])
             frequency = st.selectbox(
-                page_translations[language]['frequency_label'],
-                page_translations[language]['frequency_options']
+                pt['frequency_label'],
+                pt['frequency_options']
             )
             last_received = st.selectbox(
-                page_translations[language]['last_received_label'],
-                page_translations[language]['last_received_options']
+                pt['last_received_label'],
+                pt['last_received_options']
             )
             confidence_level = st.slider(
-                page_translations[language]['confidence_label'],
+                pt['confidence_label'],
                 1, 5, 3,
-                help=page_translations[language]['confidence_help']
+                help=pt['confidence_help']
             )
             unexpected = st.radio(
-                page_translations[language]['unexpected_label'],
-                page_translations[language]['unexpected_options']
+                pt['unexpected_label'],
+                pt['unexpected_options']
             )
             additional_notes = st.text_area(
-                page_translations[language]['additional_notes_label'],
+                pt['additional_notes_label'],
                 height=100
             )
-    
-            submitted = st.form_submit_button(page_translations[language]['analyze_button'])
-    
+
+            submitted = st.form_submit_button(pt['analyze_button'])
+
         if submitted:
             if not message.strip():
-                st.error(page_translations[language]['missing_message'])
+                st.error(pt['missing_message'])
             elif not phone_number.strip():
-                st.error(page_translations[language]['missing_phone'])
+                st.error(pt['missing_phone'])
             else:
                 # Informacje o numerze telefonu
                 country, operator = get_phone_info(phone_number)
                 if country and operator:
-                    st.write(f"**{ 'Country' if language == 'English' else ('Kraj' if language == 'Polish' else 'Land') }:** {country}")
-                    st.write(f"**{ 'Operator' if language == 'English' else ('Operator' if language == 'Polish' else 'Betreiber') }:** {operator}")
+                    st.write(f"**{'Country' if language == 'English' else ('Kraj' if language == 'Polish' else 'Land')}:** {country}")
+                    st.write(f"**{'Operator' if language == 'English' else ('Operator' if language == 'Polish' else 'Betreiber')}:** {operator}")
                 else:
-                    st.write(page_translations[language]['invalid_phone'])
-    
+                    st.error(pt['invalid_phone'])
+
                 # Sprawdzenie, czy numer jest w bazie fałszywych numerów
                 if is_fake_number(phone_number):
-                    st.warning(page_translations[language]['warning_fraud'])
+                    st.warning(pt['warning_fraud'])
                 else:
-                    st.info(page_translations[language]['safe_message'])
-    
+                    st.info(pt['safe_message'])
+
                 # 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}")
-    
+
                 # Zbierz dodatkowe informacje
                 additional_info = f"""
 **How often do you receive messages from this number?** {frequency}
@@ -179,24 +183,24 @@ def show_analysis(language):
 **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(page_translations[language]['api_key_missing'])
+                    st.error(pt['api_key_missing'])
                 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("📑 " + ("Message Content Analysis:" if language == 'English' else ("Analiza treści wiadomości:" if language == 'Polish' else "Analyse des Nachrichtentextes:")))
                         st.markdown(analysis_text)
-    
+
                         st.subheader("⚠️ " + ("Fraud Risk Assessment:" if language == 'English' else ("Ocena ryzyka oszustwa:" if language == 'Polish' else "Betrugsrisikobewertung:")))
                         st.markdown(risk_text)
-    
+
                         st.subheader("✅ " + ("User Recommendations:" if language == 'English' else ("Zalecenia dla użytkownika:" if language == 'Polish' else "Benutzerempfehlungen:")))
                         st.markdown(recommendations_text)
-    
+
                         # Aktualizacja statystyk na podstawie oceny ryzyka
                         try:
                             # Zakładając, że risk_text zawiera np. "Ryzyko: 7/10"
@@ -210,32 +214,31 @@ def show_analysis(language):
                                 fraud_detected = False
                         except:
                             fraud_detected = False
-    
+
                         # Dodanie analizy do historii
                         add_to_history(message, phone_number, analysis_text, risk_text, recommendations_text)
-    
-                    st.info(page_translations[language]['remember_note'])
-    
+
+                    st.info(pt['remember_note'])
+
                 # Zgłoś numer jako oszustwo
                 if fraud_detected:
-                    if st.button(page_translations[language]['report_button']):
+                    if st.button(pt['report_button']):
                         success = add_fake_number(phone_number)
                         if success:
-                            st.success(page_translations[language]['reported_success'])
+                            st.success(pt['reported_success'])
                         else:
-                            st.info(page_translations[language]['reported_exists'])
-    
+                            st.info(pt['reported_exists'])
+
         # Wyświetlanie historii analiz
-        st.subheader(page_translations[language]['history_header'])
+        st.subheader(pt['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"**{ 'Message' if language == 'English' else ('Wiadomość' if language == 'Polish' else 'Nachricht') }:** {entry['message']}")
-                    st.write(f"**{ 'Analysis' if language == 'English' else ('Analiza' if language == 'Polish' else 'Analyse') }:** {entry['analysis']}")
-                    st.write(f"**{ 'Risk Assessment' if language == 'English' else ('Ocena Ryzyka' if language == 'Polish' else 'Risikobewertung') }:** {entry['risk_assessment']}")
-                    st.write(f"**{ 'Recommendations' if language == 'English' else ('Zalecenia' if language == 'Polish' else 'Empfehlungen') }:** {entry['recommendations']}")
+                    st.write(f"**{'Message' if language == 'English' else ('Wiadomość' if language == 'Polish' else 'Nachricht')}:** {entry['message']}")
+                    st.write(f"**{'Analysis' if language == 'English' else ('Analiza' if language == 'Polish' else 'Analyse')}:** {entry['analysis']}")
+                    st.write(f"**{'Risk Assessment' if language == 'English' else ('Ocena Ryzyka' if language == 'Polish' else 'Risikobewertung')}:** {entry['risk_assessment']}")
+                    st.write(f"**{'Recommendations' if language == 'English' else ('Zalecenia' if language == 'Polish' else 'Empfehlungen')}:** {entry['recommendations']}")
         else:
-            st.info(page_translations[language]['no_history'])
-
-# Usuń blok "if __name__ == '__main__':" z podstron
+            st.info(pt['no_history'])
+