mariesig commited on
Commit
377e7f5
·
1 Parent(s): bc483dc

rmv gradio created audio

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -3,6 +3,7 @@ import time
3
  from typing import Optional, Any
4
 
5
  import gradio as gr
 
6
  from loguru import logger
7
  from PIL import Image
8
 
@@ -144,6 +145,7 @@ def delete_enhancement(path_to_delete: str):
144
  Consider adding a flag to files to indicate whether files can be deleted, if disk usage becomes an issue.
145
  """
146
  if not path_to_delete or not os.path.exists(path_to_delete):
 
147
  logger.warning(f"Path to delete is invalid or does not exist: {path_to_delete}")
148
  return
149
  filename_no_ext = os.path.splitext(os.path.basename(path_to_delete))[0]
@@ -188,7 +190,9 @@ def start_processing(sample_path: str, model_radio: str, api_key: str) -> tuple[
188
  return gr.update(visible=False), gr.update(interactive=False), input_enhancement_path
189
 
190
 
191
- def enable_new_input():
 
 
192
  return gr.update(visible=True), gr.update(interactive=True)
193
 
194
 
@@ -290,7 +294,7 @@ with gr.Blocks() as demo:
290
  api_key,
291
  ],
292
  outputs=[noisy_audio, noisy_image, enhanced_audio, enhanced_image],
293
- ).then(enable_new_input, None, [examples_group, btn])
294
  noise_type.change(toggle_SNR, noise_type, noise_level)
295
 
296
  cleanup_tmp(minutes_keep=0, filter=[])
 
3
  from typing import Optional, Any
4
 
5
  import gradio as gr
6
+ from httpx import delete
7
  from loguru import logger
8
  from PIL import Image
9
 
 
145
  Consider adding a flag to files to indicate whether files can be deleted, if disk usage becomes an issue.
146
  """
147
  if not path_to_delete or not os.path.exists(path_to_delete):
148
+ cleanup_tmp(minutes_keep=120, filter=EXAMPLES)
149
  logger.warning(f"Path to delete is invalid or does not exist: {path_to_delete}")
150
  return
151
  filename_no_ext = os.path.splitext(os.path.basename(path_to_delete))[0]
 
190
  return gr.update(visible=False), gr.update(interactive=False), input_enhancement_path
191
 
192
 
193
+ def enable_new_input(old_input: str):
194
+ os.remove(old_input)
195
+ print(f"Removed old input file: {old_input}")
196
  return gr.update(visible=True), gr.update(interactive=True)
197
 
198
 
 
294
  api_key,
295
  ],
296
  outputs=[noisy_audio, noisy_image, enhanced_audio, enhanced_image],
297
+ ).then(enable_new_input, audio_file, [examples_group, btn])
298
  noise_type.change(toggle_SNR, noise_type, noise_level)
299
 
300
  cleanup_tmp(minutes_keep=0, filter=[])