From b087f32690c4738e4cfd811c08ee5156db37a92d Mon Sep 17 00:00:00 2001
From: Lucas <lbometon@hotmail.fr>
Date: Wed, 8 Jan 2025 16:30:41 +0100
Subject: [PATCH] Update MessageBox.tsx

CTA link doc + web
---
 ui/components/MessageBox.tsx | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/ui/components/MessageBox.tsx b/ui/components/MessageBox.tsx
index 5d4980c..83d8d83 100644
--- a/ui/components/MessageBox.tsx
+++ b/ui/components/MessageBox.tsx
@@ -54,10 +54,16 @@ const MessageBox = ({
         message.content.replace(
           regex,
           (_, number) => {
-            const url = message.sources?.[number - 1]?.metadata?.url || '';
-            // Extraire le nom de domaine sans l'extension
-            const sourceName = url.replace(/^(?:https?:\/\/)?(?:www\.)?([^./]+).*$/, '$1');
-            return `<a href="${url}" target="_blank" class="ml-2 px-3 py-1 text-xs bg-blue-500 hover:bg-blue-600 text-white rounded-md transition-colors duration-200 no-underline inline-flex items-center">${sourceName}</a>`;
+            const source = message.sources?.[number - 1];
+            const url = source?.metadata?.url || '';
+            const isDocument = source?.metadata?.isFile;
+            
+            // Utiliser "Voir la source" pour les documents, sinon le nom de domaine
+            const linkText = isDocument 
+              ? "Voir la source"
+              : url.replace(/^(?:https?:\/\/)?(?:www\.)?([^./]+).*$/, '$1');
+            
+            return `<a href="${url}" target="_blank" class="ml-2 px-3 py-1 text-xs bg-blue-500 hover:bg-blue-600 text-white rounded-md transition-colors duration-200 no-underline inline-flex items-center">${linkText}</a>`;
           }
         ),
       );