From 7f1fa73bb04d417bf133a9febe5b0b4c1e2a9cbd Mon Sep 17 00:00:00 2001
From: Christopher Moyer <cm4181@gmail.com>
Date: Tue, 17 Dec 2024 23:25:00 -0500
Subject: [PATCH] delete files after processing

---
 src/main.rs | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/src/main.rs b/src/main.rs
index 4a1096d..52aa3ae 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -91,11 +91,13 @@ _|"""""|_|"""""|_|"""""|_|"""""|_|"""""|_|"""""|_|"""""| {======|_|"""""|_|"""""
                         error!("Stdout: {}", String::from_utf8_lossy(&result.stdout));
                         error!("Stderr: {}", String::from_utf8_lossy(&result.stderr));
 
+                        delete_file(path);
                         return delivery.ack(BasicAckOptions::default()).await.unwrap();
                     }
                     Err(error) => {
                         error!("Failed to transcribe {}, {}", &transcription_request.audio_file_path, error);
 
+                        delete_file(path);
                         return delivery.ack(BasicAckOptions::default()).await.unwrap();
                     }
                 }
@@ -106,6 +108,7 @@ _|"""""|_|"""""|_|"""""|_|"""""|_|"""""|_|"""""|_|"""""| {======|_|"""""|_|"""""
                     Err(_) => {
                         error!("Failed to read transcript file {}", &transcript_path.display());
 
+                        delete_file(path);
                         return delivery.ack(BasicAckOptions::default()).await.unwrap();
                     }
                 };
@@ -115,6 +118,8 @@ _|"""""|_|"""""|_|"""""|_|"""""|_|"""""|_|"""""|_|"""""| {======|_|"""""|_|"""""
                     Err(error) => {
                         error!("Failed to send message to meilisearch, {}", error);
 
+                        delete_file(path);
+                        delete_file(transcript_path.as_path());
                         return delivery.ack(BasicAckOptions::default()).await.unwrap();
                     }
                 };
@@ -128,6 +133,8 @@ _|"""""|_|"""""|_|"""""|_|"""""|_|"""""|_|"""""|_|"""""| {======|_|"""""|_|"""""
                     _ => {}
                 }
 
+                delete_file(path);
+                delete_file(transcript_path.as_path());
                 delivery.ack(BasicAckOptions::default()).await.unwrap()
             }
         }
@@ -178,6 +185,10 @@ async fn wait_for_task_to_complete(client: &Client, task: &TaskInfo) -> Result<T
     }
 }
 
+fn delete_file(file_path: &Path) {
+    fs::remove_file(file_path).unwrap_or_default();
+}
+
 #[derive(Serialize, Deserialize, Default, Debug)]
 struct TranscriptionRequest {
     audio_file_path: String,