smallestai 2.0.0__py3-none-any.whl → 2.1.0__py3-none-any.whl

smallest/async_tts.py

@@ -8,7 +8,7 @@ from typing import Optional, Union, List
 
 from smallest.exceptions import TTSError, APIError
 from smallest.utils import (TTSOptions, validate_input, preprocess_text, add_wav_header, chunk_text,
-                     get_smallest_languages, get_smallest_models, API_BASE_URL)
+                     get_smallest_languages, get_smallest_models, ALLOWED_AUDIO_EXTENSIONS, API_BASE_URL)
 
 
 class AsyncSmallest:
@@ -19,9 +19,7 @@ class AsyncSmallest:
         sample_rate: Optional[int] = 24000,
         voice_id: Optional[str] = "emily",
         speed: Optional[float] = 1.0,
-        add_wav_header: Optional[bool] = True,
-        transliterate: Optional[bool] = False,
-        remove_extra_silence: Optional[bool] = False
+        add_wav_header: Optional[bool] = True
     ) -> None:
         """
         AsyncSmallest Instance for asynchronous text-to-speech synthesis.
@@ -37,8 +35,6 @@ class AsyncSmallest:
         - voice_id (TTSVoices): The voice to be used for synthesis.
         - speed (float): The speed of the speech synthesis.
         - add_wav_header (bool): Whether to add a WAV header to the output audio.
-        - transliterate (bool): Whether to transliterate the text.
-        - remove_extra_silence (bool): Whether to remove extra silence from the synthesized audio.
 
         Methods:
         - get_languages: Returns a list of available languages for synthesis.
@@ -49,6 +45,9 @@ class AsyncSmallest:
         self.api_key = api_key or os.environ.get("SMALLEST_API_KEY")
         if not self.api_key:
             raise TTSError()
+        if model == "lightning-large":
+            voice_id = "lakshya"
+
         self.chunk_size = 250
 
         self.opts = TTSOptions(
@@ -57,9 +56,7 @@ class AsyncSmallest:
             voice_id=voice_id,
             api_key=self.api_key,
             add_wav_header=add_wav_header,
-            speed=speed,
-            transliterate=transliterate,
-            remove_extra_silence=remove_extra_silence,
+            speed=speed
         )
         self.session = None
 
@@ -124,6 +121,9 @@ class AsyncSmallest:
     async def synthesize(
             self,
             text: str,
+            consistency: Optional[float] = 0.5,
+            similarity: Optional[float] = 0,
+            enhancement: Optional[bool] = False,
             save_as: Optional[str] = None,
             **kwargs
         ) -> Union[bytes, None]:
@@ -134,6 +134,9 @@ class AsyncSmallest:
         - text (str): The text to be converted to speech.
         - save_as (Optional[str]): If provided, the synthesized audio will be saved to this file path.
                                    The file must have a .wav extension.
+        - consistency (Optional[float]): This parameter controls word repetition and skipping. Decrease it to prevent skipped words, and increase it to prevent repetition. Only supported in `lightning-large` model.
+        - similarity (Optional[float]): This parameter controls the similarity between the synthesized audio and the reference audio. Increase it to make the speech more similar to the reference audio. Only supported in `lightning-large` model.
+        - enhancement (Optional[bool]): Enhances speech quality at the cost of increased latency. Only supported in `lightning-large` model.
         - kwargs: Additional optional parameters to override `__init__` options for this call.
 
         Returns:
@@ -143,18 +146,29 @@ class AsyncSmallest:
         Raises:
         - TTSError: If the provided file name does not have a .wav extension when `save_as` is specified.
         - APIError: If the API request fails or returns an error.
+        - ValueError: If an unexpected parameter is passed in `kwargs`.
         """
-        should_cleanup = await self._ensure_session()
+        should_cleanup = False
+
+        if self.session is None or self.session.closed:
+            self.session = aiohttp.ClientSession()
+            should_cleanup = True  # Cleanup only if we created a new session
 
         try:
             opts = copy.deepcopy(self.opts)
+            valid_keys = set(vars(opts).keys())
+
+            invalid_keys = [key for key in kwargs if key not in valid_keys]
+            if invalid_keys:
+                raise ValueError(f"Invalid parameter(s) in kwargs: {', '.join(invalid_keys)}. Allowed parameters are: {', '.join(valid_keys)}")
+
             for key, value in kwargs.items():
                 setattr(opts, key, value)
 
