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

webscout/Provider/TTS/parler.py

@@ -0,0 +1,111 @@
+import time
+import tempfile
+from pathlib import Path
+from webscout import exceptions
+from gradio_client import Client
+import os
+from .base import BaseTTSProvider
+
+
+class ParlerTTS(BaseTTSProvider):
+    """
+    A class to interact with the Parler TTS API through Gradio Client.
+    """
+
+    def __init__(self, timeout: int = 20, proxies: dict = None):
+        """Initializes the Parler TTS client."""
+        super().__init__()
+        self.api_endpoint = "/gen_tts"
+        self.client = Client("parler-tts/parler_tts")  # Initialize the Gradio client
+        self.timeout = timeout
+
+    def tts(self, text: str, description: str = "", use_large: bool = False, verbose: bool = True) -> str:
+        """
+        Converts text to speech using the Parler TTS API.
+
+        Args:
+            text (str): The text to be converted to speech.
+            description (str, optional): Description of the desired voice characteristics. Defaults to "".
+            use_large (bool, optional): Whether to use the large model variant. Defaults to False.
+            verbose (bool, optional): Whether to log detailed information. Defaults to True.
+
+        Returns:
+            str: The filename of the saved audio file.
+
+        Raises:
+            exceptions.FailedToGenerateResponseError: If there is an error generating or saving the audio.
+        """
+        filename = Path(tempfile.mktemp(suffix=".wav", dir=self.temp_dir))
+
+        try:
+            if verbose:
+                print(f"[debug] Generating TTS with description: {description}")
+
+            result = self.client.predict(
+                text=text,
+                description=description,
+                use_large=use_large,
+                api_name=self.api_endpoint,
+            )
+
+            if isinstance(result, bytes):
+                audio_bytes = result
+            elif isinstance(result, str) and os.path.isfile(result):
+                with open(result, "rb") as f:
+                    audio_bytes = f.read()
+            else:
+                raise ValueError(f"Unexpected response from API: {result}")
+
+            self._save_audio(audio_bytes, filename, verbose)
+
+            if verbose:
+                print(f"[debug] Audio generated successfully: {filename}")
+
+            return filename.as_posix()
+
+        except Exception as e:
+            if verbose:
+                print(f"[debug] Error generating audio: {e}")
+            raise exceptions.FailedToGenerateResponseError(
+                f"Error generating audio after multiple retries: {e}"
+            ) from e
+
+    def _save_audio(self, audio_data: bytes, filename: Path, verbose: bool = True):
+        """
+        Saves the audio data to a WAV file.
+
+        Args:
+            audio_data (bytes): Audio data to save
+            filename (Path): Path to save the audio file
+            verbose (bool): Whether to print debug information
+
+        Raises:
+            exceptions.FailedToGenerateResponseError: If there is an error saving the audio.
+        """
+        try:
+            with open(filename, "wb") as f:
+                f.write(audio_data)
+            if verbose:
+                print(f"[debug] Audio saved to {filename}")
+
+        except Exception as e:
+            if verbose:
+                print(f"[debug] Error saving audio: {e}")
+            raise exceptions.FailedToGenerateResponseError(f"Error saving audio: {e}")
+
+# Example usage
+if __name__ == "__main__":
+    parlertts = ParlerTTS()
+    text = (
+        "All of the data, pre-processing, training code, and weights are released "
+        "publicly under a permissive license, enabling the community to build on our work "
+        "and develop their own powerful models."
+    )
+    voice_description = (
+        "Laura's voice is monotone yet slightly fast in delivery, with a very close "
+        "recording that almost has no background noise."
+    )
+
+    print("[debug] Generating audio...")
+    audio_file = parlertts.tts(text, description=voice_description, use_large=False)
+    print(f"Audio saved to: {audio_file}")