Update pages/Analysis.py

pages/Analysis.py

@@ -1,3 +1,5 @@
+# pages/Analysis.py
+
 import streamlit as st
 from utils.functions import (
     get_phone_info,
@@ -188,6 +190,16 @@ def analyze_website(url, language):
     st.write(get_translation('website_analysis_in_progress', language))
     # Tutaj możesz dodać logikę analizy stron internetowych
 
+def analyze_image(image_file, language):
+    st.subheader(get_translation('image_analysis_header', language))
+    st.write(get_translation('image_analysis_in_progress', language))
+    # Tutaj możesz dodać logikę analizy obrazów
+
+def analyze_phone_number(phone_number, language):
+    st.subheader(get_translation('phone_number_analysis_header', language))
+    st.write(get_translation('phone_number_analysis_in_progress', language))
+    # Tutaj możesz dodać logikę analizy numerów telefonów
+
 def show_history(language):
     st.subheader(get_translation('history_header', language))
     st.write(get_translation('history_in_progress', language))
@@ -201,11 +213,33 @@ def show_statistics(language):
 def get_translation(key, language):
     translations = {
         'Polish': {
+            'analysis_center_title': "Centrum Analizy",
+            'select_analysis_type': "Wybierz typ analizy:",
+            'sms_analysis': "Analiza SMS",
+            'email_analysis': "Analiza E-mail",
+            'website_analysis': "Analiza Strony Internetowej",
+            'phone_number_analysis': "Analiza Numeru Telefonu",
+            'image_analysis': "Analiza Obrazu",
+            'history': "Historia Analiz",
+            'statistics': "Statystyki",
             '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_number_analysis_header': "Analiza Numeru Telefonu",
+            'image_analysis_header': "Analiza Obrazu",
+            'enter_sms_content': "Wprowadź treść SMS",
+            'sender_phone_number': "Numer telefonu nadawcy (z kodem kierunkowym kraju, np. +48123456789)",
+            'analyze_button': "Analizuj",
+            'enter_email_content': "Wprowadź treść e-maila",
+            'sender_email_address': "Adres e-mail nadawcy",
+            'enter_website_url': "Wprowadź adres URL strony do analizy",
+            'upload_image': "Prześlij obraz do analizy",
+            'no_image_selected': "Proszę wybrać obraz do analizy.",
+            'enter_phone_number': "Wprowadź numer telefonu do analizy",
+            'email_analysis_in_progress': "Funkcja analizy e-maili jest w trakcie implementacji.",
+            'website_analysis_in_progress': "Funkcja analizy stron internetowych jest w trakcie implementacji.",
+            'image_analysis_in_progress': "Funkcja analizy obrazów jest w trakcie implementacji.",
+            'phone_number_analysis_in_progress': "Funkcja analizy numerów telefonów jest w trakcie implementacji.",
             'phone_info': "Informacje o numerze telefonu:",
             'country_label': "Kraj",
             'operator_label': "Operator",
@@ -229,126 +263,37 @@ def get_translation(key, language):
             '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
-        }
+        # Dodaj tłumaczenia dla innych języków, jeśli potrzebujesz
     }
-    return translations.get(language, translations['English']).get(key, key)
+    return translations.get(language, translations['Polish']).get(key, key)
 
 def show_analysis(language):
     st.title(get_translation('analysis_center_title', language))
-    
-    analysis_type = st.sidebar.radio(
+
+    analysis_options = [
+        get_translation('sms_analysis', language),
+        get_translation('email_analysis', language),
+        get_translation('website_analysis', language),
+        get_translation('phone_number_analysis', language),
+        get_translation('image_analysis', language),
+        get_translation('history', language),
+        get_translation('statistics', language)
+    ]
+
+    selected_analysis = st.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)
-        )
+        analysis_options
     )
 
-    if analysis_type == get_translation('sms_analysis_header', language):
+    if selected_analysis == get_translation('sms_analysis', 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(
-                get_translation('frequency_label', language),
-                get_translation('frequency_options', language)
-            )
-            last_received = st.selectbox(
-                get_translation('last_received_label', language),
-                get_translation('last_received_options', language)
-            )
-            confidence_level = st.slider(
-                get_translation('confidence_level_label', language),
-                1, 5, 3,
-                help=get_translation('confidence_level_help', language)
-            )
-            unexpected = st.radio(
-                get_translation('unexpected_label', language),
-                get_translation('unexpected_options', language)
-            )
-            additional_notes = st.text_area(get_translation('additional_notes_label', language), height=100)
+            # Dodaj dodatkowe pola i informacje, jeśli potrzebujesz
 
             submitted = st.form_submit_button(get_translation('analyze_button', language))
 
@@ -358,19 +303,14 @@ def show_analysis(language):
             elif not phone_number.strip():
                 st.error(get_translation('sender_phone_number', language))
             else:
-                additional_info = f"""
-**{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}
-                """
+                additional_info = ""  # Dodaj dodatkowe informacje, jeśli potrzebujesz
                 analyze_sms(message, phone_number, additional_info, language)
 
-    elif analysis_type == get_translation('email_analysis_header', language):
+    elif selected_analysis == get_translation('email_analysis', 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))
 
         if submitted:
@@ -381,9 +321,10 @@ def show_analysis(language):
             else:
                 analyze_email(email_content, sender_email, language)
 
-    elif analysis_type == get_translation('website_analysis_header', language):
+    elif selected_analysis == get_translation('website_analysis', 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))
 
         if submitted:
@@ -392,8 +333,32 @@ def show_analysis(language):
             else:
                 analyze_website(url, language)
 
-    elif analysis_type == get_translation('history_header', language):
+    elif selected_analysis == get_translation('phone_number_analysis', language):
+        with st.form("phone_number_form"):
+            phone_number = st.text_input(get_translation('enter_phone_number', language))
+
+            submitted = st.form_submit_button(get_translation('analyze_button', language))
+
+        if submitted:
+            if not phone_number.strip():
+                st.error(get_translation('enter_phone_number', language))
+            else:
+                analyze_phone_number(phone_number, language)
+
+    elif selected_analysis == get_translation('image_analysis', language):
+        with st.form("image_form"):
+            image_file = st.file_uploader(get_translation('upload_image', language), type=['png', 'jpg', 'jpeg'])
+
+            submitted = st.form_submit_button(get_translation('analyze_button', language))
+
+        if submitted:
+            if image_file is None:
+                st.error(get_translation('no_image_selected', language))
+            else:
+                analyze_image(image_file, language)
+
+    elif selected_analysis == get_translation('history', language):
         show_history(language)
 
-    elif analysis_type == get_translation('statistics_header', language):
+    elif selected_analysis == get_translation('statistics', language):
         show_statistics(language)