Update src/main.py

src/main.py

@@ -232,6 +232,13 @@ def combine_audio(audio_paths, output_path, main_gain, backup_gain, inst_gain, o
     #instrumental_audio = AudioSegment.from_wav(audio_paths[2]) - 7 + inst_gain
     main_vocal_audio.overlay(backup_vocal_audio).export(output_path, format=output_format)
 
+def remove_files_and_folders(folder_path):
+    try:
+        shutil.rmtree(folder_path)
+        print(f"All files and folders inside {folder_path} have been removed.")
+    except Exception as e:
+        print(f"An error occurred: {e}")
+
 # song_input is the main vocals that is going through the rvc processs
 # backup vocals are the backup vocals to be added at the end
 def song_cover_pipeline(main_vocals, backup_vocals, voice_model, pitch_change, keep_files,
@@ -239,6 +246,9 @@ def song_cover_pipeline(main_vocals, backup_vocals, voice_model, pitch_change, k
                         rms_mix_rate=0.25, f0_method='rmvpe', crepe_hop_length=128, protect=0.33, pitch_change_all=0,
                         reverb_rm_size=0.15, reverb_wet=0.2, reverb_dry=0.8, reverb_damping=0.7, output_format='mp3',
                         progress=gr.Progress()):
+    print("Cleaning gradio cache ...")
+    remove_files_and_folders("/tmp/gradio/")
+    print("Done")
     try:
         print(f"DEBUG PRINT: main_vocals: {main_vocals}, backup_vocals: {backup_vocals}")
         if not main_vocals or not voice_model: