akshansh36
commited on
Commit
•
edf4d66
1
Parent(s):
2b1f02f
Update app.py
Browse files
app.py
CHANGED
@@ -216,20 +216,16 @@ def apply_noisereduce(audio_list):
|
|
216 |
|
217 |
return result
|
218 |
|
219 |
-
def run(
|
220 |
-
if not
|
221 |
raise ValueError("Please provide an audio file.")
|
222 |
|
223 |
-
if isinstance(audio_files, str):
|
224 |
-
audio_files = [audio_files]
|
225 |
-
|
226 |
try:
|
227 |
-
duration_base = librosa.get_duration(filename=
|
228 |
print("Duration:", duration_base)
|
229 |
except Exception as e:
|
230 |
print(e)
|
231 |
|
232 |
-
# Construct the full path to the model and index files
|
233 |
file_m = os.path.join(model_dir, file_m)
|
234 |
file_index = os.path.join(model_dir, file_index) if file_index else None
|
235 |
|
@@ -256,16 +252,16 @@ def run(audio_files, file_m, file_index):
|
|
256 |
respiration_median_filtering=r_m_f,
|
257 |
envelope_ratio=e_r,
|
258 |
consonant_breath_protection=c_b_p,
|
259 |
-
resample_sr=44100 if
|
260 |
)
|
261 |
time.sleep(0.1)
|
262 |
|
263 |
-
result = convert_now(
|
264 |
|
265 |
-
# Always apply noise reduction
|
266 |
result = apply_noisereduce(result)
|
267 |
|
268 |
-
return
|
|
|
269 |
|
270 |
|
271 |
def model_conf():
|
@@ -301,22 +297,17 @@ def button_conf():
|
|
301 |
)
|
302 |
|
303 |
def output_conf():
|
304 |
-
return gr.File(
|
305 |
-
label="Result",
|
306 |
-
file_count="multiple",
|
307 |
-
interactive=False,
|
308 |
-
)
|
309 |
|
310 |
def get_gui(theme):
|
311 |
with gr.Blocks(theme=theme, delete_cache=(3200, 3200)) as app:
|
312 |
gr.Markdown(title)
|
313 |
|
314 |
aud = audio_conf()
|
315 |
-
|
316 |
model = model_conf()
|
317 |
indx = index_conf()
|
318 |
button_base = button_conf()
|
319 |
-
|
320 |
|
321 |
button_base.click(
|
322 |
run,
|
@@ -325,7 +316,7 @@ def get_gui(theme):
|
|
325 |
model,
|
326 |
indx,
|
327 |
],
|
328 |
-
outputs=[
|
329 |
)
|
330 |
|
331 |
gr.Examples(
|
@@ -347,7 +338,7 @@ def get_gui(theme):
|
|
347 |
model,
|
348 |
indx,
|
349 |
],
|
350 |
-
outputs=[
|
351 |
cache_examples=False,
|
352 |
)
|
353 |
|
|
|
216 |
|
217 |
return result
|
218 |
|
219 |
+
def run(audio_file, file_m, file_index):
|
220 |
+
if not audio_file:
|
221 |
raise ValueError("Please provide an audio file.")
|
222 |
|
|
|
|
|
|
|
223 |
try:
|
224 |
+
duration_base = librosa.get_duration(filename=audio_file)
|
225 |
print("Duration:", duration_base)
|
226 |
except Exception as e:
|
227 |
print(e)
|
228 |
|
|
|
229 |
file_m = os.path.join(model_dir, file_m)
|
230 |
file_index = os.path.join(model_dir, file_index) if file_index else None
|
231 |
|
|
|
252 |
respiration_median_filtering=r_m_f,
|
253 |
envelope_ratio=e_r,
|
254 |
consonant_breath_protection=c_b_p,
|
255 |
+
resample_sr=44100 if audio_file.endswith('.mp3') else 0,
|
256 |
)
|
257 |
time.sleep(0.1)
|
258 |
|
259 |
+
result = convert_now([audio_file], random_tag, converter)
|
260 |
|
|
|
261 |
result = apply_noisereduce(result)
|
262 |
|
263 |
+
# Since there's only one file, return its path for both download and playback
|
264 |
+
return result[0]
|
265 |
|
266 |
|
267 |
def model_conf():
|
|
|
297 |
)
|
298 |
|
299 |
def output_conf():
|
300 |
+
return gr.File(label="Download Processed Audio", interactive=False), gr.Audio(label="Play Processed Audio")
|
|
|
|
|
|
|
|
|
301 |
|
302 |
def get_gui(theme):
|
303 |
with gr.Blocks(theme=theme, delete_cache=(3200, 3200)) as app:
|
304 |
gr.Markdown(title)
|
305 |
|
306 |
aud = audio_conf()
|
|
|
307 |
model = model_conf()
|
308 |
indx = index_conf()
|
309 |
button_base = button_conf()
|
310 |
+
output_file, output_audio = output_conf()
|
311 |
|
312 |
button_base.click(
|
313 |
run,
|
|
|
316 |
model,
|
317 |
indx,
|
318 |
],
|
319 |
+
outputs=[output_file, output_audio],
|
320 |
)
|
321 |
|
322 |
gr.Examples(
|
|
|
338 |
model,
|
339 |
indx,
|
340 |
],
|
341 |
+
outputs=[output_file, output_audio],
|
342 |
cache_examples=False,
|
343 |
)
|
344 |
|