Update utils/functions.py

utils/functions.py

@@ -73,6 +73,13 @@ def is_fake_number(phone_number):
     logging.info(f"Sprawdzanie numeru {phone_number}: {'znaleziony' if exists else 'nie znaleziony'}.")
     return exists
 
+def get_fake_numbers():
+    """
+    Pobiera listę fałszywych numerów z pliku fake_numbers.json.
+    """
+    fake_numbers = load_json(FAKE_NUMBERS_FILE)
+    return fake_numbers
+
 def add_to_history(message, phone_number, analysis, risk, recommendations):
     """
     Dodaje wpis do historii analiz w pliku history.json.
@@ -132,7 +139,7 @@ def get_phone_info(phone_number):
         return country, operator
     except phonenumbers.NumberParseException as e:
         logging.error(f"Nie udało się przetworzyć numeru telefonu {phone_number}: {e}")
-        return None, None
+        return "Nieznany", "Nieznany"
 
 def simple_checks(message, language):
     """
@@ -150,11 +157,11 @@ def simple_checks(message, language):
 
     message_lower = message.lower()
 
-    if any(keyword in message_lower for keyword in selected_keywords):
+    if any(keyword.lower() in message_lower for keyword in selected_keywords):
         warnings.append("Wiadomość zawiera słowa kluczowe związane z potencjalnym oszustwem.")
     if re.search(r'http[s]?://', message):
         warnings.append("Wiadomość zawiera link.")
-    if re.search(r'\b(podaj|prześlij|udostępnij|sende|übermittle|teile)\b.*\b(hasło|kod|dane osobowe|numer konta|Passwort|Code|persönliche Daten|Kontonummer)\b', message_lower):
+    if re.search(r'\b(podaj|prześlij|udostępnij|sende|übermittle|teile|send|provide|share)\b.*\b(hasło|kod|dane osobowe|numer konta|Passwort|Code|persönliche Daten|Kontonummer|password|code|personal information|account number)\b', message_lower):
         warnings.append("Wiadomość zawiera prośbę o poufne informacje.")
     return warnings
 
@@ -250,7 +257,7 @@ Your response should be formatted exactly as specified above, using the <analysi
         """
     }
 
-    system_prompt = system_prompts.get(language, system_prompts['English'])  # Default to English if language not found
+    system_prompt = system_prompts.get(language, system_prompts['English'])  # Domyślnie angielski, jeśli język nie jest obsługiwany
 
     user_prompt = f"""Analyze the following message for potential fraud:
 
@@ -284,15 +291,14 @@ Provide your analysis and conclusions following the guidelines above."""
             risk_assessment = re.search(r'<risk_assessment>(.*?)</risk_assessment>', ai_response, re.DOTALL)
             recommendations = re.search(r'<recommendations>(.*?)</recommendations>', ai_response, re.DOTALL)
 
-            analysis_text = analysis.group(1).strip() if analysis else "No analysis available."
-            risk_text = risk_assessment.group(1).strip() if risk_assessment else "No risk assessment available."
-            recommendations_text = recommendations.group(1).strip() if recommendations else "No recommendations available."
+            analysis_text = analysis.group(1).strip() if analysis else "Brak analizy."
+            risk_text = risk_assessment.group(1).strip() if risk_assessment else "Brak oceny ryzyka."
+            recommendations_text = recommendations.group(1).strip() if recommendations else "Brak zaleceń."
 
             return analysis_text, risk_text, recommendations_text
         else:
-            logging.error(f"API Error: {response.status_code} - {response.text}")
-            return f"API Error: {response.status_code} - {response.text}", "Analysis Error.", "Analysis Error."
+            logging.error(f"Błąd API: {response.status_code} - {response.text}")
+            return f"Błąd API: {response.status_code} - {response.text}", "Błąd analizy.", "Błąd analizy."
     except Exception as e:
-        logging.error(f"API Connection Error: {e}")
-        return f"API Connection Error: {e}", "Analysis Error.", "Analysis Error."
-        
+        logging.error(f"Błąd połączenia z API: {e}")
+        return f"Błąd połączenia z API: {e}", "Błąd analizy.", "Błąd analizy."