From 0607337cbb2000cc757440fe300d829d44792894 Mon Sep 17 00:00:00 2001
From: Dev Khant <devkhant24@gmail.com>
Date: Mon, 1 Apr 2024 15:56:20 +0530
Subject: [PATCH] Fix for saving audio files

---
 docs/docs/audio/speech.md | 14 +++++---------
 docs/welcome/tutorial.md  |  2 +-
 src/marvin/ai/audio.py    |  6 +++---
 3 files changed, 9 insertions(+), 13 deletions(-)

diff --git a/docs/docs/audio/speech.md b/docs/docs/audio/speech.md
index 47b5f4ae1..5046f0ae6 100644
--- a/docs/docs/audio/speech.md
+++ b/docs/docs/audio/speech.md
@@ -24,7 +24,7 @@ Marvin can generate speech from text.
 
         !!! success "Result"
             ```python
-            audio.stream_to_file("fancy_computer.mp3")
+            audio.save("fancy_computer.mp3")
             ```
             <audio controls>
               <source src="/assets/audio/fancy_computer.mp3" type="audio/mpeg">
@@ -47,7 +47,7 @@ Marvin can generate speech from text.
 
         !!! success "Result"
             ```python
-            audio.stream_to_file("hello_arthur.mp3")
+            audio.save("hello_arthur.mp3")
             ```
             <audio controls>
               <source src="/assets/audio/hello_arthur.mp3" type="audio/mpeg">
@@ -100,14 +100,10 @@ ai_say('hello')
 Both `speak` and `@speech` accept a `voice` parameter that allows you to choose from a variety of voices. You can preview the available voices [here](https://platform.openai.com/docs/guides/text-to-speech/voice-options).
 
 ```python
+The result of the `speak` function and `@speech` decorator is an audio stream.
 
-## Saving audio files
-
-The result of the `speak` function and `@speech` decorator is an audio stream. You can save this stream to disk like this:
-
-```python
-audio = marvin.speak("Hello, world!")
-audio.stream_to_file("hello_world.mp3")
+audio = marvin.speak("Hello, world!", voice="nova")
+audio.save("hello_world.mp3") # Saving audio files
 ```
 
 
diff --git a/docs/welcome/tutorial.md b/docs/welcome/tutorial.md
index 1ce38d93a..ebd2f34fb 100644
--- a/docs/welcome/tutorial.md
+++ b/docs/welcome/tutorial.md
@@ -360,7 +360,7 @@ And remember:
     import marvin
 
     audio = marvin.speak("and above all else... don't panic!")
-    audio.stream_to_file("dont_panic.mp3")
+    audio.save("dont_panic.mp3")
     ```
 
     !!! success "Result"
diff --git a/src/marvin/ai/audio.py b/src/marvin/ai/audio.py
index e35d6a99d..12103f310 100644
--- a/src/marvin/ai/audio.py
+++ b/src/marvin/ai/audio.py
@@ -5,7 +5,7 @@
 
 import marvin
 from marvin.client.openai import AsyncMarvinClient
-from marvin.types import Audio, HttpxBinaryResponseContent, SpeechRequest
+from marvin.types import Audio, SpeechRequest
 from marvin.utilities.asyncio import run_sync
 from marvin.utilities.jinja import Environment
 from marvin.utilities.logging import get_logger
@@ -86,7 +86,7 @@ def speak(
     text: str,
     voice: Literal["alloy", "echo", "fable", "onyx", "nova", "shimmer"] = None,
     model_kwargs: Optional[dict[str, Any]] = None,
-) -> HttpxBinaryResponseContent:
+) -> Audio:
     """
     Generates audio from text using an AI.
 
@@ -101,7 +101,7 @@ def speak(
             language model. Defaults to None.
 
     Returns:
-        HttpxBinaryResponseContent: The generated audio.
+        Audio: The generated audio.
     """
     return run_sync(speak_async(text, voice, model_kwargs))