Update app.py

app.py

@@ -5,6 +5,7 @@ import os
 from dotenv import load_dotenv
 from pydub import AudioSegment
 import tempfile
+import shutil
 
 # Load environment variables
 load_dotenv()
@@ -26,16 +27,22 @@ def isolate_audio(audio_file):
     if not is_valid_audio(audio_file):
         return None, "The uploaded file is not a valid audio file. Please ensure it is a playable audio file."
 
+    temp_dir = tempfile.mkdtemp()
+    input_file_path = os.path.join(temp_dir, "input_audio.mp3")
+    output_file_path = os.path.join(temp_dir, "isolated_audio.mp3")
+
     try:
+        # Copy the input file to our temporary directory
+        shutil.copy2(audio_file, input_file_path)
+
         # Perform audio isolation
-        with open(audio_file, "rb") as file:
+        with open(input_file_path, "rb") as file:
             isolated_audio_iterator = client.audio_isolation.audio_isolation(audio=file)
 
-        # Save the isolated audio to a temporary file
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as temp_file:
-            for chunk in isolated_audio_iterator:
-                temp_file.write(chunk)
-            output_file_path = temp_file.name
+            # Save the isolated audio to the output file
+            with open(output_file_path, "wb") as output_file:
+                for chunk in isolated_audio_iterator:
+                    output_file.write(chunk)
 
         return output_file_path, "Audio isolation completed successfully."
     except ApiError as e:
@@ -44,6 +51,9 @@ def isolate_audio(audio_file):
     except Exception as e:
         error_message = f"An unexpected error occurred: {str(e)}"
         return None, error_message
+    finally:
+        # Clean up temporary files
+        shutil.rmtree(temp_dir, ignore_errors=True)
 
 # Create Gradio interface
 iface = gr.Interface(