From 8177f9eef26be9e37066536bfc8cb71cfad6e231 Mon Sep 17 00:00:00 2001
From: Leonard Excoffier <excoffier.leonard@gmail.com>
Date: Tue, 13 Feb 2024 22:30:16 -0500
Subject: [PATCH] put in a function the cleanup

---
 main.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/main.py b/main.py
index 2d878f4..9348ad7 100644
--- a/main.py
+++ b/main.py
@@ -29,8 +29,6 @@ def transcription(OPENAI_API_KEY, audio_filename):
         )
 
         print("Transcription finished")
-        
-        os.remove(audio_filename)
 
         return transcript
     except Exception as e:
@@ -56,12 +54,21 @@ def summarize(OPENAI_API_KEY, transcript):
         print(f"Failed to summarize transcript: {e}")
         return None
 
+def cleanup(audio_filename):
+    try:
+        os.remove(audio_filename)
+        print(f"Deleted audio: {audio_filename}")
+    except Exception as e:
+        print(f"Failed to delete audio: {audio_filename}: {e}")
+        return None
+
 def main():
     OPENAI_API_KEY = os.environ.get('OPENAI_API_KEY')
     video_url = str(input("video url:"))
     audio_filename = download_audio(video_url)
     transcript = transcription(OPENAI_API_KEY, audio_filename)
     summarize(OPENAI_API_KEY, transcript)
+    cleanup(audio_filename)
 
 if __name__ == "__main__":
     main()