Update src/webui.py

src/webui.py

@@ -10,6 +10,7 @@ import gradio as gr
 
 from main import song_cover_pipeline
 from my_utils import show_stored_files
+from my_utils import remove_files_and_folders
 
 BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
 
@@ -243,11 +244,17 @@ if __name__ == '__main__':
                             outputs=[pitch, main_gain, backup_gain, inst_gain, index_rate, filter_radius, rms_mix_rate,
                                      protect, f0_method, crepe_hop_length, pitch_all, reverb_rm_size, reverb_wet,
                                      reverb_dry, reverb_damping, output_format, ai_cover])
+            
             with gr.Accordion('For Debug Purposes', open=False):
                 with gr.Row():
                     folder = gr.Text(label='folder')
                     debug_button = gr.Button(value='Debug')
             debug_button.click(show_stored_files, inputs =[folder])
+            with gr.Row():
+                clear_output = gr.Text()
+                clear_cache_button = gr.Button(value='Clear Gradio Cache')
+            clear_cache_button.click(remove_files_and_folders, inputs=["/tmp/gradio/"])
+