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()