diff --git a/crates/service/job/inference_job/src/job/job_types/vc/rvc_v2/process_job.rs b/crates/service/job/inference_job/src/job/job_types/vc/rvc_v2/process_job.rs index 19a8d08b7a..f1c0c0d7c7 100644 --- a/crates/service/job/inference_job/src/job/job_types/vc/rvc_v2/process_job.rs +++ b/crates/service/job/inference_job/src/job/job_types/vc/rvc_v2/process_job.rs @@ -3,6 +3,7 @@ use std::time::Instant; use anyhow::anyhow; use log::{error, info, warn}; +use buckets::public::media_files::original_file::MediaFileBucketPath; use buckets::public::media_uploads::original_file::MediaUploadOriginalFilePath; use buckets::public::voice_conversion_results::original_file::VoiceConversionResultOriginalFilePath; @@ -12,13 +13,14 @@ use container_common::filesystem::safe_delete_temp_file::safe_delete_temp_file; use enums::by_table::generic_inference_jobs::inference_result_type::InferenceResultType; use filesys::create_dir_all_if_missing::create_dir_all_if_missing; use filesys::file_size::file_size; +use hashing::sha256::sha256_hash_file::sha256_hash_file; use media::decode_basic_audio_info::decode_basic_audio_file_info; use mimetypes::mimetype_for_file::get_mimetype_for_file; use mysql_queries::payloads::generic_inference_args::generic_inference_args::PolymorphicInferenceArgs; use mysql_queries::queries::generic_inference::job::list_available_generic_inference_jobs::AvailableInferenceJob; +use mysql_queries::queries::media_files::insert_media_file_from_voice_conversion::{insert_media_file_from_voice_conversion, InsertMediaFileArgs, VoiceConversionModelType}; use mysql_queries::queries::media_uploads::get_media_upload_for_inference::MediaUploadRecordForInference; use mysql_queries::queries::voice_conversion::inference::get_voice_conversion_model_for_inference::VoiceConversionModelForInference; -use mysql_queries::queries::voice_conversion::results::insert_voice_conversion_result::{insert_voice_conversion_result, InsertArgs}; use tokens::tokens::media_uploads::MediaUploadToken; use tokens::tokens::users::UserToken; @@ -28,6 +30,9 @@ use crate::job::job_types::vc::rvc_v2::rvc_v2_inference_command::InferenceArgs; use crate::job_dependencies::JobDependencies; use crate::util::maybe_download_file_from_bucket::maybe_download_file_from_bucket; +const BUCKET_FILE_PREFIX : &str = "fakeyou_rvc_"; +const BUCKET_FILE_EXTENSION : &str = ".wav"; + pub struct RvcV2ProcessJobArgs<'a> { pub job_dependencies: &'a JobDependencies, pub job: &'a AvailableInferenceJob, @@ -267,8 +272,6 @@ pub async fn process_job(args: RvcV2ProcessJobArgs<'_>) -> Result) -> Result) -> Result) -> Result) -> Result