Update utils/functions.py

utils/functions.py

@@ -35,7 +35,7 @@ def load_json(file_path):
     """Ładuje dane z pliku JSON. Jeśli plik nie istnieje, zwraca pustą listę lub domyślny obiekt."""
     if not os.path.exists(file_path):
         if file_path.endswith('stats.json'):
-            return {"total": 0, "fraud": 0, "safe": 0}
+            return {"total_analyses": 0, "total_frauds_detected": 0}
         else:
             return []
     with open(file_path, 'r', encoding='utf-8') as file:
@@ -45,7 +45,7 @@ def load_json(file_path):
         except json.JSONDecodeError:
             logging.error(f"Nie można załadować danych z {file_path}. Plik jest uszkodzony.")
             if file_path.endswith('stats.json'):
-                return {"total": 0, "fraud": 0, "safe": 0}
+                return {"total_analyses": 0, "total_frauds_detected": 0}
             return []
 
 def save_json(file_path, data):
@@ -84,21 +84,18 @@ def get_fake_numbers():
     fake_numbers = load_json(FAKE_NUMBERS_FILE)
     return fake_numbers
 
-def add_to_history(content, sender_info, analysis_text, risk_text, recommendations_text):
+def add_to_history(message, sender_info, analysis_text, risk_text, recommendations_text):
     """
     Dodaje wpis do historii analiz w pliku history.json.
     """
     history = load_json(HISTORY_FILE)
     history.append({
-        "date": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
-        "type": "Analysis",
-        "details": {
-            "content": content,
-            "sender_info": sender_info,
-            "analysis": analysis_text,
-            "risk_assessment": risk_text,
-            "recommendations": recommendations_text
-        }
+        "timestamp": datetime.now().isoformat(),
+        "message": message,
+        "sender_info": sender_info,
+        "analysis": analysis_text,
+        "risk_assessment": risk_text,
+        "recommendations": recommendations_text
     })
     save_json(HISTORY_FILE, history)
     logging.info(f"Dodano wpis do history.json dla nadawcy {sender_info}.")
@@ -111,18 +108,22 @@ def get_history():
     logging.info("Historia analiz została pobrana pomyślnie.")
     return history
 
+def get_analysis_history():
+    """
+    Alias funkcji get_history dla zgodności z innymi modułami.
+    """
+    return get_history()
+
 def update_stats(fraud_detected=False):
     """
     Aktualizuje statystyki analiz w pliku stats.json.
     """
     stats = load_json(STATS_FILE)
-    stats["total"] += 1
+    stats["total_analyses"] += 1
     if fraud_detected:
-        stats["fraud"] += 1
-    else:
-        stats["safe"] += 1
+        stats["total_frauds_detected"] += 1
     save_json(STATS_FILE, stats)
-    logging.info(f"Statystyki zostały zaktualizowane: Analiz {stats['total']}, Oszustw {stats['fraud']}.")
+    logging.info(f"Statystyki zostały zaktualizowane: Analiz {stats['total_analyses']}, Oszustw {stats['total_frauds_detected']}.")
 
 def get_stats():
     """
@@ -190,79 +191,79 @@ def analyze_message(content, sender_info, additional_info, api_key, language):
 
     system_prompts = {
         'Polish': """
-Jesteś zaawansowanym asystentem AI specjalizującym się w identyfikacji fałszywych wiadomości. Twoim zadaniem jest przeprowadzenie szczegółowej analizy poniższej wiadomości, wykorzystując głęboki proces myślenia i dostarczając kompleksową ocenę. Twoja odpowiedź powinna być podzielona na trzy sekcje:
-
-<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.
-</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ę.
-</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.
-</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.
+    Jesteś zaawansowanym asystentem AI specjalizującym się w identyfikacji fałszywych wiadomości. Twoim zadaniem jest przeprowadzenie szczegółowej analizy poniższej wiadomości, wykorzystując głęboki proces myślenia i dostarczając kompleksową ocenę. Twoja odpowiedź powinna być podzielona na trzy sekcje:
+
+    <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.
+    </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ę.
+    </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.
+    </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.
         """,
         'German': """
