diff --git a/src/lib/components/chat/MessageInput.svelte b/src/lib/components/chat/MessageInput.svelte index 5ea541c7f..2fa835acd 100644 --- a/src/lib/components/chat/MessageInput.svelte +++ b/src/lib/components/chat/MessageInput.svelte @@ -98,6 +98,7 @@ const uploadFileHandler = async (file) => { console.log(file); + // Check if the file is an audio file and transcribe/convert it to text file if (['audio/mpeg', 'audio/wav'].includes(file['type'])) { const res = await transcribeAudio(localStorage.token, file).catch((error) => { @@ -112,40 +113,49 @@ } } - // Upload the file to the server - const uploadedFile = await uploadFile(localStorage.token, file).catch((error) => { - toast.error(error); - return null; - }); + const fileItem = { + type: 'file', + file: '', + id: null, + url: '', + name: file.name, + collection_name: '', + status: '', + size: file.size, + error: '' + }; + files = [...files, fileItem]; - if (uploadedFile) { - const fileItem = { - type: 'file', - file: uploadedFile, - id: uploadedFile.id, - url: `${WEBUI_API_BASE_URL}/files/${uploadedFile.id}`, - name: file.name, - collection_name: '', - status: 'uploaded', - error: '' - }; - files = [...files, fileItem]; + try { + const uploadedFile = await uploadFile(localStorage.token, file); - // TODO: Check if tools & functions have files support to skip this step to delegate file processing - // Default Upload to VectorDB - if ( - SUPPORTED_FILE_TYPE.includes(file['type']) || - SUPPORTED_FILE_EXTENSIONS.includes(file.name.split('.').at(-1)) - ) { - processFileItem(fileItem); + if (uploadedFile) { + fileItem.status = 'uploaded'; + fileItem.file = uploadedFile; + fileItem.id = uploadedFile.id; + fileItem.url = `${WEBUI_API_BASE_URL}/files/${uploadedFile.id}`; + + // TODO: Check if tools & functions have files support to skip this step to delegate file processing + // Default Upload to VectorDB + if ( + SUPPORTED_FILE_TYPE.includes(file['type']) || + SUPPORTED_FILE_EXTENSIONS.includes(file.name.split('.').at(-1)) + ) { + processFileItem(fileItem); + } else { + toast.error( + $i18n.t(`Unknown file type '{{file_type}}'. Proceeding with the file upload anyway.`, { + file_type: file['type'] + }) + ); + processFileItem(fileItem); + } } else { - toast.error( - $i18n.t(`Unknown file type '{{file_type}}'. Proceeding with the file upload anyway.`, { - file_type: file['type'] - }) - ); - processFileItem(fileItem); + files = files.filter((item) => item.status !== null); } + } catch (e) { + toast.error(e); + files = files.filter((item) => item.status !== null); } }; @@ -162,7 +172,6 @@ // Remove the failed doc from the files array // files = files.filter((f) => f.id !== fileItem.id); toast.error(e); - fileItem.status = 'processed'; files = files; }