From b5ec9ef99139fb9df21e661b5b8cc160e66dc611 Mon Sep 17 00:00:00 2001
From: Thorsten Mueller <MrThorstenM@gmx.net>
Date: Tue, 15 Feb 2022 17:52:03 +0100
Subject: [PATCH] Fixed minor issues

---
 helperScripts/MRS2LJSpeech.py | 19 ++++++++++++-------
 1 file changed, 12 insertions(+), 7 deletions(-)

diff --git a/helperScripts/MRS2LJSpeech.py b/helperScripts/MRS2LJSpeech.py
index c15f234..1c53915 100644
--- a/helperScripts/MRS2LJSpeech.py
+++ b/helperScripts/MRS2LJSpeech.py
@@ -61,6 +61,9 @@ def convert_audio():
   
   print('→ Converting %s Audio Files to 22050 Hz, 16 Bit, Mono\n' % "{:,}".format(recordings))
 
+  # Please use `pip install ffmpeg-python`
+  import ffmpeg
+
   for idx, wav in enumerate(glob.glob(os.path.join(output_dir_audio_temp, "*.wav"))):
 
     percent = (idx + 1) / recordings
@@ -75,11 +78,6 @@ def convert_audio():
       .run(capture_stdout=True)
     )
 
-    # Delete Temp File
-    os.remove(wav)
-
-  # Remove Temp Folder
-  rmtree(output_dir_audio_temp)
 
 def copy_audio():
   global output_dir_audio
@@ -90,7 +88,7 @@ def copy_audio():
   print('→ Copy %s Audio Files to LJSpeech Dataset\n' % "{:,}".format(recordings))
 
   for idx, wav in enumerate(glob.glob(os.path.join(output_dir_audio_temp, "*.wav"))):    
-    shutil.copyfile(wav,os.path.join(output_dir_audio, os.path.basename(wav)))
+    copyfile(wav,os.path.join(output_dir_audio, os.path.basename(wav)))
 
 def create_meta_data(mrs_dir):
   print('→ Creating META Data')
@@ -125,6 +123,12 @@ def create_meta_data(mrs_dir):
   metadata.close()
   conn.close()
 
+def cleanup():
+  global output_dir_audio_temp
+
+  # Remove Temp Folder
+  rmtree(output_dir_audio_temp)
+
 def main():
   parser = argparse.ArgumentParser()
   parser.add_argument('--mrs_dir', required=True)
@@ -140,11 +144,12 @@ def main():
   create_meta_data(args.mrs_dir)
 
   if(args.ffmpeg):
-    import ffmpeg
     convert_audio()
   
   else:
     copy_audio()
+  
+  cleanup()
 
   print('\n\033[38;5;86;1m✔\033[0m COMPLETE【ツ】\n')