From 4306a4763ddd9d699ee0c3b0feb52c68973df7bb Mon Sep 17 00:00:00 2001
From: Brandon Thomas <echelon@gmail.com>
Date: Mon, 2 Oct 2023 00:40:12 -0400
Subject: [PATCH] Add extension

---
 .../media_files/insert_media_file_from_face_animation.rs      | 3 +++
 .../src/job/job_types/lipsync/sad_talker/process_job.rs       | 4 ++--
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/crates/lib/mysql_queries/src/queries/media_files/insert_media_file_from_face_animation.rs b/crates/lib/mysql_queries/src/queries/media_files/insert_media_file_from_face_animation.rs
index cad9f10f36..8cf2bdb54d 100644
--- a/crates/lib/mysql_queries/src/queries/media_files/insert_media_file_from_face_animation.rs
+++ b/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,
diff --git a/crates/service/job/inference_job/src/job/job_types/lipsync/sad_talker/process_job.rs b/crates/service/job/inference_job/src/job/job_types/lipsync/sad_talker/process_job.rs
index 20497e11b1..3a33cb5f54 100644
--- a/crates/service/job/inference_job/src/job/job_types/lipsync/sad_talker/process_job.rs
+++ b/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,