From 489aabda3df959a54cf874f74be987b8e145dd52 Mon Sep 17 00:00:00 2001
From: Kasisnu <5044120+kasisnu@users.noreply.github.com>
Date: Fri, 31 May 2024 10:43:59 +0530
Subject: [PATCH] generated thumbnails extension fix (#336)

---
 .../video_style_transfer/validate_and_save_results.rs    | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/crates/service/job/inference_job/src/job/job_types/workflow/comfy_ui/video_style_transfer/validate_and_save_results.rs b/crates/service/job/inference_job/src/job/job_types/workflow/comfy_ui/video_style_transfer/validate_and_save_results.rs
index 147b39f5e8..076e3f2f7e 100644
--- a/crates/service/job/inference_job/src/job/job_types/workflow/comfy_ui/video_style_transfer/validate_and_save_results.rs
+++ b/crates/service/job/inference_job/src/job/job_types/workflow/comfy_ui/video_style_transfer/validate_and_save_results.rs
@@ -142,13 +142,20 @@ pub async fn validate_and_save_results(args: SaveResultsArgs<'_>) -> Result<Medi
   info!("Generating thumbnail tasks...");
 
   for output_type in thumbnail_types {
+    let output_ext = match get_file_extension(output_type) {
+        Ok(ext) => format!(".{}", ext),
+        Err(e) => {
+            error!("Failed to get extension for output type: {:?}", e);
+            continue;
+        }
+    };
     let thumbnail_task_result = ThumbnailTaskBuilder::new()
         .with_bucket(&*args.deps.buckets.public_bucket_client.bucket_name())
         .with_path(&*path_to_string(result_bucket_object_pathbuf.clone()))
         .with_source_mimetype(mimetype.as_str())
         .with_output_mimetype(output_type)
         .with_output_suffix("thumb")
-        .with_output_extension(&ext)
+        .with_output_extension(&output_ext)
         .with_event_id(&args.job.id.0.to_string())
         .send()
         .await;