Update app.py

app.py

@@ -2,6 +2,8 @@ import gradio as gr
 
 import glob
 import os
+import shutil
+import tempfile
 from pydub import AudioSegment
 
 def process_audio(input_file):
@@ -46,24 +48,38 @@ def find_audio_chunks():
     return wav_files  # Returning the list of file paths
 
 def concatenate_audio(output_filename="final_output.wav"):
-    """Concatenates all audio chunks and saves them to a final output file."""
+    """Concatenates all audio chunks and saves them to a final output file in a temporary directory."""
+    
     wav_files = find_audio_chunks()  # Get sorted audio file paths
     
     if not wav_files:
         print("No audio files found.")
         return []
 
+    # Create a temporary directory
+    temp_dir = tempfile.mkdtemp()
+    
     # Load and concatenate all audio files
     combined = AudioSegment.empty()
+    temp_wav_files = []
+    
     for file in wav_files:
         audio = AudioSegment.from_wav(file)
         combined += audio
-
+        
+        # Move individual files to the temp directory
+        temp_file_path = os.path.join(temp_dir, os.path.basename(file))
+        shutil.move(file, temp_file_path)
+        temp_wav_files.append(temp_file_path)
+
+    # Define the final output path in the temporary directory
+    temp_output_path = os.path.join(temp_dir, output_filename)
+    
     # Export the final combined audio
-    combined.export(output_filename, format="wav")
-    print(f"Concatenated audio saved as {output_filename}")
+    combined.export(temp_output_path, format="wav")
+    print(f"Concatenated audio saved at {temp_output_path}")
     
-    return output_filename, wav_files # Returning the list of processed file paths
+    return temp_output_path, temp_wav_files  # Returning temp paths
 
 
 def infer(audio_input_path):