webscout 8.2.9__py3-none-any.whl → 2026.1.19__py3-none-any.whl

webscout/Provider/TTS/deepgram.py

@@ -1,156 +1,286 @@
-import time
-import requests
-import pathlib
-import base64
-import tempfile
-from io import BytesIO
-from webscout import exceptions
-from concurrent.futures import ThreadPoolExecutor, as_completed
-from webscout.litagent import LitAgent
-from . import utils
-from .base import BaseTTSProvider
-
-class DeepgramTTS(BaseTTSProvider):
-    """
-    Text-to-speech provider using the DeepgramTTS API.
-    """
-    # Request headers
-    headers: dict[str, str] = {
-        "User-Agent": LitAgent().random()
-    }
-    all_voices: dict[str, str] = {
-        "Asteria": "aura-asteria-en", "Arcas": "aura-arcas-en", "Luna": "aura-luna-en",
-        "Zeus": "aura-zeus-en", "Orpheus": "aura-orpheus-en", "Angus": "aura-angus-en",
-        "Athena": "aura-athena-en", "Helios": "aura-helios-en", "Hera": "aura-hera-en",
-        "Orion": "aura-orion-en", "Perseus": "aura-perseus-en", "Stella": "aura-stella-en"
-    }
-
-    def __init__(self, timeout: int = 20, proxies: dict = None):
-        """Initializes the DeepgramTTS TTS client."""
-        super().__init__()
-        self.session = requests.Session()
-        self.session.headers.update(self.headers)
-        if proxies:
-            self.session.proxies.update(proxies)
-        self.timeout = timeout
-
-    def tts(self, text: str, voice: str = "Brian", verbose: bool = True) -> str:
-        """
-        Converts text to speech using the DeepgramTTS API and saves it to a file.
-
-        Args:
-            text (str): The text to convert to speech
-            voice (str): The voice to use for TTS (default: "Brian")
-            verbose (bool): Whether to print progress messages (default: True)
-
-        Returns:
-            str: Path to the generated audio file
-
-        Raises:
-            AssertionError: If the specified voice is not available
-            requests.RequestException: If there's an error communicating with the API
-            RuntimeError: If there's an error processing the audio
-        """
-        assert (
-            voice in self.all_voices
-        ), f"Voice '{voice}' not one of [{', '.join(self.all_voices.keys())}]"
-
-        url = "https://deepgram.com/api/ttsAudioGeneration"
-        filename = pathlib.Path(tempfile.mktemp(suffix=".mp3", dir=self.temp_dir))
-
-        # Split text into sentences using the utils module
-        sentences = utils.split_sentences(text)
-        if verbose:
-            for index, sen in enumerate(sentences):
-                print(f"[debug] Sentence {index}: {sen}")
-
-        def generate_audio_for_chunk(part_text: str, part_number: int):
-            """
-            Generate audio for a single chunk of text.
-
-            Args:
-                part_text (str): The text chunk to convert
-                part_number (int): The chunk number for ordering
-
-            Returns:
-                tuple: (part_number, audio_data)
-
-            Raises:
-                requests.RequestException: If there's an API error
-            """
-            max_retries = 3
-            retry_count = 0
-
-            while retry_count < max_retries:
-                try:
-                    payload = {"text": part_text, "model": self.all_voices[voice]}
-                    response = self.session.post(
-                        url=url,
-                        headers=self.headers,
-                        json=payload,
-                        stream=True,
-                        timeout=self.timeout
-                    )
-                    response.raise_for_status()
-
-                    response_data = response.json().get('data')
-                    if response_data:
-                        audio_data = base64.b64decode(response_data)
-                        if verbose:
-                            print(f"[debug] Chunk {part_number} processed successfully")
-                        return part_number, audio_data
-
-                    if verbose:
-                        print(f"[debug] No data received for chunk {part_number}. Attempt {retry_count + 1}/{max_retries}")
-
-                except requests.RequestException as e:
-                    if verbose:
-                        print(f"[debug] Error processing chunk {part_number}: {str(e)}. Attempt {retry_count + 1}/{max_retries}")
-                    if retry_count == max_retries - 1:
-                        raise
-
-                retry_count += 1
-                time.sleep(1)
-
-            raise RuntimeError(f"Failed to generate audio for chunk {part_number} after {max_retries} attempts")
-
-        try:
-            # Using ThreadPoolExecutor to handle requests concurrently
-            with ThreadPoolExecutor() as executor:
-                futures = {
-                    executor.submit(generate_audio_for_chunk, sentence.strip(), chunk_num): chunk_num
-                    for chunk_num, sentence in enumerate(sentences, start=1)
-                }
-
-                # Dictionary to store results with order preserved
-                audio_chunks = {}
-
-                for future in as_completed(futures):
-                    chunk_num = futures[future]
-                    try:
-                        part_number, audio_data = future.result()
-                        audio_chunks[part_number] = audio_data
-                    except Exception as e:
-                        raise RuntimeError(f"Failed to generate audio for chunk {chunk_num}: {str(e)}")
-
-                # Combine all audio chunks in order
-                with open(filename, 'wb') as f:
-                    for chunk_num in sorted(audio_chunks.keys()):
-                        f.write(audio_chunks[chunk_num])
-
-                if verbose:
-                    print(f"[debug] Audio saved to {filename}")
-                return str(filename)
-
-        except Exception as e:
-            print(f"[debug] Failed to generate audio: {str(e)}") if verbose else None
-            raise RuntimeError(f"Failed to generate audio: {str(e)}")
-
-# Example usage
-if __name__ == "__main__":
-    deepgram = DeepgramTTS()
-    text = "This is a test of the DeepgramTTS text-to-speech API. It supports multiple sentences. Let's see how it works!"
-
-    print("[debug] Generating audio...")
-    audio_file = deepgram.tts(text, voice="Asteria")
-    print(f"Audio saved to: {audio_file}")
+##################################################################################
+##  Deepgram TTS Provider                                                      ##
+##################################################################################
+import pathlib
+import tempfile
+import time
+from concurrent.futures import ThreadPoolExecutor, as_completed
+from typing import Any, Optional, Union, cast
+
+import requests
+from litprinter import ic
+
+from webscout import exceptions
+from webscout.litagent import LitAgent
+
+try:
+    from . import utils
+    from .base import BaseTTSProvider
+except ImportError:
+    # Handle direct execution
+    import os
+    import sys
+
+    sys.path.insert(0, os.path.join(os.path.dirname(__file__), "..", "..", ".."))
+    from webscout.Provider.TTS import utils
+    from webscout.Provider.TTS.base import BaseTTSProvider
+
+
+class DeepgramTTS(BaseTTSProvider):
+    """
+    Text-to-speech provider using the Deepgram Aura-2 API.
+
+    This provider follows the OpenAI TTS API structure with support for:
+    - Aura-2 next-gen voices
+    - Low-latency real-time performance
+    - Multiple output formats
+    - Concurrent generation for long texts
+    """
+
+    required_auth = False
+
+    # Request headers
+    headers: dict[str, str] = {
+        "Accept": "*/*",
+        "Accept-Language": "en-US,en;q=0.9",
+        "Content-Type": "application/json",
+        "Origin": "https://deepgram.com",
+        "Referer": "https://deepgram.com/ai-voice-generator",
+        "User-Agent": LitAgent().random(),
+    }
+
+    # Supported Aura-2 voices
+    SUPPORTED_MODELS = ["aura-2"]
+
+    SUPPORTED_VOICES = [
+        "thalia",
+        "odysseus",
+        "harmonia",
+        "theia",
+        "electra",
+        "arcas",
+        "amalthea",
+        "helena",
+        "hyperion",
+        "apollo",
+        "luna",
+        # Legacy Aura-1 voices (if still supported by the endpoint)
+        "asteria",
+        "luna",
+        "stella",
+        "athena",
+        "hera",
+        "zeus",
+        "orpheus",
+        "arcas",
+        "perseus",
+        "angus",
+        "orion",
+        "helios",
+    ]
+
+    # Voice mapping for Deepgram API compatibility
+    voice_mapping = {
+        # Aura-2
+        "thalia": "aura-2-thalia-en",
+        "odysseus": "aura-2-odysseus-en",
+        "harmonia": "aura-2-harmonia-en",
+        "theia": "aura-2-theia-en",
+        "electra": "aura-2-electra-en",
+        "arcas": "aura-2-arcas-en",
+        "amalthea": "aura-2-amalthea-en",
+        "helena": "aura-2-helena-en",
+        "hyperion": "aura-2-hyperion-en",
+        "apollo": "aura-2-apollo-en",
+        "luna": "aura-2-luna-en",
+        # Aura-1 (Backward compatibility)
+        "asteria": "aura-asteria-en",
+        "stella": "aura-stella-en",
+        "athena": "aura-athena-en",
+        "hera": "aura-hera-en",
+        "zeus": "aura-zeus-en",
+        "orpheus": "aura-orpheus-en",
+        "perseus": "aura-perseus-en",
+        "angus": "aura-angus-en",
+        "orion": "aura-orion-en",
+        "helios": "aura-helios-en",
+    }
+
+    def __init__(self, timeout: int = 30, proxies: Optional[dict] = None):
+        """
+        Initialize the Deepgram TTS client.
+
+        Args:
+            timeout (int): Request timeout in seconds
+            proxies (dict): Proxy configuration
+        """
+        super().__init__()
+        self.api_url = "https://deepgram.com/api/ttsAudioGeneration"
+        self.session = requests.Session()
+        self.session.headers.update(self.headers)
+        if proxies:
+            self.session.proxies.update(proxies)
+        self.timeout = timeout
+        self.default_voice = "thalia"
+
+    def tts(self, text: str, voice: Optional[str] = None, verbose: bool = False, **kwargs) -> str:
+        """
+        Convert text to speech using Deepgram Aura-2 API.
+
+        Args:
+            text (str): The text to convert to speech
+            voice (str): The voice to use (thalia, odysseus, etc.)
+            verbose (bool): Whether to print debug information
+            **kwargs: Additional parameters
+
+        Returns:
+            str: Path to the generated audio file
+        """
+        # Extract parameters from kwargs with defaults
+        voice = voice or kwargs.get("voice", "thalia")
+        response_format = kwargs.get("response_format", "mp3")
+        verbose = verbose if verbose is not None else kwargs.get("verbose", True)
+
+        if not text:
+            raise ValueError("Input text must be a non-empty string")
+
+        # Map voice to Deepgram API format
+        voice_id = self.voice_mapping.get(voice.lower(), f"aura-2-{voice.lower()}-en")
+
+        # Create temporary file
+        file_extension = f".{response_format}"
+        filename = pathlib.Path(
+            tempfile.NamedTemporaryFile(suffix=file_extension, dir=self.temp_dir, delete=False).name
+        )
+        sentences = utils.split_sentences(text)
+        if verbose:
+            ic.configureOutput(prefix="DEBUG| ")
+            ic(f"DeepgramTTS: Processing {len(sentences)} chunks")
+            ic.configureOutput(prefix="DEBUG| ")
+            ic(f"Voice: {voice} -> {voice_id}")
+
+        def generate_audio_for_chunk(part_text: str, part_number: int):
+            max_retries = 3
+            for attempt in range(max_retries):
+                try:
+                    payload = {
+                        "text": part_text,
+                        "model": voice_id,
+                        "demoType": "voice-generator",
+                        "params": "tag=landingpage-aivoicegenerator",
+                    }
+                    response = self.session.post(self.api_url, json=payload, timeout=self.timeout)
+                    response.raise_for_status()
+
+                    if response.content:
+                        if verbose:
+                            ic.configureOutput(prefix="DEBUG| ")
+                            ic(f"Chunk {part_number} processed successfully")
+                        return part_number, response.content
+
+                except requests.RequestException as e:
+                    if verbose:
+                        ic.configureOutput(prefix="WARNING| ")
+                        ic(
+                            f"Error processing chunk {part_number}: {e}. Retrying {attempt + 1}/{max_retries}"
+                        )
+                    time.sleep(1)
+
+            raise exceptions.FailedToGenerateResponseError(
+                f"Failed to generate audio for chunk {part_number}"
+            )
+
+        try:
+            with ThreadPoolExecutor(max_workers=3) as executor:
+                futures = {
+                    executor.submit(generate_audio_for_chunk, sentence.strip(), i): i
+                    for i, sentence in enumerate(sentences)
+                }
+
+                audio_chunks = {}
+                for future in as_completed(futures):
+                    part_num, data = future.result()
+                    audio_chunks[part_num] = data
+
+                with open(filename, "wb") as f:
+                    for i in sorted(audio_chunks.keys()):
+                        f.write(audio_chunks[i])
+
+                if verbose:
+                    ic.configureOutput(prefix="INFO| ")
+                    ic(f"Audio saved to {filename}")
+
+                return str(filename)
+
+        except Exception as e:
+            raise exceptions.FailedToGenerateResponseError(f"Deepgram TTS failed: {e}")
+
+    def create_speech(
+        self,
+        input_text: str,
+        model: Optional[str] = None,
+        voice: Optional[str] = None,
+        response_format: Optional[str] = None,
+        instructions: Optional[str] = None,
+        verbose: bool = False,
+    ) -> str:
+        return self.tts(
+            text=input_text,
+            model=model or "aura-2",
+            voice=voice or "thalia",
+            response_format=response_format or "mp3",
+            instructions=instructions,
+            verbose=verbose,
+        )
+
+    def with_streaming_response(self):
+        return StreamingResponseContextManager(self)
+
+
+class StreamingResponseContextManager:
+    def __init__(self, tts_provider: DeepgramTTS):
+        self.tts_provider = tts_provider
+
+    def create(self, **kwargs):
+        audio_file = self.tts_provider.create_speech(**kwargs)
+        return StreamingResponse(audio_file)
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        pass
+
+
+class StreamingResponse:
+    def __init__(self, audio_file: str):
+        self.audio_file = audio_file
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        pass
+
+    def stream_to_file(self, file_path: str):
+        import shutil
+
+        shutil.copy2(self.audio_file, file_path)
+
+    def iter_bytes(self, chunk_size: int = 1024):
+        with open(self.audio_file, "rb") as f:
+            while chunk := f.read(chunk_size):
+                yield chunk
+
+
+if __name__ == "__main__":
+    dg = DeepgramTTS()
+    try:
+        path = dg.tts(
+            "Deepgram Aura-2 test successful. Generating high quality speech.",
+            voice="thalia",
+            verbose=True,
+        )
+        print(f"Saved to: {path}")
+    except Exception as e:
+        print(f"Error: {e}")