Update app.py

app.py

@@ -3,6 +3,8 @@ from elevenlabs.client import ElevenLabs
 from elevenlabs.core.api_error import ApiError
 import os
 from dotenv import load_dotenv
+from pydub import AudioSegment
+import tempfile
 
 # Load environment variables
 load_dotenv()
@@ -10,19 +12,30 @@ load_dotenv()
 # Initialize the client with your API key
 client = ElevenLabs(api_key=os.getenv("ELEVENLABS_API_KEY"))
 
+def is_valid_audio(file_path):
+    try:
+        AudioSegment.from_file(file_path)
+        return True
+    except:
+        return False
+
 def isolate_audio(audio_file):
     if audio_file is None:
         return None, "Please upload an 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."
+
     try:
         # Perform audio isolation
-        isolated_audio_iterator = client.audio_isolation.audio_isolation(audio=audio_file)
+        with open(audio_file, "rb") as file:
+            isolated_audio_iterator = client.audio_isolation.audio_isolation(audio=file)
 
         # Save the isolated audio to a temporary file
-        output_file_path = "isolated_audio.mp3"
-        with open(output_file_path, "wb") as output_file:
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as temp_file:
             for chunk in isolated_audio_iterator:
-                output_file.write(chunk)
+                temp_file.write(chunk)
+            output_file_path = temp_file.name
 
         return output_file_path, "Audio isolation completed successfully."
     except ApiError as e:
@@ -45,4 +58,5 @@ iface = gr.Interface(
 )
 
 # Launch the app
-iface.launch()
+iface.launch()
+