Update app.py

app.py

@@ -1,7 +1,52 @@
 import gradio as gr 
 
+import glob
+import os
+from pydub import AudioSegment
+
+def cleanup_old_audio():
+    """Remove old audio files before starting a new inference."""
+    files_to_remove = glob.glob("out_en-*.wav") + glob.glob("final_output.wav")
+    
+    if files_to_remove:
+        print(f"Cleaning up {len(files_to_remove)} old audio files...")
+        for file in files_to_remove:
+            try:
+                os.remove(file)
+                print(f"Deleted: {file}")
+            except Exception as e:
+                print(f"Error deleting {file}: {e}")
+    else:
+        print("No old audio files found.")
+
+def concatenate_audio(output_filename="final_output.wav"):
+    # Find all WAV files matching the pattern
+    wav_files = glob.glob("out_en-*.wav")
+
+    # Extract numbers and sort properly
+    wav_files.sort(key=lambda x: int(x.split('-')[-1].split('.')[0]))
+
+    if not wav_files:
+        print("No audio files found.")
+        return
+    
+    print(f"Found {len(wav_files)} audio chunks: {wav_files}")
+
+    # Load and concatenate all audio files
+    combined = AudioSegment.empty()
+    for file in wav_files:
+        audio = AudioSegment.from_wav(file)
+        combined += audio
+
+    # Export the final combined audio
+    combined.export(output_filename, format="wav")
+    print(f"Concatenated audio saved as {output_filename}")
+
+
 def infer(audio_input_path):
 
+    cleanup_old_audio()
+
     import subprocess
 
     command = [
@@ -19,10 +64,12 @@ def infer(audio_input_path):
     # Check if the command was successful
     if result.returncode == 0:
         print("Command executed successfully.")
+        concatenate_audio()
+        return "final_output.wav"
     else:
         print("Error executing command.")
-
-    return "out_en-1.wav"
+        raise gr.Error("Error executing command")
+    
 
 with gr.Blocks() as demo:
     with gr.Column(elem_id="col-container"):