From 2faee0f6dffe0f288eb7d949927c7085efdcaa1f Mon Sep 17 00:00:00 2001
From: Mews <ar754456@gmail.com>
Date: Sat, 8 Jun 2024 16:57:03 +0100
Subject: [PATCH] Delete invalid files when they're invalid

---
 src/simplestretch/__init__.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/simplestretch/__init__.py b/src/simplestretch/__init__.py
index c74f2c7..057d1fe 100644
--- a/src/simplestretch/__init__.py
+++ b/src/simplestretch/__init__.py
@@ -2,7 +2,7 @@
 
 import soundfile
 from numpy import ndarray
-
+import pathlib
 
 
 def stretch_audio(
@@ -60,6 +60,9 @@ def stretch_audio(
         try:
             soundfile.write(output, audio, samplerate=stretched_samplerate)
         except soundfile.LibsndfileError as exc:
+            # Delete invalid file
+            pathlib.Path.unlink(output)
+
             exc.add_note("(Try saving it as a .wav file instead)")
             raise exc