-            validate_input(preprocess_text(text), opts.model, opts.sample_rate, opts.speed)
+            validate_input(preprocess_text(text), opts.model, opts.sample_rate, opts.speed, consistency, similarity, enhancement)
 
             self.chunk_size = 250
-            if opts.model == 'ligtning-large':
+            if opts.model == 'lightning-large':
                 self.chunk_size = 140
 
             chunks = chunk_text(text, self.chunk_size)
@@ -167,19 +181,23 @@ class AsyncSmallest:
                     "voice_id": opts.voice_id,
                     "add_wav_header": False,
                     "speed": opts.speed,
-                    "model": opts.model,
-                    "transliterate": opts.transliterate,
-                    "remove_extra_silence": opts.remove_extra_silence
+                    "model": opts.model
                 }
+                
+                if opts.model == "lightning-large":
+                    if consistency:
+                        payload["consistency"] = consistency
+                    if similarity:
+                        payload["similarity"] = similarity
+                    if enhancement:
+                        payload["enhancement"] = enhancement
+
 
                 headers = {
                     "Authorization": f"Bearer {self.api_key}",
                     "Content-Type": "application/json",
                 }
 
-                if not self.session:
-                    self.session = aiohttp.ClientSession()
-
                 async with self.session.post(f"{API_BASE_URL}/{opts.model}/get_speech", json=payload, headers=headers) as res:
                     if res.status != 200:
                         raise APIError(f"Failed to synthesize speech: {await res.text()}. For more information, visit https://waves.smallest.ai/")
@@ -199,7 +217,7 @@ class AsyncSmallest:
                 return add_wav_header(audio_content, opts.sample_rate)
 
             return audio_content
-        
+
         finally:
             if should_cleanup and self.session:
                 await self.session.close()
@@ -226,7 +244,6 @@ class AsyncSmallest:
         if not os.path.exists(file_path):
             raise TTSError("Invalid file path. File does not exist.")
 
-        ALLOWED_AUDIO_EXTENSIONS = ['.mp3', '.wav']
         file_extension = os.path.splitext(file_path)[1].lower()
         if file_extension not in ALLOWED_AUDIO_EXTENSIONS:
             raise TTSError(f"Invalid file type. Supported formats are: {ALLOWED_AUDIO_EXTENSIONS}")
@@ -257,4 +274,38 @@ class AsyncSmallest:
             if should_cleanup and self.session:
                 await self.session.close()
                 self.session = None
+    
+    
+    async def delete_voice(self, voice_id: str) -> str:
+        """
+        Delete a cloned voice asynchronously.
 
+        Args:
+        - voice_id (str): The ID of the voice to be deleted.
+
+        Returns:
+        - str: The response from the API.
+
+        Raises:
+        - APIError: If the API request fails or returns an error.
+        """
+        url = f"{API_BASE_URL}/lightning-large"
+        payload = {'voiceId': voice_id}
+
+        headers = {
+            "Authorization": f"Bearer {self.api_key}",
+        }
+
+        should_cleanup = await self._ensure_session()
+
+        try:
+            async with self.session.delete(url, headers=headers, json=payload) as res:
+                if res.status != 200:
+                    raise APIError(f"Failed to delete voice: {await res.text()}. For more information, visit https://waves.smallest.ai/")
+
+                return await res.text()
+        
+        finally:
+            if should_cleanup and self.session:
+                await self.session.close()
+                self.session = None

smallest/stream_tts.py

@@ -30,7 +30,7 @@ class TextToAudioStream:
 
         Args:
             tts_instance: The text-to-speech engine to use (Smallest or AsyncSmallest)
-            queue_timeout: How long to wait for new text (seconds, default: 1.0)
+            queue_timeout: How long to wait for new text (seconds, default: 5.0)
             max_retries: Number of retry attempts for failed synthesis (default: 3)
         """
         self.tts_instance = tts_instance
@@ -48,36 +48,43 @@ class TextToAudioStream:
 
     async def _stream_llm_output(self, llm_output: AsyncGenerator[str, None]) -> None:
         """
-        Streams the LLM output, splitting it into sentences based on the regex 
-        and chunk size, and adding each chunk to the queue.
+        Streams the LLM output, splitting it into chunks based on sentence boundaries
+        or space characters if no sentence boundary is found before reaching buffer_size.
 
         Parameters:
         - llm_output (AsyncGenerator[str, None]): An async generator yielding LLM output.
         """
         buffer = ""
-        last_break_index = 0
 
         async for chunk in llm_output:
             buffer += chunk
-            i = 0
-            while i < len(buffer):
-                current_chunk = buffer[:i + 1]
-                if self.sentence_end_regex.match(current_chunk):
-                    last_break_index = i
-                if len(current_chunk) >= self.buffer_size:
-                    if last_break_index > 0:
-                        self.queue.put(f'{buffer[:last_break_index + 1].replace("—", " ").strip()} ')
-                        buffer = buffer[last_break_index + 1:] 
+
+            while len(buffer) > self.buffer_size:
+                chunk_text = buffer[:self.buffer_size]
+                last_break_index = -1
+
+                # Find last sentence boundary using regex
+                for i in range(len(chunk_text) - 1, -1, -1):
+                    if self.sentence_end_regex.match(chunk_text[:i + 1]):
+                        last_break_index = i
+                        break
+
+                if last_break_index == -1:
+                    # Fallback to space if no sentence boundary found
+                    last_space = chunk_text.rfind(' ')
+                    if last_space != -1:
+                        last_break_index = last_space
                     else:
-                        # No sentence boundary, split at max chunk size
-                        self.queue.put(f'{buffer[:self.buffer_size].replace("—", " ").strip()} ')
-                        buffer = buffer[self.buffer_size:] 
-                    last_break_index = 0
-                    i = -1 
-                i += 1
-                
+                        last_break_index = self.buffer_size - 1
+
+                # Add chunk to queue and update buffer
+                self.queue.put(f'{buffer[:last_break_index + 1].replace("—", " ").strip()} ')
+                buffer = buffer[last_break_index + 1:].strip()
+
+        # Don't forget the remaining text
         if buffer:
             self.queue.put(f'{buffer.replace("—", " ").strip()} ')
+
         self.stop_flag = True
 
 
@@ -89,8 +96,7 @@ class TextToAudioStream:
             if retries < self.max_retries:
                 return self._synthesize_sync(sentence, retries + 1)
             else:
-                print(f"Synthesis failed for sentence: {sentence} - Error: {e}. Retries Exhausted, for more information, visit https://waves.smallest.ai/")
-                return None
+                raise APIError(f"Error: {e}. Retries Exhausted, for more information, visit https://waves.smallest.ai/")
             
 
     async def _synthesize_async(self, sentence: str, retries: int = 0) -> Optional[bytes]:
@@ -101,8 +107,7 @@ class TextToAudioStream:
             if retries < self.max_retries:
                 return await self._synthesize_async(sentence, retries + 1)
             else:
-                print(f"Synthesis failed for sentence: {sentence} - Error: {e}. Retries Exhausted, for more information, visit https://waves.smallest.ai/")
-                return None
+                raise APIError(f"Error: {e}. Retries Exhausted, for more information, visit https://waves.smallest.ai/")
 
 
     async def _run_synthesis(self) -> AsyncGenerator[bytes, None]:
@@ -144,10 +149,13 @@ class TextToAudioStream:
             - Streamed over a network
             - Further processed as needed
         """
-        llm_thread = Thread(target=asyncio.run, args=(self._stream_llm_output(llm_output),))
-        llm_thread.start()
+        stream_task = asyncio.create_task(self._stream_llm_output(llm_output))
 
-        async for audio_content in self._run_synthesis():
-            yield audio_content
+        try:
+            async for audio_content in self._run_synthesis():
+                yield audio_content
+        except Exception as e:
+            raise APIError(f"Error during synthesis processing: {e}")
 
-        llm_thread.join()
+        finally:
+            await stream_task

smallest/tts.py

@@ -7,7 +7,7 @@ from typing import Optional, Union, List
 
 from smallest.exceptions import TTSError, APIError
 from smallest.utils import (TTSOptions, validate_input, preprocess_text, add_wav_header, chunk_text,
-get_smallest_languages, get_smallest_models, API_BASE_URL)
+get_smallest_languages, get_smallest_models, ALLOWED_AUDIO_EXTENSIONS, API_BASE_URL)
 
 class Smallest:
     def __init__(
@@ -17,9 +17,7 @@ class Smallest:
         sample_rate: Optional[int] = 24000,
         voice_id: Optional[str] = "emily",
         speed: Optional[float] = 1.0,
-        add_wav_header: Optional[bool] = True,
-        transliterate: Optional[bool] = False,
-        remove_extra_silence: Optional[bool] = True
+        add_wav_header: Optional[bool] = True
     ) -> None:
         """
         Smallest Instance for text-to-speech synthesis.
@@ -34,8 +32,6 @@ class Smallest:
         - voice_id (TTSVoices): The voice to be used for synthesis.
         - speed (float): The speed of the speech synthesis.
         - add_wav_header (bool): Whether to add a WAV header to the output audio.
-        - transliterate (bool): Whether to transliterate the text.
-        - remove_extra_silence (bool): Whether to remove extra silence from the synthesized audio.
 
         Methods:
         - get_languages: Returns a list of available languages for synthesis.
@@ -46,7 +42,9 @@ class Smallest:
         self.api_key = api_key or os.environ.get("SMALLEST_API_KEY")
         if not self.api_key:
             raise TTSError()
-        
+        if model == "lightning-large":
+            voice_id = "lakshya"
+
         self.chunk_size = 250
         
         self.opts = TTSOptions(
@@ -55,9 +53,7 @@ class Smallest:
             voice_id=voice_id,
             api_key=self.api_key,
             add_wav_header=add_wav_header,
-            speed=speed,
-            transliterate=transliterate,
-            remove_extra_silence=remove_extra_silence
+            speed=speed
         )
     
         
@@ -102,6 +98,9 @@ class Smallest:
     def synthesize(
             self,
             text: str,
+            consistency: Optional[float] = 0.5,
+            similarity: Optional[float] = 0,
+            enhancement: Optional[bool] = False,
             save_as: Optional[str] = None,
             **kwargs
         ) -> Union[bytes, None]:
@@ -112,6 +111,9 @@ class Smallest:
         - text (str): The text to be converted to speech.
         - save_as (Optional[str]): If provided, the synthesized audio will be saved to this file path. 
                                    The file must have a .wav extension.
+        - consistency (Optional[float]): This parameter controls word repetition and skipping. Decrease it to prevent skipped words, and increase it to prevent repetition. Only supported in `lightning-large` model.
+        - similarity (Optional[float]): This parameter controls the similarity between the synthesized audio and the reference audio. Increase it to make the speech more similar to the reference audio. Only supported in `lightning-large` model.
+        - enhancement (Optional[bool]): Enhances speech quality at the cost of increased latency. Only supported in `lightning-large` model.
         - kwargs: Additional optional parameters to override `__init__` options for this call.
 
         Returns:
@@ -123,10 +125,16 @@ class Smallest:
         - APIError: If the API request fails or returns an error.
         """
         opts = copy.deepcopy(self.opts)
+        valid_keys = set(vars(opts).keys())
+
+        invalid_keys = [key for key in kwargs if key not in valid_keys]
+        if invalid_keys:
+            raise ValueError(f"Invalid parameter(s) in kwargs: {', '.join(invalid_keys)}. Allowed parameters are: {', '.join(valid_keys)}")
+                                
         for key, value in kwargs.items():
             setattr(opts, key, value)
 
-        validate_input(preprocess_text(text), opts.model, opts.sample_rate, opts.speed)
+        validate_input(preprocess_text(text), opts.model, opts.sample_rate, opts.speed, consistency, similarity, enhancement)
 
         self.chunk_size = 250
         if opts.model == "lightning-large":
@@ -142,11 +150,16 @@ class Smallest:
                 "voice_id": opts.voice_id,
                 "add_wav_header": False,
                 "speed": opts.speed,
-                "model": opts.model,
-                "transliterate": opts.transliterate,
-                "remove_extra_silence": opts.remove_extra_silence,
             }
 
+            if opts.model == "lightning-large":
+                if consistency:
+                    payload["consistency"] = consistency
+                if similarity:
+                    payload["similarity"] = similarity
+                if enhancement:
+                    payload["enhancement"] = enhancement
+
             headers = {
                 "Authorization": f"Bearer {self.api_key}",
                 "Content-Type": "application/json",
@@ -154,7 +167,7 @@ class Smallest:
 
             res = requests.post(f"{API_BASE_URL}/{opts.model}/get_speech", json=payload, headers=headers)
             if res.status_code != 200:
-                raise APIError(f"Failed to synthesize speech: {res.text}. Please check if you have set the correct API key. For more information, visit https://waves.smallest.ai/")
+                raise APIError(f"Failed to synthesize speech: {res.text}. For more information, visit https://waves.smallest.ai/")
             
             audio_content += res.content
 
@@ -193,7 +206,6 @@ class Smallest:
         if not os.path.isfile(file_path):
             raise TTSError("Invalid file path. File does not exist.")
         
-        ALLOWED_AUDIO_EXTENSIONS = ['.mp3', '.wav']
         file_extension = os.path.splitext(file_path)[1].lower()
         if file_extension not in ALLOWED_AUDIO_EXTENSIONS:
             raise TTSError(f"Invalid file type. Supported formats are: {ALLOWED_AUDIO_EXTENSIONS}")
@@ -212,3 +224,30 @@ class Smallest:
             raise APIError(f"Failed to add voice: {response.text}. For more information, visit https://waves.smallest.ai/")
         
         return json.dumps(response.json(), indent=4, ensure_ascii=False)
+
+
+    def delete_voice(self, voice_id: str) -> str:
+        """
+        Delete a cloned voice synchronously.
+
+        Args:
+        - voice_id (str): The ID of the voice to be deleted.
+
+        Returns:
+        - str: The response from the API.
+
+        Raises:
+        - APIError: If the API request fails or returns an error.
+        """
+        url = f"{API_BASE_URL}/lightning-large"
+        payload = {'voiceId': voice_id}
+
+        headers = {
+            'Authorization': f"Bearer {self.api_key}",
+        }
+
+        response = requests.delete(url, headers=headers, json=payload)
+        if response.status_code != 200:
+            raise APIError(f"Failed to delete voice: {response.text}. For more information, visit https://waves.smallest.ai/")
+        
+        return json.dumps(response.json(), indent=4, ensure_ascii=False)

smallest/utils.py

@@ -1,6 +1,7 @@
 import re
 import io
 from typing import List
+from typing import Optional
 from pydub import AudioSegment
 from dataclasses import dataclass
 from sacremoses import MosesPunctNormalizer
@@ -14,6 +15,7 @@ SENTENCE_END_REGEX = re.compile(r'.*[-.—!?,;:…।|]$')
 mpn = MosesPunctNormalizer()
 SAMPLE_WIDTH = 2
 CHANNELS = 1
+ALLOWED_AUDIO_EXTENSIONS = ['.mp3', '.wav']
 
 
 @dataclass
@@ -24,11 +26,9 @@ class TTSOptions:
     api_key: str
     add_wav_header: bool
     speed: float
-    transliterate: bool
-    remove_extra_silence: bool
 
 
-def validate_input(text: str, model: str, sample_rate: int, speed: float):
+def validate_input(text: str, model: str, sample_rate: int, speed: float, consistency: Optional[float] = None, similarity: Optional[float] = None, enhancement: Optional[bool] = None):
     if not text:
         raise ValidationError("Text cannot be empty.")
     if model not in TTSModels:
@@ -37,14 +37,20 @@ def validate_input(text: str, model: str, sample_rate: int, speed: float):
         raise ValidationError(f"Invalid sample rate: {sample_rate}. Must be between 8000 and 24000")
     if not 0.5 <= speed <= 2.0:
         raise ValidationError(f"Invalid speed: {speed}. Must be between 0.5 and 2.0")
+    if consistency is not None and not 0.0 <= consistency <= 1.0:
+        raise ValidationError(f"Invalid consistency: {consistency}. Must be between 0.0 and 1.0")
+    if similarity is not None and not 0.0 <= similarity <= 1.0:
+        raise ValidationError(f"Invalid similarity: {similarity}. Must be between 0.0 and 1.0")
+    if enhancement is not None and not isinstance(enhancement, bool):
+        raise ValidationError(f"Invalid enhancement: {enhancement}. Must be a boolean value.")
 
 
 def add_wav_header(frame_input: bytes, sample_rate: int = 24000, sample_width: int = 2, channels: int = 1) -> bytes:
-        audio = AudioSegment(data=frame_input, sample_width=sample_width, frame_rate=sample_rate, channels=channels)
-        wav_buf = io.BytesIO()
-        audio.export(wav_buf, format="wav")
-        wav_buf.seek(0)
-        return wav_buf.read()
+    audio = AudioSegment(data=frame_input, sample_width=sample_width, frame_rate=sample_rate, channels=channels)
+    wav_buf = io.BytesIO()
+    audio.export(wav_buf, format="wav")
+    wav_buf.seek(0)
+    return wav_buf.read()
 
 
 def preprocess_text(text: str) -> str:
@@ -55,11 +61,6 @@ def preprocess_text(text: str) -> str:
 
 
 def chunk_text(text: str, chunk_size: int = 250) -> List[str]:
-    """
-    Splits the input text into chunks based on sentence boundaries
-    defined by SENTENCE_END_REGEX and the maximum chunk size.
-    Only splits at valid sentence boundaries to avoid breaking words.
-    """
     chunks = []
     while text:
         if len(text) <= chunk_size:

{smallestai-2.0.0.dist-info → smallestai-2.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: smallestai
-Version: 2.0.0
+Version: 2.1.0
 Summary: Official Python client for the Smallest AI API
 Author-email: Smallest <support@smallest.ai>
 License: MIT
@@ -59,8 +59,11 @@ Currently, the library supports direct synthesis and the ability to synthesize s
   - [Aynchronous](#Synchronous)
   - [LLM to Speech](#llm-to-speech)
   - [Add your Voice](#add-your-voice)
-    - [Synchronously](#synchronously)
-    - [Asynchronously](#asynchronously)
+    - [Synchronously](#add-synchronously)
+    - [Asynchronously](#add-asynchronously)
+  - [Delete your Voice](#delete-your-voice)
+    - [Synchronously](#delete-synchronously)
+    - [Asynchronously](#delete-asynchronously)
 - [Available Methods](#available-methods)
 - [Technical Note: WAV Headers in Streaming Audio](#technical-note-wav-headers-in-streaming-audio)
 
@@ -80,14 +83,6 @@ When using an SDK in your application, make sure to pin to at least the major ve
 3. Create a new API Key and copy it.
 4. Export the API Key in your environment with the name `SMALLEST_API_KEY`, ensuring that your application can access it securely for authentication.
 
-## Best Practices for Input Text
-While the `transliterate` parameter is provided, please note that it is not fully supported and may not perform consistently across all cases. It is recommended to use the model without relying on this parameter.
-
-For optimal voice generation results:
-
-1. For English, provide the input in Latin script (e.g., "Hello, how are you?").
-2. For Hindi, provide the input in Devanagari script (e.g., "नमस्ते, आप कैसे हैं?").
-3. For code-mixed input, use Latin script for English and Devanagari script for Hindi (e.g., "Hello, आप कैसे हैं?").
 
 ## Examples
 
@@ -115,9 +110,10 @@ if __name__ == "__main__":
 - `sample_rate`: Audio sample rate (default: 24000)
 - `voice_id`: Voice ID (default: "emily")
 - `speed`: Speech speed multiplier (default: 1.0)
-- `add_wav_header`: Include WAV header in output (default: True)
-- `transliterate`: Enable text transliteration (default: False)
-- `remove_extra_silence`: Remove additional silence (default: True)  
+- `consistency`: Controls word repetition and skipping. Decrease it to prevent skipped words, and increase it to prevent repetition. Only supported in `lightning-large` model. (default: 0.5)
+- `similarity`: Controls the similarity between the synthesized audio and the reference audio. Increase it to make the speech more similar to the reference audio. Only supported in `lightning-large` model. (default: 0)
+- `enhancement`: Enhances speech quality at the cost of increased latency. Only supported in `lightning-large` model. (default: False)
+- `add_wav_header`: Whether to add a WAV header to the output audio.
 
 These parameters are part of the `Smallest` instance. They can be set when creating the instance (as shown above). However, the `synthesize` function also accepts `kwargs`, allowing you to override these parameters for a specific synthesis request.
 
@@ -141,9 +137,8 @@ import asyncio
 import aiofiles
 from smallest import AsyncSmallest
 
-client = AsyncSmallest(api_key="SMALLEST_API_KEY")
-
 async def main():
+    client = AsyncSmallest(api_key="SMALLEST_API_KEY")
     async with client as tts:
         audio_bytes = await tts.synthesize("Hello, this is a test of the async synthesis function.") 
         async with aiofiles.open("async_synthesize.wav", "wb") as f:
@@ -153,15 +148,33 @@ if __name__ == "__main__":
     asyncio.run(main())
 ```
 
+**Running Asynchronously in a Jupyter Notebook**   
+If you are using a Jupyter Notebook, use the following approach to execute the asynchronous function within an existing event loop:
+```python
+import asyncio
+import aiofiles
+from smallest import AsyncSmallest
+
+async def main():
+    client = AsyncSmallest(api_key="SMALLEST_API_KEY")
+    async with client as tts:
+        audio_bytes = await tts.synthesize("Hello, this is a test of the async synthesis function.") 
+        async with aiofiles.open("async_synthesize.wav", "wb") as f:
+            await f.write(audio_bytes) # alternatively you can use the `save_as` parameter.
+
+await main()
+```
+
 **Parameters:**    
 - `api_key`: Your API key (can be set via SMALLEST_API_KEY environment variable)
 - `model`: TTS model to use (default: "lightning")
 - `sample_rate`: Audio sample rate (default: 24000)
 - `voice_id`: Voice ID (default: "emily")
 - `speed`: Speech speed multiplier (default: 1.0)
-- `add_wav_header`: Include WAV header in output (default: True)
-- `transliterate`: Enable text transliteration (default: False)
-- `remove_extra_silence`: Remove additional silence (default: True)   
+- `consistency`: Controls word repetition and skipping. Decrease it to prevent skipped words, and increase it to prevent repetition. Only supported in `lightning-large` model.
+- `similarity`: Controls the similarity between the synthesized audio and the reference audio. Increase it to make the speech more similar to the reference audio. Only supported in `lightning-large` model.
+- `enhancement`: Enhances speech quality at the cost of increased latency. Only supported in `lightning-large` model.
+- `add_wav_header`: Whether to add a WAV header to the output audio.
 
 These parameters are part of the `AsyncSmallest` instance. They can be set when creating the instance (as shown above). However, the `synthesize` function also accepts `kwargs`, allowing you to override any of these parameters on a per-request basis.  
 
@@ -178,6 +191,58 @@ audio_bytes = await tts.synthesize(
 
 The `TextToAudioStream` class provides real-time text-to-speech processing, converting streaming text into audio output. It's particularly useful for applications like voice assistants, live captioning, or interactive chatbots that require immediate audio feedback from text generation. Supports both synchronous and asynchronous TTS instance.
 
+#### Stream through a WebSocket
+
+```python
+import asyncio
+import websockets
+from groq import Groq
+from smallest import Smallest, TextToAudioStream  
+
+# Initialize Groq (LLM) and Smallest (TTS) instances
+llm = Groq(api_key="GROQ_API_KEY")
+tts = Smallest(api_key="SMALLEST_API_KEY")
+WEBSOCKET_URL = "wss://echo.websocket.events" # Mock WebSocket server
+
+# Async function to stream text generation from LLM
+async def generate_text(prompt):
+    completion = llm.chat.completions.create(
+        messages=[{"role": "user", "content": prompt}],
+        model="llama3-8b-8192",
+        stream=True,
+    )
+
+    # Yield text as it is generated
+    for chunk in completion:
+        text = chunk.choices[0].delta.content
+        if text:
+            yield text
+
+# Main function to run the process
+async def main():
+    # Initialize the TTS processor
+    processor = TextToAudioStream(tts_instance=tts)
+
+    # Generate text from LLM
+    llm_output = generate_text("Explain text to speech like I am five in 5 sentences.")
+
+    # Stream the generated speech throught a websocket
+    async with websockets.connect(WEBSOCKET_URL) as ws:
+        print("Connected to WebSocket server.")
+
+        # Stream the generated speech
+        async for audio_chunk in processor.process(llm_output):
+            await ws.send(audio_chunk)  # Send audio chunk
+            echoed_data = await ws.recv()  # Receive the echoed message
+            print("Received from server:", echoed_data[:20], "...")  # Print first 20 bytes
+
+        print("WebSocket connection closed.")
+
+if __name__ == "__main__":
+    asyncio.run(main())
+```
+
+#### Save to a File
 ```python
 import wave
 import asyncio
@@ -245,12 +310,12 @@ The processor yields raw audio data chunks without WAV headers for streaming eff
 ## Add your Voice   
 The Smallest AI SDK allows you to clone your voice by uploading an audio file. This feature is available both synchronously and asynchronously, making it flexible for different use cases. Below are examples of how to use this functionality.  
 
-### Synchronously
+### Add Synchronously
 ```python
 from smallest import Smallest
 
 def main():
-    client = Smallest(api_key="YOUR_API_KEY")
+    client = Smallest(api_key="SMALLEST_API_KEY")
     res = client.add_voice(display_name="My Voice", file_path="my_voice.wav")
     print(res)
 
@@ -258,13 +323,13 @@ if __name__ == "__main__":
     main()
 ```  
 
-### Asynchronously
+### Add Asynchronously
 ```python
 import asyncio
 from smallest import AsyncSmallest
 
 async def main():
-    client = AsyncSmallest(api_key="YOUR_API_KEY")
+    client = AsyncSmallest(api_key="SMALLEST_API_KEY")
     res = await client.add_voice(display_name="My Voice", file_path="my_voice.wav")
     print(res)
 
@@ -272,6 +337,36 @@ if __name__ == "__main__":
     asyncio.run(main())
 ```
 
+## Delete your Voice
+The Smallest AI SDK allows you to delete your cloned voice. This feature is available both synchronously and asynchronously, making it flexible for different use cases. Below are examples of how to use this functionality.
+
+### Delete Synchronously
+```python
+from smallest import Smallest
+
+def main():
+    client = Smallest(api_key="SMALLEST_API_KEY")
+    res = client.delete_voice(voice_id="voice_id")
+    print(res)
+
+if __name__ == "__main__":
+    main()
+```
+
+### Delete Asynchronously
+```python
+import asyncio
+from smallest import AsyncSmallest
+
+async def main():
+    client = AsyncSmallest(api_key="SMALLEST_API_KEY")
+    res = await client.delete_voice(voice_id="voice_id")
+    print(res)
+
+if __name__ == "__main__":
+    asyncio.run(main())
+```
+
 ## Available Methods
 
 ```python

smallestai-2.1.0.dist-info/RECORD

@@ -0,0 +1,12 @@
+smallest/__init__.py,sha256=vaoIBml_IobavpVvFazB86iikg2iEy4h3ddxqv_0Fy4,190
+smallest/async_tts.py,sha256=Jr7IID5tJrnMx_d2217foUJqfFvAFsddvy_0HG5tKGc,11905
+smallest/exceptions.py,sha256=nY6I8fCXe2By54CytQ0-i3hFiYtt8TYAKj0g6OYsCjc,585
+smallest/models.py,sha256=g2e_4nU5P48vyXZandKLWqZC1TkoEGeLvYKqJIqurSI,83
+smallest/stream_tts.py,sha256=dUxoY0VkXecsMZ41QA8RkX4t_pD5-7mMIJhaB01tQrk,6512
+smallest/tts.py,sha256=bSL7EYmLpd5yT42dbUXVb-IgZ_xIcXpyHvCu2-hHtMs,10024
+smallest/utils.py,sha256=HDpDjPkUeeQLqDhrV-zPTLtOH9hJueae0q9SNq486GQ,3396
+smallestai-2.1.0.dist-info/LICENSE,sha256=kK3HNKhN7luQhkjkNWIvy9_gizbEDUM4mSv_HWq9uuM,1068
+smallestai-2.1.0.dist-info/METADATA,sha256=BwCUFiVZTRActimZBQcPJg8vHJy0M-6vYA_yHvaFpDk,14904
+smallestai-2.1.0.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
+smallestai-2.1.0.dist-info/top_level.txt,sha256=i5ktbWkG-2aS28vrYTeuhKtA-tY2ZG7SHgLHi87QTLw,9
+smallestai-2.1.0.dist-info/RECORD,,

{smallestai-2.0.0.dist-info → smallestai-2.1.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.8.0)
+Generator: setuptools (75.8.2)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

smallestai-2.0.0.dist-info/RECORD

@@ -1,12 +0,0 @@
-smallest/__init__.py,sha256=vaoIBml_IobavpVvFazB86iikg2iEy4h3ddxqv_0Fy4,190
-smallest/async_tts.py,sha256=5qW7owlMeSWFx0rpn9dYfbO76mmNY0DXcytNjLfbbz8,9727
-smallest/exceptions.py,sha256=nY6I8fCXe2By54CytQ0-i3hFiYtt8TYAKj0g6OYsCjc,585
-smallest/models.py,sha256=g2e_4nU5P48vyXZandKLWqZC1TkoEGeLvYKqJIqurSI,83
-smallest/stream_tts.py,sha256=SeP9A9zXJWiV62Eezv0L1J5sRIR304Llc_mwVtOOSUI,6348
-smallest/tts.py,sha256=xBBEk_byRPGT6SYkE6qvhfEupgHl6XBdAqtxmzw2rF8,8311
-smallest/utils.py,sha256=FCZkvbbHJBoN0jpBSqmt1hJjvks56t8i82we4XnqjYk,3016
-smallestai-2.0.0.dist-info/LICENSE,sha256=kK3HNKhN7luQhkjkNWIvy9_gizbEDUM4mSv_HWq9uuM,1068
-smallestai-2.0.0.dist-info/METADATA,sha256=EIyZZqzAvHgQ7jfEs5x5LUx3HjzoCUhzJoXfkb3CuoI,11538
-smallestai-2.0.0.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-smallestai-2.0.0.dist-info/top_level.txt,sha256=i5ktbWkG-2aS28vrYTeuhKtA-tY2ZG7SHgLHi87QTLw,9
-smallestai-2.0.0.dist-info/RECORD,,