diff --git a/koboldcpp.py b/koboldcpp.py index 6754bce3a..a69011cdb 100644 --- a/koboldcpp.py +++ b/koboldcpp.py @@ -2743,8 +2743,8 @@ class KcppServerRequestHandler(http.server.SimpleHTTPRequestHandler): if boundary: fparts = body.split(boundary) for fpart in fparts: - detected_upload_filename = re.findall(r'Content-Disposition[^;]*;\s*name=(?:"file"|file)\s*;\s*filename="([^"]+)"', fpart.decode('utf-8',errors='ignore'),flags=re.IGNORECASE) - detected_upload_filename_comfy = re.findall(r'Content-Disposition[^;]*;\s*name=(?:"image"|image)\s*;\s*filename="([^"]+)""', fpart.decode('utf-8',errors='ignore'),flags=re.IGNORECASE) + detected_upload_filename = re.findall(r'Content-Disposition[^;]*;\s*name=(?:"file"|file)\s*;\s*filename=(?:"([^"]+)"|([^\s";]+))', fpart.decode('utf-8',errors='ignore'),flags=re.IGNORECASE) + detected_upload_filename_comfy = re.findall(r'Content-Disposition[^;]*;\s*name=(?:"image"|image)\s*;\s*filename=(?:"([^"]+)"|([^\s";]+))', fpart.decode('utf-8',errors='ignore'),flags=re.IGNORECASE) if detected_upload_filename and len(detected_upload_filename)>0: utfprint(f"Detected uploaded file: {detected_upload_filename[0]}") file_content_start = fpart.find(b'\r\n\r\n') + 4 # Position after headers