From fa74b9715f98fd738dfda281534554ecd425be47 Mon Sep 17 00:00:00 2001
From: Brandon Thomas <echelon@gmail.com>
Date: Fri, 15 Sep 2023 09:28:41 -0400
Subject: [PATCH] logging for job

---
 .../src/job/job_types/lipsync/sad_talker/process_job.rs  | 9 ++++++++-
 .../job/inference_job/src/util/model_downloader.rs       | 3 +++
 2 files changed, 11 insertions(+), 1 deletion(-)

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 e62e360a74..8367ee9d46 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
@@ -43,7 +43,14 @@ pub async fn process_job(args: SadTalkerProcessJobArgs<'_>) -> Result<JobSuccess
 
   info!("Download models (if not present)...");
 
+  let mut i : usize = 0;
+
   for downloader in deps.job_type_details.sad_talker.downloaders.all_downloaders() {
+
+    // Temporary debugging
+    info!("Downloader {}", i);
+    i = i + 1;
+
     let result = downloader.download_if_not_on_filesystem(
       &args.job_dependencies.private_bucket_client,
       &args.job_dependencies.fs.scoped_temp_dir_creator_for_downloads,
@@ -57,7 +64,7 @@ pub async fn process_job(args: SadTalkerProcessJobArgs<'_>) -> Result<JobSuccess
 
   // ==================== TEMP DIR ==================== //
 
-  let work_temp_dir = format!("temp_rvc_v2_inference_{}", job.id.0);
+  let work_temp_dir = format!("temp_sad_talker_inference_{}", job.id.0);
 
   // NB: TempDir exists until it goes out of scope, at which point it should delete from filesystem.
   let work_temp_dir = args.job_dependencies
diff --git a/crates/service/job/inference_job/src/util/model_downloader.rs b/crates/service/job/inference_job/src/util/model_downloader.rs
index 0740614910..36aa9a56e4 100644
--- a/crates/service/job/inference_job/src/util/model_downloader.rs
+++ b/crates/service/job/inference_job/src/util/model_downloader.rs
@@ -35,6 +35,9 @@ pub trait ModelDownloader {
       return Ok(());
     }
 
+    info!("Model not on filesystem: {}", self.get_model_name());
+    info!("Needs to be in: {:?}", self.get_filesystem_path());
+
     if let Some(parent_directory_path) = filesystem_path.parent() {
       create_dir_all_if_missing(parent_directory_path)?;
     }