-Du bist ein fortgeschrittener KI-Assistent, spezialisiert auf die Identifizierung gefälschter Nachrichten. Deine Aufgabe ist es, eine detaillierte Analyse der folgenden Nachricht durchzuführen, indem du einen tiefgreifenden Denkprozess nutzt und eine umfassende Bewertung lieferst. Deine Antwort sollte in drei Abschnitte unterteilt sein:
-
-<analysis>
-**Nachrichteninhaltsanalyse:**
-- Führe eine detaillierte Analyse des Nachrichteninhalts durch und identifiziere potenzielle rote Flaggen wie sprachliche Fehler, Aufforderungen zur Preisgabe persönlicher Daten, dringende Kontaktanfragen usw.
-- Beschreibe den sprachlichen und kulturellen Kontext der Nachricht.
-- Identifiziere alle Elemente, die darauf hindeuten könnten, dass die Nachricht ein Versuch ist, Informationen oder Geld zu erlangen.
-</analysis>
-
-<risk_assessment>
-**Betrugsrisikobewertung:**
-- Basierend auf der Inhaltsanalyse und den verfügbaren Informationen, bewerte die Wahrscheinlichkeit, dass die Nachricht ein Betrug ist. Verwende eine Skala von 1 bis 10, wobei 1 sehr geringes Risiko und 10 sehr hohes Risiko bedeutet.
-- Erkläre, welche Faktoren diese Bewertung beeinflussen.
-</risk_assessment>
-
-<recommendations>
-**Empfehlungen für den Benutzer:**
-- Gib klare und konkrete Empfehlungen zu den nächsten Schritten, die der Benutzer unternehmen sollte.
-- Berücksichtige Sicherheitsempfehlungen wie das Blockieren des Absenders, das Melden der Nachricht an entsprechende Behörden oder das Ignorieren der Nachricht.
-- Wenn möglich, schlage zusätzliche Vorsichtsmaßnahmen vor, die der Benutzer ergreifen kann, um seine persönlichen und finanziellen Daten zu schützen.
-</recommendations>
-
-Deine Antwort sollte genau nach den oben genannten Richtlinien formatiert sein und die Markierungen <analysis>, <risk_assessment> und <recommendations> verwenden. Stelle sicher, dass jeder Abschnitt vollständig und detailliert ausgefüllt ist.
+    Du bist ein fortgeschrittener KI-Assistent, spezialisiert auf die Identifizierung gefälschter Nachrichten. Deine Aufgabe ist es, eine detaillierte Analyse der folgenden Nachricht durchzuführen, indem du einen tiefgreifenden Denkprozess nutzt und eine umfassende Bewertung lieferst. Deine Antwort sollte in drei Abschnitte unterteilt sein:
+
+    <analysis>
+    **Nachrichteninhaltsanalyse:**
+    - Führe eine detaillierte Analyse des Nachrichteninhalts durch und identifiziere potenzielle rote Flaggen wie sprachliche Fehler, Aufforderungen zur Preisgabe persönlicher Daten, dringende Kontaktanfragen usw.
+    - Beschreibe den sprachlichen und kulturellen Kontext der Nachricht.
+    - Identifiziere alle Elemente, die darauf hindeuten könnten, dass die Nachricht ein Versuch ist, Informationen oder Geld zu erlangen.
+    </analysis>
+
+    <risk_assessment>
+    **Betrugsrisikobewertung:**
+    - Basierend auf der Inhaltsanalyse und den verfügbaren Informationen, bewerte die Wahrscheinlichkeit, dass die Nachricht ein Betrug ist. Verwende eine Skala von 1 bis 10, wobei 1 sehr geringes Risiko und 10 sehr hohes Risiko bedeutet.
+    - Erkläre, welche Faktoren diese Bewertung beeinflussen.
+    </risk_assessment>
+
+    <recommendations>
+    **Empfehlungen für den Benutzer:**
+    - Gib klare und konkrete Empfehlungen zu den nächsten Schritten, die der Benutzer unternehmen sollte.
+    - Berücksichtige Sicherheitsempfehlungen wie das Blockieren des Absenders, das Melden der Nachricht an entsprechende Behörden oder das Ignorieren der Nachricht.
+    - Wenn möglich, schlage zusätzliche Vorsichtsmaßnahmen vor, die der Benutzer ergreifen kann, um seine persönlichen und finanziellen Daten zu schützen.
+    </recommendations>
+
+    Deine Antwort sollte genau nach den oben genannten Richtlinien formatiert sein und die Markierungen <analysis>, <risk_assessment> und <recommendations> verwenden. Stelle sicher, dass jeder Abschnitt vollständig und detailliert ausgefüllt ist.
         """,
         'English': """
-You are an advanced AI assistant specializing in identifying fake messages. Your task is to conduct a detailed analysis of the following message, utilizing a deep thinking process and providing a comprehensive assessment. Your response should be divided into three sections:
-
-<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.
-</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.
-</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.
-</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.
+    You are an advanced AI assistant specializing in identifying fake messages. Your task is to conduct a detailed analysis of the following message, utilizing a deep thinking process and providing a comprehensive assessment. Your response should be divided into three sections:
+
+    <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.
+    </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.
+    </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.
+    </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.
         """
     }
 
@@ -360,3 +361,4 @@ def extract_text_from_image(image_file):
     except Exception as e:
         logging.error(f"Błąd podczas ekstrakcji tekstu z obrazu: {e}")
         return f"Błąd podczas ekstrakcji tekstu z obrazu: {e}"
+