Update pages/Analysis.py

pages/Analysis.py

@@ -1,26 +1,145 @@
 # pages/Analysis.py
 
 import streamlit as st
+from utils.functions import (
+    analyze_message,
+    get_phone_info,
+    simple_checks,
+    add_fake_number,
+    is_fake_number,
+    update_stats,
+    add_to_history
+)
 
 # 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 poniżej, aby sprawdzić, czy jest ona potencjalnym oszustwem.""",
+        'input_placeholder': "Wprowadź treść SMS...",
+        'analyze_button': "Analizuj",
+        '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"
     },
     '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 unten ein, um zu überprüfen, ob es sich um einen potenziellen Betrugsversuch handelt.""",
+        'input_placeholder': "Geben Sie den SMS-Inhalt ein...",
+        'analyze_button': "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"
     },
     '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 content of the message below to check if it is a potential fraud attempt.""",
+        'input_placeholder': "Enter SMS content...",
+        'analyze_button': "Analyze",
+        '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"
     }
 }
 
 def main(language):
     st.header(page_translations[language]['header'])
     st.write(page_translations[language]['content'])
-    # Tutaj dodaj resztę funkcjonalności analizy SMS
 
-# Do not include "if __name__ == '__main__':" block in pages
+    # Input box for SMS content
+    sms_content = st.text_area(
+        label="",
+        placeholder=page_translations[language]['input_placeholder'],
+        height=150
+    )
+
+    # 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.")
+        else:
+            # Pobierz numer telefonu (przykładowy, możesz dodać input)
+            phone_number = st.text_input("Numer telefonu nadawcy:", value="", max_chars=15)
+
+            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.")
+                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.subheader("📄 Analiza Treści")
+                    st.write(analysis)
+
+                    st.subheader("⚖️ Ocena Ryzyka")
+                    st.write(risk)
+
+                    st.subheader("✅ Zalecenia")
+                    st.write(recommendations)
+
+                    # 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)
+
+                    # Dodanie do historii
+                    add_to_history(sms_content, phone_number, analysis, risk, recommendations)
+
+                    # 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.")
+
+    # Wyświetlanie historii analiz
+    st.subheader(page_translations[language]['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:
+        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