UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-06 08:56:39 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

refac: chat input file handling

Browse Source
This commit is contained in:
Timothy J. Baek 2024-10-26 12:56:37 -07:00
parent 032abba88e
commit 15cfdc69a8
3 changed files with 22 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
backend/open_webui/apps/webui/models/files.py
View File

@ -73,6 +73,8 @@ class FileModelResponse(BaseModel):
created_at: int # timestamp in epoch
updated_at: int # timestamp in epoch
model_config = ConfigDict(extra="allow")
class FileMetadataResponse(BaseModel):
id: str

8
backend/open_webui/apps/webui/routers/files.py
View File

@ -38,7 +38,7 @@ router = APIRouter()
############################
@router.post("/")
@router.post("/", response_model=FileModelResponse)
def upload_file(file: UploadFile = File(...), user=Depends(get_verified_user)):
log.info(f"file.content_type: {file.content_type}")
try:
@ -73,6 +73,12 @@ def upload_file(file: UploadFile = File(...), user=Depends(get_verified_user)):
except Exception as e:
log.exception(e)
log.error(f"Error processing file: {file_item.id}")
file_item = FileModelResponse(
**{
**file_item.model_dump(),
"error": e,
}
)
if file_item:
return file_item

17
src/lib/components/chat/MessageInput.svelte
View File

@ -1,5 +1,7 @@
<script lang="ts">
import { toast } from 'svelte-sonner';
import { v4 as uuidv4 } from 'uuid';
import { onMount, tick, getContext, createEventDispatcher, onDestroy } from 'svelte';
const dispatch = createEventDispatcher();
@ -89,6 +91,7 @@
const uploadFileHandler = async (file) => {
console.log(file);
const tempItemId = uuidv4();
const fileItem = {
type: 'file',
file: '',
@ -98,10 +101,16 @@
collection_name: '',
status: 'uploading',
size: file.size,
error: ''
error: '',
itemId: tempItemId
};
files = [...files, fileItem];
if (fileItem.size == 0) {
toast.error($i18n.t('You cannot upload an empty file.'));
return null;
}
files = [...files, fileItem];
// Check if the file is an audio file and transcribe/convert it to text file
if (['audio/mpeg', 'audio/wav', 'audio/ogg', 'audio/x-m4a'].includes(file['type'])) {
const res = await transcribeAudio(localStorage.token, file).catch((error) => {
@ -132,11 +141,11 @@
files = files;
} else {
files = files.filter((item) => item.status !== null);
files = files.filter((item) => item?.itemId !== tempItemId);
}
} catch (e) {
toast.error(e);
files = files.filter((item) => item.status !== null);
files = files.filter((item) => item?.itemId !== tempItemId);
}
};