From 32ea31144ee75d8c04334ca6964d49d6fd266e94 Mon Sep 17 00:00:00 2001 From: Timothy Jaeryang Baek Date: Wed, 14 May 2025 22:00:42 +0400 Subject: [PATCH] refac: file upload handling behaviour --- backend/open_webui/routers/files.py | 56 ++++++++++++++--------------- 1 file changed, 28 insertions(+), 28 deletions(-) diff --git a/backend/open_webui/routers/files.py b/backend/open_webui/routers/files.py index d37455415..ba589070f 100644 --- a/backend/open_webui/routers/files.py +++ b/backend/open_webui/routers/files.py @@ -125,37 +125,37 @@ def upload_file( ) if process: try: + if file.content_type: + if file.content_type.startswith( + ( + "audio/mpeg", + "audio/wav", + "audio/ogg", + "audio/x-m4a", + "audio/webm", + "video/webm", + ) + ): + file_path = Storage.get_file(file_path) + result = transcribe(request, file_path) - if file.content_type.startswith( - ( - "audio/mpeg", - "audio/wav", - "audio/ogg", - "audio/x-m4a", - "audio/webm", - "video/webm", - ) - ): - file_path = Storage.get_file(file_path) - result = transcribe(request, file_path) - - process_file( - request, - ProcessFileForm(file_id=id, content=result.get("text", "")), - user=user, - ) - elif file.content_type not in [ - "image/png", - "image/jpeg", - "image/gif", - "video/mp4", - "video/ogg", - "video/quicktime", - ]: - process_file(request, ProcessFileForm(file_id=id), user=user) + process_file( + request, + ProcessFileForm(file_id=id, content=result.get("text", "")), + user=user, + ) + elif file.content_type not in [ + "image/png", + "image/jpeg", + "image/gif", + "video/mp4", + "video/ogg", + "video/quicktime", + ]: + process_file(request, ProcessFileForm(file_id=id), user=user) else: log.info( - f"File type {file.content_type} is not supported for processing, but trying to process anyway" + f"File type {file.content_type} is not provided, but trying to process anyway" ) process_file(request, ProcessFileForm(file_id=id), user=user)