Update app.py

app.py

@@ -15,6 +15,9 @@ from io import BytesIO
 
 import uuid
 
+from cryptography.fernet import Fernet
+from cryptography.hazmat.primitives import hashes
+from cryptography.hazmat.primitives.kdf.pbkdf2 import PBKDF2HMAC
 
 
 
@@ -950,10 +953,13 @@ def send_message():
     message_text = data.get('message', '').strip()
     
     if message_text:
+        # Шифруем сообщение перед сохранением
+        encrypted_text = encrypt_message(message_text)
+        
         message = {
             'id': len(messages) + 1,
             'user_id': user_id,
-            'text': message_text,
+            'encrypted_text': encrypted_text,  # Сохраняем зашифрованный текст
             'timestamp': datetime.now().strftime("%H:%M:%S"),
             'username': f"User_{user_id}"
         }
@@ -968,13 +974,35 @@ def send_message():
 
 @app.route('/get_messages')
 def get_messages():
-    return jsonify({'messages': messages})
+    # Расшифровываем сообщения перед отправкой клиенту
+    decrypted_messages = []
+    for msg in messages:
+        decrypted_msg = msg.copy()
+        decrypted_msg['text'] = decrypt_message(msg['encrypted_text'])
+        decrypted_messages.append(decrypted_msg)
+    
+    return jsonify({'messages': decrypted_messages})
 
 @app.route('/get_new_messages')
 def get_new_messages():
     last_id = request.args.get('last_id', 0, type=int)
     new_messages = [msg for msg in messages if msg['id'] > last_id]
-    return jsonify({'messages': new_messages})
+    
+    # Расшифровываем новые сообщения
+    decrypted_messages = []
+    for msg in new_messages:
+        decrypted_msg = msg.copy()
+        decrypted_msg['text'] = decrypt_message(msg['encrypted_text'])
+        decrypted_messages.append(decrypted_msg)
+    
+    return jsonify({'messages': decrypted_messages})
+
+@app.route('/clear_chat', methods=['POST'])
+def clear_chat():
+    """Очистка всех сообщений"""
+    global messages
+    messages.clear()
+    return jsonify({'status': 'success', 'message': 'Чат очищен'})