Add extension

crates/lib/mysql_queries/src/queries/media_files/insert_media_file_from_face_animation.rs

@@ -24,6 +24,7 @@ pub struct InsertArgs<'a> {
   //pub duration_millis: u64,
 
   pub public_bucket_directory_hash: &'a str,
+  pub maybe_public_bucket_extension: Option<&'a str>,
 
   pub is_on_prem: bool,
   pub worker_hostname: &'a str,
@@ -85,6 +86,7 @@ SET
   checksum_sha2 = ?,
 
   public_bucket_directory_hash = ?,
+  maybe_public_bucket_extension = ?,
 
   maybe_creator_user_token = ?,
   creator_ip_address = ?,
@@ -111,6 +113,7 @@ SET
       args.sha256_checksum,
 
       args.public_bucket_directory_hash,
+      args.maybe_public_bucket_extension,
 
       args.job.maybe_creator_user_token,
       args.job.creator_ip_address,

crates/service/job/inference_job/src/job/job_types/lipsync/sad_talker/process_job.rs

@@ -227,7 +227,7 @@ pub async fn process_job(args: SadTalkerProcessJobArgs<'_>) -> Result<JobSuccess
 
   check_file_exists(&output_video_fs_path).map_err(|e| ProcessSingleJobError::Other(e))?;
 
-  // ==================== WATERMARK ==================== //
+  // ==================== OPTIONAL WATERMARK ==================== //
 
   let mut finished_file = output_video_fs_path.clone();
 
@@ -259,7 +259,6 @@ pub async fn process_job(args: SadTalkerProcessJobArgs<'_>) -> Result<JobSuccess
     }
   }
 
-
   // ==================== CHECK ALL FILES EXIST AND GET METADATA ==================== //
 
   info!("Checking that output watermark file exists: {:?} ...", finished_file);
@@ -324,6 +323,7 @@ pub async fn process_job(args: SadTalkerProcessJobArgs<'_>) -> Result<JobSuccess
     file_size_bytes,
     sha256_checksum: &file_checksum,
     public_bucket_directory_hash: result_bucket_location.get_object_hash(),
+    maybe_public_bucket_extension: None, // TODO(bt,2023-10-02): Add extension '.mp4'
     is_on_prem: args.job_dependencies.container.is_on_prem,
     worker_hostname: &args.job_dependencies.container.hostname,
     worker_cluster: &args.job_dependencies.container.cluster_name,