Update utils/functions.py

utils/functions.py

@@ -189,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')  # Zmieniono na 'en' dla uniwersalności
-        operator = carrier.name_for_number(parsed_number, 'en')
+        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'
         logging.info(f"Numer {phone_number} - Kraj: {country}, Operator: {operator}.")
         return country, operator
     except phonenumbers.NumberParseException as e:
@@ -212,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("Message contains keywords related to potential fraud.")
+        warnings.append("Wiadomość zawiera słowa kluczowe związane z potencjalnym oszustwem.")
     if re.search(r'http[s]?://', message):
-        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.")
+        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.")
     return warnings
 
 def analyze_message(message, phone_number, additional_info, api_key, language):
@@ -225,39 +225,40 @@ def analyze_message(message, phone_number, additional_info, api_key, language):
     """
     if not api_key:
         logging.error("Brak klucza API.")
-        return "API key is missing.", "API key is missing.", "API key is missing."
+        return "Brak klucza API.", "Brak klucza API.", "Brak klucza API."
 
     url = "https://api.sambanova.ai/v1/chat/completions"
     headers = {
-        "Authorization": f"Bearer {api_key}"
+        "Authorization": f"Bearer {api_key}",
+        "Content-Type": "application/json"
     }
 
     # System prompts w trzech językach
     system_prompts = {
         'Polish': """
-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:
+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:
 
 <analysis>
-**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.
+**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.
 </analysis>
 
 <risk_assessment>
-**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.
+**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ę.
 </risk_assessment>
 
 <recommendations>
-**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.
+**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.
 </recommendations>
 
-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.
+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.
         """,
         '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:
@@ -324,7 +325,7 @@ Additional Information:
 Provide your analysis and conclusions following the guidelines above."""
 
     payload = {
-        "model": "Meta-Llama-3.1-8B-Instruct",
+        "model": "Meta-Llama-3.1-8B-Instruct",  # Upewnij się, że to poprawny model API
         "messages": [
             {"role": "system", "content": system_prompt},
             {"role": "user", "content": user_prompt}