Update utils/functions.py

utils/functions.py

@@ -61,7 +61,6 @@ def add_fake_number(phone_number):
     """
     db = SessionLocal()
     try:
-        logging.info(f"Próbuję dodać numer: {phone_number}")
         existing_number = db.query(FakeNumber).filter(FakeNumber.phone_number == phone_number).first()
         if not existing_number:
             new_number = FakeNumber(phone_number=phone_number)
@@ -190,8 +189,8 @@ def get_phone_info(phone_number):
     """
     try:
         parsed_number = phonenumbers.parse(phone_number, None)
-        country = geocoder.description_for_number(parsed_number, 'en')  # Możesz zmienić język na 'pl' lub 'de'
-        operator = carrier.name_for_number(parsed_number, 'en')        # Możesz zmienić język na 'pl' lub 'de'
+        country = geocoder.description_for_number(parsed_number, 'en')  # Zmieniono na 'en' dla uniwersalności
+        operator = carrier.name_for_number(parsed_number, 'en')
         logging.info(f"Numer {phone_number} - Kraj: {country}, Operator: {operator}.")
         return country, operator
     except phonenumbers.NumberParseException as e:
@@ -213,11 +212,11 @@ def simple_checks(message, language):
     selected_keywords = scam_keywords.get(language, scam_keywords['English'])
 
     if any(keyword in message.lower() for keyword in selected_keywords):
-        warnings.append("Wiadomość zawiera słowa kluczowe związane z potencjalnym oszustwem.")
+        warnings.append("Message contains keywords related to potential fraud.")
     if re.search(r'http[s]?://', message):
-        warnings.append("Wiadomość zawiera link.")
-    if re.search(r'\b(podaj|prześlij|udostępnij)\b.*\b(hasło|kod|dane osobowe|numer konta)\b', message.lower()):
-        warnings.append("Wiadomość zawiera prośbę o poufne informacje.")
+        warnings.append("Message contains a link.")
+    if re.search(r'\b(provide|send|share)\b.*\b(password|code|personal information|account number)\b', message.lower()):
+        warnings.append("Message contains a request for confidential information.")
     return warnings
 
 def analyze_message(message, phone_number, additional_info, api_key, language):
@@ -226,7 +225,7 @@ def analyze_message(message, phone_number, additional_info, api_key, language):
     """
     if not api_key:
         logging.error("Brak klucza API.")
-        return "Brak klucza API.", "Brak klucza API.", "Brak klucza API."
+        return "API key is missing.", "API key is missing.", "API key is missing."
 
     url = "https://api.sambanova.ai/v1/chat/completions"
     headers = {
@@ -236,29 +235,29 @@ def analyze_message(message, phone_number, additional_info, api_key, language):
     # System prompts w trzech językach
     system_prompts = {
         'Polish': """
-Jesteś zaawansowanym asystentem AI specjalizującym się w identyfikacji fałszywych wiadomości SMS. Twoim zadaniem jest przeprowadzenie szczegółowej analizy wiadomości, wykorzystując głęboki proces myślenia i dostarczając kompleksową ocenę. Twoja odpowiedź powinna być podzielona na trzy sekcje:
+You are an advanced AI assistant specializing in identifying fake SMS messages. Your task is to conduct a detailed analysis of the message, utilizing a deep thinking process and providing a comprehensive assessment. Your response should be divided into three sections:
 
 <analysis>
-**Analiza Treści Wiadomości:**
-- Przeprowadź szczegółową analizę treści wiadomości, identyfikując potencjalne czerwone flagi, takie jak błędy językowe, prośby o dane osobowe, pilne prośby o kontakt itp.
-- Opisz kontekst językowy i kulturowy wiadomości.
-- Zidentyfikuj wszelkie elementy, które mogą sugerować, że wiadomość jest próbą wyłudzenia informacji lub pieniędzy.
+**Message Content Analysis:**
+- Conduct a detailed analysis of the message content, identifying potential red flags such as language errors, requests for personal information, urgent contact requests, etc.
+- Describe the linguistic and cultural context of the message.
+- Identify any elements that may suggest the message is an attempt to solicit information or money.
 </analysis>
 
 <risk_assessment>
-**Ocena Ryzyka Oszustwa:**
-- Na podstawie analizy treści i dostępnych informacji oceń prawdopodobieństwo, że wiadomość jest oszustwem. Użyj skali od 1 do 10, gdzie 1 oznacza bardzo niskie ryzyko, a 10 bardzo wysokie ryzyko.
-- Wyjaśnij, jakie czynniki wpływają na tę ocenę.
+**Fraud Risk Assessment:**
+- Based on the content analysis and available information, assess the likelihood that the message is fraudulent. Use a scale from 1 to 10, where 1 indicates very low risk and 10 indicates very high risk.
+- Explain the factors that influence this assessment.
 </risk_assessment>
 
 <recommendations>
-**Zalecenia dla Użytkownika:**
-- Podaj jasne i konkretne zalecenia dotyczące dalszych kroków, które użytkownik powinien podjąć.
-- Uwzględnij sugestie dotyczące bezpieczeństwa, takie jak blokowanie nadawcy, zgłaszanie wiadomości do odpowiednich instytucji, czy też ignorowanie wiadomości.
-- Jeśli to możliwe, zasugeruj dodatkowe środki ostro��ności, które użytkownik może podjąć, aby chronić swoje dane osobowe i finansowe.
+**User Recommendations:**
+- Provide clear and concrete recommendations regarding the next steps the user should take.
+- Include security suggestions such as blocking the sender, reporting the message to appropriate authorities, or ignoring the message.
+- If possible, suggest additional precautionary measures the user can take to protect their personal and financial information.
 </recommendations>
 
-Twoja odpowiedź powinna być sformatowana dokładnie w powyższy sposób, używając znaczników <analysis>, <risk_assessment> i <recommendations>. Upewnij się, że każda sekcja jest wypełniona kompletnie i szczegółowo.
+Your response should be formatted exactly as specified above, using the <analysis>, <risk_assessment>, and <recommendations> tags. Ensure that each section is thoroughly and comprehensively filled out.
         """,
         'German': """
 Du bist ein fortgeschrittener KI-Assistent, spezialisiert auf die Identifizierung gefälschter SMS-Nachrichten. Deine Aufgabe ist es, eine detaillierte Analyse der Nachricht durchzuführen, indem du einen tiefgreifenden Denkprozess nutzt und eine umfassende Bewertung lieferst. Deine Antwort sollte in drei Abschnitte unterteilt sein: