cartesia 1.0.2__tar.gz → 1.0.4__tar.gz

{cartesia-1.0.2 → cartesia-1.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cartesia
-Version: 1.0.2
+Version: 1.0.4
 Summary: The official Python library for the Cartesia API.
 Home-page: 
 Author: Cartesia, Inc.
@@ -227,6 +227,101 @@ p.terminate()
 ws.close()  # Close the websocket connection
 ```
 
+#### Conditioning speech on previous generations using WebSocket
+
+In some cases, input text may need to be streamed in. In these cases, it would be slow to wait for all the text to buffer before sending it to Cartesia's TTS service.
+
+To mitigate this, Cartesia offers audio continuations. In this setting, users can send input text, as it becomes available, over a websocket connection.
+
+To do this, we will create a `context` and sending multiple requests without awaiting the response. Then you can listen to the responses in the order they were sent.
+
+Each `context` will be closed automatically after 5 seconds of inactivity or when the `no_more_inputs` method is called. `no_more_inputs` sends a request with the `continue_=False`, which indicates no more inputs will be sent over this context
+
+```python
+import asyncio
+import os
+import pyaudio
+from cartesia import AsyncCartesia
+
+async def send_transcripts(ctx):
+    # Check out voice IDs by calling `client.voices.list()` or on [play.cartesia.ai](https://play.cartesia.ai/)
+    voice_id = "87748186-23bb-4158-a1eb-332911b0b708"
+
+    # You can check out our models at [docs.cartesia.ai](https://docs.cartesia.ai/getting-started/available-models).
+    model_id = "sonic-english"
+    
+    # You can find the supported `output_format`s in our [API Reference](https://docs.cartesia.ai/api-reference/endpoints/stream-speech-server-sent-events).
+    output_format = {
+        "container": "raw",
+        "encoding": "pcm_f32le",
+        "sample_rate": 44100,
+    }
+
+    transcripts = [
+        "Sonic and Yoshi team up in a dimension-hopping adventure! ",
+        "Racing through twisting zones, they dodge Eggman's badniks and solve ancient puzzles. ",
+        "In the Echoing Caverns, they find the Harmonic Crystal, unlocking new powers. ",
+        "Sonic's speed creates sound waves, while Yoshi's eggs become sonic bolts. ",
+        "As they near Eggman's lair, our heroes charge their abilities for an epic boss battle. ",
+        "Get ready to spin, jump, and sound-blast your way to victory in this high-octane crossover!"
+    ]
+    
+    for transcript in transcripts:
+        # Send text inputs as they become available
+        await ctx.send(
+            model_id=model_id,
+            transcript=transcript,
+            voice_id=voice_id,
+            continue_=True,
+            output_format=output_format,
+        )
+
+    # Indicate that no more inputs will be sent. Otherwise, the context will close after 5 seconds of inactivity.
+    await ctx.no_more_inputs() 
+
+async def receive_and_play_audio(ctx):
+    p = pyaudio.PyAudio()
+    stream = None
+    rate = 44100
+
+    async for output in ctx.receive():
+        buffer = output["audio"]
+
+        if not stream:
+            stream = p.open(
+                format=pyaudio.paFloat32,
+                channels=1,
+                rate=rate,
+                output=True
+            )
+
+        stream.write(buffer)
+
+    stream.stop_stream()
+    stream.close()
+    p.terminate()
+
+async def stream_and_listen():
+    client = AsyncCartesia(api_key=os.environ.get("CARTESIA_API_KEY"))
+
+    # Set up the websocket connection
+    ws = await client.tts.websocket()
+
+    # Create a context to send and receive audio
+    ctx = ws.context() # Generates a random context ID if not provided
+
+    send_task = asyncio.create_task(send_transcripts(ctx))
+    listen_task = asyncio.create_task(receive_and_play_audio(ctx))
+
+    # Call the two coroutine tasks concurrently
+    await asyncio.gather(send_task, listen_task)
+
+    await ws.close()
+    await client.close()
+
+asyncio.run(stream_and_listen())
+```
+
 ### Multilingual Text-to-Speech [Alpha]
 
 You can use our `sonic-multilingual` model to generate audio in multiple languages. The languages supported are available at [docs.cartesia.ai](https://docs.cartesia.ai/getting-started/available-models).

{cartesia-1.0.2 → cartesia-1.0.4}/README.md

@@ -211,6 +211,101 @@ p.terminate()
 ws.close()  # Close the websocket connection
 ```
 
+#### Conditioning speech on previous generations using WebSocket
+
+In some cases, input text may need to be streamed in. In these cases, it would be slow to wait for all the text to buffer before sending it to Cartesia's TTS service.
+
+To mitigate this, Cartesia offers audio continuations. In this setting, users can send input text, as it becomes available, over a websocket connection.
+
+To do this, we will create a `context` and sending multiple requests without awaiting the response. Then you can listen to the responses in the order they were sent.
+
+Each `context` will be closed automatically after 5 seconds of inactivity or when the `no_more_inputs` method is called. `no_more_inputs` sends a request with the `continue_=False`, which indicates no more inputs will be sent over this context
+
+```python
+import asyncio
+import os
+import pyaudio
+from cartesia import AsyncCartesia
+
+async def send_transcripts(ctx):
+    # Check out voice IDs by calling `client.voices.list()` or on [play.cartesia.ai](https://play.cartesia.ai/)
+    voice_id = "87748186-23bb-4158-a1eb-332911b0b708"
+
+    # You can check out our models at [docs.cartesia.ai](https://docs.cartesia.ai/getting-started/available-models).
+    model_id = "sonic-english"
+    
+    # You can find the supported `output_format`s in our [API Reference](https://docs.cartesia.ai/api-reference/endpoints/stream-speech-server-sent-events).
+    output_format = {
+        "container": "raw",
+        "encoding": "pcm_f32le",
+        "sample_rate": 44100,
+    }
+
+    transcripts = [
+        "Sonic and Yoshi team up in a dimension-hopping adventure! ",
+        "Racing through twisting zones, they dodge Eggman's badniks and solve ancient puzzles. ",
+        "In the Echoing Caverns, they find the Harmonic Crystal, unlocking new powers. ",
+        "Sonic's speed creates sound waves, while Yoshi's eggs become sonic bolts. ",
+        "As they near Eggman's lair, our heroes charge their abilities for an epic boss battle. ",
+        "Get ready to spin, jump, and sound-blast your way to victory in this high-octane crossover!"
+    ]
+    
+    for transcript in transcripts:
+        # Send text inputs as they become available
+        await ctx.send(
+            model_id=model_id,
+            transcript=transcript,
+            voice_id=voice_id,
+            continue_=True,
+            output_format=output_format,
+        )
+
+    # Indicate that no more inputs will be sent. Otherwise, the context will close after 5 seconds of inactivity.
+    await ctx.no_more_inputs() 
+
+async def receive_and_play_audio(ctx):
+    p = pyaudio.PyAudio()
+    stream = None
+    rate = 44100
+
+    async for output in ctx.receive():
+        buffer = output["audio"]
+
+        if not stream:
+            stream = p.open(
+                format=pyaudio.paFloat32,
+                channels=1,
+                rate=rate,
+                output=True
+            )
+
+        stream.write(buffer)
+
+    stream.stop_stream()
+    stream.close()
+    p.terminate()
+
+async def stream_and_listen():
+    client = AsyncCartesia(api_key=os.environ.get("CARTESIA_API_KEY"))
+
+    # Set up the websocket connection
+    ws = await client.tts.websocket()
+
+    # Create a context to send and receive audio
+    ctx = ws.context() # Generates a random context ID if not provided
+
+    send_task = asyncio.create_task(send_transcripts(ctx))
+    listen_task = asyncio.create_task(receive_and_play_audio(ctx))
+
+    # Call the two coroutine tasks concurrently
+    await asyncio.gather(send_task, listen_task)
+
+    await ws.close()
+    await client.close()
+
+asyncio.run(stream_and_listen())
+```
+
 ### Multilingual Text-to-Speech [Alpha]
 
 You can use our `sonic-multilingual` model to generate audio in multiple languages. The languages supported are available at [docs.cartesia.ai](https://docs.cartesia.ai/getting-started/available-models).

{cartesia-1.0.2 → cartesia-1.0.4}/cartesia/_types.py

@@ -45,15 +45,16 @@ class DeprecatedOutputFormatMapping:
         "mulaw_8000": {"container": "raw", "encoding": "pcm_mulaw", "sample_rate": 8000},
         "alaw_8000": {"container": "raw", "encoding": "pcm_alaw", "sample_rate": 8000},
     }
-
+    
+    @classmethod
     @deprecated(
         vdeprecated="1.0.1",
         vremove="1.2.0",
         reason="Old output format names are being deprecated in favor of names aligned with the Cartesia API. Use names from `OutputFormatMapping` instead.",
     )
-    def get_format_deprecated(self, format_name):
-        if format_name in self._format_mapping:
-            return self._format_mapping[format_name]
+    def get_format_deprecated(cls, format_name):
+        if format_name in cls._format_mapping:
+            return cls._format_mapping[format_name]
         else:
             raise ValueError(f"Unsupported format: {format_name}")