Update app.py

app.py

@@ -216,20 +216,16 @@ def apply_noisereduce(audio_list):
 
     return result
 
-def run(audio_files, file_m, file_index):
-    if not audio_files:
+def run(audio_file, file_m, file_index):
+    if not audio_file:
         raise ValueError("Please provide an audio file.")
 
-    if isinstance(audio_files, str):
-        audio_files = [audio_files]
-
     try:
-        duration_base = librosa.get_duration(filename=audio_files[0])
+        duration_base = librosa.get_duration(filename=audio_file)
         print("Duration:", duration_base)
     except Exception as e:
         print(e)
 
-    # Construct the full path to the model and index files
     file_m = os.path.join(model_dir, file_m)
     file_index = os.path.join(model_dir, file_index) if file_index else None
 
@@ -256,16 +252,16 @@ def run(audio_files, file_m, file_index):
         respiration_median_filtering=r_m_f,
         envelope_ratio=e_r,
         consonant_breath_protection=c_b_p,
-        resample_sr=44100 if audio_files[0].endswith('.mp3') else 0,
+        resample_sr=44100 if audio_file.endswith('.mp3') else 0,
     )
     time.sleep(0.1)
 
-    result = convert_now(audio_files, random_tag, converter)
+    result = convert_now([audio_file], random_tag, converter)
 
-    # Always apply noise reduction
     result = apply_noisereduce(result)
 
-    return result
+    # Since there's only one file, return its path for both download and playback
+    return result[0]
 
 
 def model_conf():
@@ -301,22 +297,17 @@ def button_conf():
     )
 
 def output_conf():
-    return gr.File(
-        label="Result",
-        file_count="multiple",
-        interactive=False,
-    )
+    return gr.File(label="Download Processed Audio", interactive=False), gr.Audio(label="Play Processed Audio")
 
 def get_gui(theme):
     with gr.Blocks(theme=theme, delete_cache=(3200, 3200)) as app:
         gr.Markdown(title)
 
         aud = audio_conf()
-
         model = model_conf()
         indx = index_conf()
         button_base = button_conf()
-        output_base = output_conf()
+        output_file, output_audio = output_conf()
 
         button_base.click(
             run,
@@ -325,7 +316,7 @@ def get_gui(theme):
                 model,
                 indx,
             ],
-            outputs=[output_base],
+            outputs=[output_file, output_audio],
         )
 
         gr.Examples(
@@ -347,7 +338,7 @@ def get_gui(theme):
                 model,
                 indx,
             ],
-            outputs=[output_base],
+            outputs=[output_file, output_audio],
             cache_examples=False,
         )