// Функция вызывается для каждого загружаемого файла
function custom_pre_upload( $file ) {
    
    // переменная для передачи названия директории
    global $directory;
    
    // получение типа передаваемого файла
    $type = $file[ 'type' ];
    
    // названия директорий по типу файла
    $directories = array(
        'text/plain' => '/files',
        'image/jpeg' => '/images',
    );
    
    // проверка $directories на наличие названия директории по типу файла
    if ( isset( $directories[ $type ] ) ) {        
        // запись названия директории
        $directory = $directories[ $type ];        
    } else {
        // сообщаем, что по типу файла директория не найдена
        $directory = false;
    }
        
    // установка фильтра для изменения директории  
    add_filter( 'upload_dir', 'custom_upload_dir' );
        
    return $file;
}
add_filter( 'wp_handle_upload_prefilter', 'custom_upload_dir' );    


// функция изменения директории
function custom_upload_dir( $upload ) {
    
    // переменная с названием нужной директории
    global $directory;
    
    // проверка на наличие названия директории
    if ( $directory ) {        
        $upload['subdir'] .= $directory;
        $upload['path'] .= $directory;
        $upload['url'] .= $directory;        
    }
    
    return $upload;
}