diff --git a/crates/lib/mysql_queries/src/queries/media_files/create/insert_media_file_from_face_animation.rs b/crates/lib/mysql_queries/src/queries/media_files/create/insert_media_file_from_face_animation.rs index 12a302e2f6..4dd2a97048 100644 --- a/crates/lib/mysql_queries/src/queries/media_files/create/insert_media_file_from_face_animation.rs +++ b/crates/lib/mysql_queries/src/queries/media_files/create/insert_media_file_from_face_animation.rs @@ -3,6 +3,7 @@ use sqlx; use sqlx::MySqlPool; use enums::by_table::generic_synthetic_ids::id_category::IdCategory; +use enums::by_table::media_files::media_file_class::MediaFileClass; use enums::by_table::media_files::media_file_origin_category::MediaFileOriginCategory; use enums::by_table::media_files::media_file_origin_model_type::MediaFileOriginModelType; use enums::by_table::media_files::media_file_origin_product_category::MediaFileOriginProductCategory; @@ -66,7 +67,6 @@ pub async fn insert_media_file_from_face_animation( const ORIGIN_CATEGORY : MediaFileOriginCategory = MediaFileOriginCategory::Inference; const ORIGIN_PRODUCT_CATEGORY : MediaFileOriginProductCategory = MediaFileOriginProductCategory::FaceAnimator; const ORIGIN_MODEL_TYPE : MediaFileOriginModelType = MediaFileOriginModelType::SadTalker; - const MEDIA_TYPE : MediaFileType = MediaFileType::Video; let record_id = { let query_result = sqlx::query!( @@ -75,11 +75,13 @@ INSERT INTO media_files SET token = ?, + media_class = ?, + media_type = ?, + origin_category = ?, origin_product_category = ?, maybe_origin_model_type = ?, - media_type = ?, maybe_mime_type = ?, file_size_bytes = ?, @@ -105,11 +107,13 @@ SET "#, result_token.as_str(), + MediaFileClass::Video.to_str(), + MediaFileType::Video.to_str(), + ORIGIN_CATEGORY.to_str(), ORIGIN_PRODUCT_CATEGORY.to_str(), ORIGIN_MODEL_TYPE.to_str(), - MEDIA_TYPE.to_str(), args.maybe_mime_type, args.file_size_bytes, diff --git a/crates/lib/mysql_queries/src/queries/media_files/create/insert_media_file_generic.rs b/crates/lib/mysql_queries/src/queries/media_files/create/insert_media_file_generic.rs index de1d15ce70..19c80109b7 100644 --- a/crates/lib/mysql_queries/src/queries/media_files/create/insert_media_file_generic.rs +++ b/crates/lib/mysql_queries/src/queries/media_files/create/insert_media_file_generic.rs @@ -3,6 +3,7 @@ use sqlx; use sqlx::MySqlPool; use enums::by_table::generic_synthetic_ids::id_category::IdCategory; +use enums::by_table::media_files::media_file_class::MediaFileClass; use enums::by_table::media_files::media_file_origin_category::MediaFileOriginCategory; use enums::by_table::media_files::media_file_origin_model_type::MediaFileOriginModelType; use enums::by_table::media_files::media_file_origin_product_category::MediaFileOriginProductCategory; @@ -22,6 +23,7 @@ pub struct InsertArgs<'a> { pub pool: &'a MySqlPool, pub job: &'a AvailableInferenceJob, + pub media_class: MediaFileClass, pub media_type: MediaFileType, // Origin and categorization @@ -99,6 +101,8 @@ pub async fn insert_media_file_generic( INSERT INTO media_files SET token = ?, + + media_class = ?, media_type = ?, origin_category = ?, @@ -140,6 +144,8 @@ pub async fn insert_media_file_generic( generated_by_cluster = ? "#, result_token, + + args.media_class.to_str(), args.media_type.to_str(), args.origin_category.to_str(), diff --git a/crates/service/job/inference_job/src/job/job_types/format_conversion/fbx_to_gltf/process_job.rs b/crates/service/job/inference_job/src/job/job_types/format_conversion/fbx_to_gltf/process_job.rs index 247072641a..b40ea1fb19 100644 --- a/crates/service/job/inference_job/src/job/job_types/format_conversion/fbx_to_gltf/process_job.rs +++ b/crates/service/job/inference_job/src/job/job_types/format_conversion/fbx_to_gltf/process_job.rs @@ -8,6 +8,7 @@ use log::{error, info, warn}; use buckets::public::media_files::bucket_file_path::MediaFileBucketPath; use enums::by_table::generic_inference_jobs::inference_result_type::InferenceResultType; use enums::by_table::generic_synthetic_ids::id_category::IdCategory; +use enums::by_table::media_files::media_file_class::MediaFileClass; use enums::by_table::media_files::media_file_origin_category::MediaFileOriginCategory; use enums::by_table::media_files::media_file_origin_product_category::MediaFileOriginProductCategory; use enums::by_table::media_files::media_file_type::MediaFileType; @@ -186,6 +187,7 @@ pub async fn process_job(args: FbxToGltfJobArgs<'_>) -> Result) -> Result