add audio isolation feature

core/audio_isolation.py

@@ -8,6 +8,7 @@
 
 from typing import Optional
 from typing import Union
+from typing import IO
 from typing import List
 from typing import Dict
 from typing import Any
@@ -15,19 +16,40 @@ import requests
 import json
 import os
 import tempfile
+from io import BytesIO
 
 from dotenv import load_dotenv
 from elevenlabs import ElevenLabs
 
-def isolate_audio(fichier_audio: str, api_key: str):
-    client = ElevenLabs(api_key=api_key)
+def isolate_audio(fichier_audio: str):
+    load_dotenv()
+    client = ElevenLabs(api_key=os.getenv("ELEVENLABS_API_KEY"))
+    buffer = BytesIO()
     
     with open(fichier_audio, 'rb') as audio_file:
-        response = client.audio_isolation.audio_isolation(audio=audio_file)
+        isolated_audio_iterator = client.audio_isolation.audio_isolation(audio=audio_file)
+
+        for chunk in isolated_audio_iterator:
+            buffer.write(chunk)
+    
+    buffer.seek(0)
+    return buffer
+    
+
+"""  
+        with tempfile.NamedTemporaryFile(delete_on_close=False) as fp_out:
+            for chunk in isolated_audio_iterator:
+                fp_out.write(chunk)
+            print(f"L'audio a été isolé avec succès et enregistré sous \'{fp_out.name}\'.")
+            fp_out.seek(0)
+            return fp_out.name
+"""
+#    else:
+#        print(f"Erreur lors de l'isolement de l'audio: {isolated_audio_iterator.#status_code}, {isolated_audio_iterator.text}")
+#        return None
+    
+if __name__ == "__main__":
+    buffer = isolate_audio("audio.mp3")
+    with open("audio_isole.mp3", "wb") as output_file:
+        output_file.write(buffer.read())
         
-    if response.status_code == 200:
-        with open("audio_isole.mp3", "wb") as output_file:
-            output_file.write(response.content)
-        print("L'audio a été isolé avec succès et enregistré sous 'audio_isole.mp3'.")
-    else:
-        print(f"Erreur lors de l'isolement de l'audio: {response.status_code}, {response.text}")