diff --git a/src/pages/chat/chat.ts b/src/pages/chat/chat.ts
index a466ef6..2501e77 100755
--- a/src/pages/chat/chat.ts
+++ b/src/pages/chat/chat.ts
@@ -396,7 +396,8 @@ class ChatElement extends HTMLElement {
                 sender: myAddresses[0],
                 text: messageText,
                 time: formattedTime,
-                type: 'text' as const
+                type: 'text',
+                class: 'message user'
             };
 
             if (this.selectedMemberId) {
@@ -538,7 +539,7 @@ class ChatElement extends HTMLElement {
                 }
 
                 const messageContent = document.createElement('div');
-                messageContent.className = 'message';
+                messageContent.className = message.class || 'message';
                 
                 if (message.type === 'file') {
                     messageContent.innerHTML = `
@@ -861,8 +862,7 @@ class ChatElement extends HTMLElement {
 
     // Send a file
     private async sendFile(file: File) {
-        // Ajouter une vérification de la taille avant la conversion en base64
-        const MAX_FILE_SIZE = 1 * 1024 * 1024; // Réduire à 1MB pour éviter les problèmes de quota
+        const MAX_FILE_SIZE = 1 * 1024 * 1024; 
         if (file.size > MAX_FILE_SIZE) {
             alert('Le fichier est trop volumineux. Taille maximum : 1MB');
             return;
@@ -874,7 +874,6 @@ class ChatElement extends HTMLElement {
                 throw new Error('No paired member found');
             }
 
-            // Compresser l'image si c'est une image
             let fileData: string;
             if (file.type.startsWith('image/')) {
                 fileData = await this.compressImage(file);
@@ -882,17 +881,16 @@ class ChatElement extends HTMLElement {
                 fileData = await this.readFileAsBase64(file);
             }
 
-            // Créer un message avec un texte descriptif au lieu du fileData
             const newMessage = {
                 id: Date.now(),
                 sender: myAddresses[0],
-                text: `Fichier envoyé: ${file.name}`, // Ajouter un texte descriptif
+                text: `Fichier envoyé: ${file.name}`,
                 fileName: file.name,
                 time: new Date().toLocaleString('fr-FR'),
-                type: 'file' as const
+                type: 'file',
+                class: 'message user'
             };
 
-            // Stocker le fileData séparément
             try {
                 const fileKey = `file_${newMessage.id}`;
                 localStorage.setItem(fileKey, fileData);
@@ -907,7 +905,6 @@ class ChatElement extends HTMLElement {
                 this.messagesMock = messageStore.getMessages();
             }
 
-            // Récupérer le process_id du parent (conversation)
             const groupItem = this.shadowRoot?.querySelector('[data-process-id]');
             const parentProcessId = groupItem?.getAttribute('data-process-id');