webscout 5.1__py3-none-any.whl → 5.2__py3-none-any.whl

webscout/Provider/TTS/streamElements.py

@@ -0,0 +1,296 @@
+import time
+import pygame
+import requests
+import pathlib
+import urllib.parse
+from typing import Union, Generator
+
+from webscout import exceptions
+from webscout.AIbase import TTSProvider
+
+class StreamElements(TTSProvider): 
+    """
+    Text-to-speech provider using the StreamElements API.
+    """
+
+    # Request headers
+    headers: dict[str, str] = {
+        "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/119.0.0.0 Safari/537.36"
+    }
+    cache_dir = pathlib.Path("./audio_cache")
+    all_voices: list[str] = [
+        "Filiz",
+        "Astrid",
+        "Tatyana",
+        "Maxim",
+        "Carmen",
+        "Ines",
+        "Cristiano",
+        "Vitoria",
+        "Ricardo",
+        "Maja",
+        "Jan",
+        "Jacek",
+        "Ewa",
+        "Ruben",
+        "Lotte",
+        "Liv",
+        "Seoyeon",
+        "Takumi",
+        "Mizuki",
+        "Giorgio",
+        "Carla",
+        "Bianca",
+        "Karl",
+        "Dora",
+        "Mathieu",
+        "Celine",
+        "Chantal",
+        "Penelope",
+        "Miguel",
+        "Mia",
+        "Enrique",
+        "Conchita",
+        "Geraint",
+        "Salli",
+        "Matthew",
+        "Kimberly",
+        "Kendra",
+        "Justin",
+        "Joey",
+        "Joanna",
+        "Ivy",
+        "Raveena",
+        "Aditi",
+        "Emma",
+        "Brian",
+        "Amy",
+        "Russell",
+        "Nicole",
+        "Vicki",
+        "Marlene",
+        "Hans",
+        "Naja",
+        "Mads",
+        "Gwyneth",
+        "Zhiyu",
+        "es-ES-Standard-A",
+        "it-IT-Standard-A",
+        "it-IT-Wavenet-A",
+        "ja-JP-Standard-A",
+        "ja-JP-Wavenet-A",
+        "ko-KR-Standard-A",
+        "ko-KR-Wavenet-A",
+        "pt-BR-Standard-A",
+        "tr-TR-Standard-A",
+        "sv-SE-Standard-A",
+        "nl-NL-Standard-A",
+        "nl-NL-Wavenet-A",
+        "en-US-Wavenet-A",
+        "en-US-Wavenet-B",
+        "en-US-Wavenet-C",
+        "en-US-Wavenet-D",
+        "en-US-Wavenet-E",
+        "en-US-Wavenet-F",
+        "en-GB-Standard-A",
+        "en-GB-Standard-B",
+        "en-GB-Standard-C",
+        "en-GB-Standard-D",
+        "en-GB-Wavenet-A",
+        "en-GB-Wavenet-B",
+        "en-GB-Wavenet-C",
+        "en-GB-Wavenet-D",
+        "en-US-Standard-B",
+        "en-US-Standard-C",
+        "en-US-Standard-D",
+        "en-US-Standard-E",
+        "de-DE-Standard-A",
+        "de-DE-Standard-B",
+        "de-DE-Wavenet-A",
+        "de-DE-Wavenet-B",
+        "de-DE-Wavenet-C",
+        "de-DE-Wavenet-D",
+        "en-AU-Standard-A",
+        "en-AU-Standard-B",
+        "en-AU-Wavenet-A",
+        "en-AU-Wavenet-B",
+        "en-AU-Wavenet-C",
+        "en-AU-Wavenet-D",
+        "en-AU-Standard-C",
+        "en-AU-Standard-D",
+        "fr-CA-Standard-A",
+        "fr-CA-Standard-B",
+        "fr-CA-Standard-C",
+        "fr-CA-Standard-D",
+        "fr-FR-Standard-C",
+        "fr-FR-Standard-D",
+        "fr-FR-Wavenet-A",
+        "fr-FR-Wavenet-B",
+        "fr-FR-Wavenet-C",
+        "fr-FR-Wavenet-D",
+        "da-DK-Wavenet-A",
+        "pl-PL-Wavenet-A",
+        "pl-PL-Wavenet-B",
+        "pl-PL-Wavenet-C",
+        "pl-PL-Wavenet-D",
+        "pt-PT-Wavenet-A",
+        "pt-PT-Wavenet-B",
+        "pt-PT-Wavenet-C",
+        "pt-PT-Wavenet-D",
+        "ru-RU-Wavenet-A",
+        "ru-RU-Wavenet-B",
+        "ru-RU-Wavenet-C",
+        "ru-RU-Wavenet-D",
+        "sk-SK-Wavenet-A",
+        "tr-TR-Wavenet-A",
+        "tr-TR-Wavenet-B",
+        "tr-TR-Wavenet-C",
+        "tr-TR-Wavenet-D",
+        "tr-TR-Wavenet-E",
+        "uk-UA-Wavenet-A",
+        "ar-XA-Wavenet-A",
+        "ar-XA-Wavenet-B",
+        "ar-XA-Wavenet-C",
+        "cs-CZ-Wavenet-A",
+        "nl-NL-Wavenet-B",
+        "nl-NL-Wavenet-C",
+        "nl-NL-Wavenet-D",
+        "nl-NL-Wavenet-E",
+        "en-IN-Wavenet-A",
+        "en-IN-Wavenet-B",
+        "en-IN-Wavenet-C",
+        "fil-PH-Wavenet-A",
+        "fi-FI-Wavenet-A",
+        "el-GR-Wavenet-A",
+        "hi-IN-Wavenet-A",
+        "hi-IN-Wavenet-B",
+        "hi-IN-Wavenet-C",
+        "hu-HU-Wavenet-A",
+        "id-ID-Wavenet-A",
+        "id-ID-Wavenet-B",
+        "id-ID-Wavenet-C",
+        "it-IT-Wavenet-B",
+        "it-IT-Wavenet-C",
+        "it-IT-Wavenet-D",
+        "ja-JP-Wavenet-B",
+        "ja-JP-Wavenet-C",
+        "ja-JP-Wavenet-D",
+        "cmn-CN-Wavenet-A",
+        "cmn-CN-Wavenet-B",
+        "cmn-CN-Wavenet-C",
+        "cmn-CN-Wavenet-D",
+        "nb-no-Wavenet-E",
+        "nb-no-Wavenet-A",
+        "nb-no-Wavenet-B",
+        "nb-no-Wavenet-C",
+        "nb-no-Wavenet-D",
+        "vi-VN-Wavenet-A",
+        "vi-VN-Wavenet-B",
+        "vi-VN-Wavenet-C",
+        "vi-VN-Wavenet-D",
+        "sr-rs-Standard-A",
+        "lv-lv-Standard-A",
+        "is-is-Standard-A",
+        "bg-bg-Standard-A",
+        "af-ZA-Standard-A",
+        "Tracy",
+        "Danny",
+        "Huihui",
+        "Yaoyao",
+        "Kangkang",
+        "HanHan",
+        "Zhiwei",
+        "Asaf",
+        "An",
+        "Stefanos",
+        "Filip",
+        "Ivan",
+        "Heidi",
+        "Herena",
+        "Kalpana",
+        "Hemant",
+        "Matej",
+        "Andika",
+        "Rizwan",
+        "Lado",
+        "Valluvar",
+        "Linda",
+        "Heather",
+        "Sean",
+        "Michael",
+        "Karsten",
+        "Guillaume",
+        "Pattara",
+        "Jakub",
+        "Szabolcs",
+        "Hoda",
+        "Naayf",
+    ]
+
+    def __init__(self, timeout: int = 20, proxies: dict = None):
+        """Initializes the StreamElements TTS client."""
+        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") -> str:
+        """
+        Converts text to speech using the StreamElements API and saves it to a file.
+        """
+        assert (
+            voice in self.all_voices
+        ), f"Voice '{voice}' not one of [{', '.join(self.all_voices)}]"
+
+        url = f"https://api.streamelements.com/kappa/v2/speech?voice={voice}&text={{{urllib.parse.quote(text)}}}"
+        filename = self.cache_dir / f"{int(time.time())}.mp3"  
+
+        try:
+            response = self.session.get(url=url, headers=self.headers, stream=True, timeout=self.timeout)
+            response.raise_for_status()
+
+            # Create the audio_cache directory if it doesn't exist
+            self.cache_dir.mkdir(parents=True, exist_ok=True) 
+
+            with open(filename, "wb") as f:
+                for chunk in response.iter_content(chunk_size=512):
+                    if chunk:
+                        f.write(chunk)
+
+            return filename.as_posix()
+
+        except requests.exceptions.RequestException as e:
+            raise exceptions.FailedToGenerateResponseError(
+                f"Failed to perform the operation: {e}"
+            )
+
+    def play_audio(self, filename: str):
+        """
+        Plays an audio file using pygame.
+
+        Args:
+            filename (str): The path to the audio file.
+
+        Raises:
+            RuntimeError: If there is an error playing the audio.
+        """
+        try:
+            pygame.mixer.init()
+            pygame.mixer.music.load(filename)
+            pygame.mixer.music.play()
+            while pygame.mixer.music.get_busy():
+                pygame.time.Clock().tick(10)
+        except Exception as e:
+            raise RuntimeError(f"Error playing audio: {e}")
+
+# Example usage
+if __name__ == "__main__":
+    streamelements = StreamElements()
+    text = "This is a test of the StreamElements text-to-speech API."
+
+    print("Generating audio...")
+    audio_file = streamelements.tts(text, voice="Brian") 
+
+    print("Playing audio...")
+    streamelements.play_audio(audio_file)

webscout/Provider/TTS/voicepod.py

@@ -0,0 +1,114 @@
+import requests
+import json
+import pygame
+import time
+from pathlib import Path
+from typing import Generator
+
+from webscout import exceptions
+from webscout.AIbase import TTSProvider
+
+class Voicepods(TTSProvider):
+    """
+    A class to interact with the Voicepods text-to-speech API.
+    """
+
+    def __init__(self, timeout: int = 20, proxies: dict = None):
+        """
+        Initializes the Voicepods API client.
+        """
+        self.api_endpoint = "https://voicepods-stream.vercel.app/api/resemble"
+        self.headers = {
+            'Accept': '*/*',
+            'Accept-Encoding': 'gzip, deflate, br, zstd',
+            'Accept-Language': 'en-US,en;q=0.9,en-IN;q=0.8',
+            'Content-Type': 'application/json',
+            'DNT': '1',
+            'Origin': 'https://voicepods-stream.vercel.app',
+            'Referer': 'https://voicepods-stream.vercel.app/',
+            'Sec-CH-UA': '"Chromium";v="128", "Not;A=Brand";v="24", "Microsoft Edge";v="128"',
+            'Sec-CH-UA-Mobile': '?0',
+            'Sec-CH-UA-Platform': '"Windows"',
+            'Sec-Fetch-Dest': 'empty',
+            'Sec-Fetch-Mode': 'cors',
+            'Sec-Fetch-Site': 'same-origin',
+            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/128.0.0.0 Safari/537.36 Edg/128.0.0.0',
+        }
+        self.session = requests.Session()
+        self.session.headers.update(self.headers)
+        if proxies:
+            self.session.proxies.update(proxies)
+        self.timeout = timeout
+        self.audio_cache_dir = Path("./audio_cache")
+
+    def tts(self, text: str) -> str:
+        """
+        Converts text to speech using the Voicepods API. 
+
+        Args:
+            text (str): The text to be converted to speech.
+
+        Returns:
+            str: The filename of the saved audio file.
+        
+        Raises:
+            exceptions.FailedToGenerateResponseError: If there is an error generating or saving the audio.
+        """
+        payload = json.dumps({"query": text})
+        filename = self.audio_cache_dir / f"{int(time.time())}.wav"  # Using timestamp for filename
+
+        try:
+            response = self.session.post(self.api_endpoint, data=payload, timeout=self.timeout)
+            response.raise_for_status()
+
+            content_type = response.headers.get('Content-Type', '')
+            if 'audio' not in content_type.lower():
+                raise ValueError(f"Unexpected content type: {content_type}")
+
+            audio_data = response.content
+            self._save_audio(audio_data, filename)
+            return filename.as_posix()  # Return the filename as a string
+
+        except requests.exceptions.RequestException as e:
+            raise exceptions.FailedToGenerateResponseError(f"Error generating audio: {e}")
+
+    def _save_audio(self, audio_data: bytes, filename: Path):
+        """Saves the audio data to a WAV file in the audio cache directory."""
+        try:
+            # Create the audio_cache directory if it doesn't exist
+            self.audio_cache_dir.mkdir(parents=True, exist_ok=True)
+
+            riff_start = audio_data.find(b'RIFF')
+            if riff_start == -1:
+                raise ValueError("RIFF header not found in audio data")
+            
+            trimmed_audio_data = audio_data[riff_start:]
+
+            with open(filename, "wb") as f:
+                f.write(trimmed_audio_data)
+
+        except Exception as e:
+            raise exceptions.FailedToGenerateResponseError(f"Error saving audio: {e}")
+
+    def play_audio(self, filename: str):
+        """Plays the audio file using pygame."""
+        try:
+            pygame.mixer.init()
+            pygame.mixer.music.load(filename)
+            pygame.mixer.music.play()
+            while pygame.mixer.music.get_busy():
+                pygame.time.Clock().tick(10)
+        except Exception as e:
+            raise RuntimeError(f"Error playing audio: {e}")
+
+# Example usage
+if __name__ == "__main__":
+
+    voicepods = Voicepods()
+    text = "Hello, this is a test of the Voicepods text-to-speech system."
+
+    print("Generating audio...")
+    audio_file = voicepods.tts(text)
+
+    print("Playing audio...")
+    voicepods.play_audio(audio_file)

webscout/Provider/TeachAnything.py

@@ -0,0 +1,177 @@
+import requests
+from requests.exceptions import RequestException
+from typing import Any, Dict
+import logging
+import random
+
+from webscout.AIutel import Conversation, Optimizers
+
+class TeachAnything:
+    """
+    A class to interact with the Teach-Anything API.
+    """
+
+
+    def __init__(
+        self,
+        is_conversation: bool = True,
+        max_tokens: int = 600,
+        timeout: int = 30,
+        intro: str = None,
+        filepath: str = None,
+        update_file: bool = True,
+        proxies: dict = {},
+        history_offset: int = 10250,
+        act: str = None,
+    ) -> None:
+        """
+        Initializes the Teach-Anything API with given parameters.
+
+        Args:
+            is_conversation (bool, optional): Flag for chatting conversationally. Defaults to True.
+            max_tokens (int, optional): Maximum number of tokens to be generated upon completion. Defaults to 600.
+            timeout (int, optional): Http request timeout. Defaults to 30.
+            intro (str, optional): Conversation introductory prompt. Defaults to None.
+            filepath (str, optional): Path to file containing conversation history. Defaults to None.
+            update_file (bool, optional): Add new prompts and responses to the file. Defaults to True.
+            proxies (dict, optional): Http request proxies. Defaults to {}.
+            history_offset (int, optional): Limit conversation history to this number of last texts. Defaults to 10250.
+            act (str|int, optional): Awesome prompt key or index. (Used as intro). Defaults to None.
+            model (str, optional): AI model to use for text generation. Defaults to "gpt4".
+        """
+
+
+        self.session = requests.Session()
+        self.is_conversation = is_conversation
+        self.max_tokens_to_sample = max_tokens
+        self.api_endpoint = "https://www.teach-anything.com/api/generate"
+        self.timeout = timeout
+        self.last_response = {}
+        self.headers = {
+            "authority": "www.teach-anything.com",
+            "path": "/api/generate",
+            "scheme": "https",
+            "accept": "*/*",
+            "accept-encoding": "gzip, deflate, br, zstd",
+            "accept-language": "en-US,en;q=0.9,en-IN;q=0.8",
+            "content-type": "application/json",
+            "origin": "https://www.teach-anything.com",
+            "referer": "https://www.teach-anything.com/",
+            "user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/127.0.0.0 Safari/537.36 Edg/127.0.0.0",
+        }
+        self.session.headers.update(self.headers)
+        self.conversation = Conversation(
+            is_conversation, self.max_tokens_to_sample, filepath, update_file
+        )
+        self.conversation.history_offset = history_offset
+        self.session.proxies = proxies
+
+    def ask(
+        self,
+        prompt: str,
+        stream: bool = False,
+        raw: bool = False,
+        optimizer: str = None,
+        conversationally: bool = False,
+    ) -> dict:
+        """Chat with AI
+
+        Args:
+            prompt (str): Prompt to be send.
+            stream (bool, optional): Whether to stream the response. Defaults to False.
+            raw (bool, optional): Whether to return the raw response. Defaults to False.
+            optimizer (str, optional): The name of the optimizer to use. Defaults to None.
+            conversationally (bool, optional): Whether to chat conversationally. Defaults to False.
+
+        Returns:
+            The response from the API.
+        """
+        conversation_prompt = self.conversation.gen_complete_prompt(prompt)
+        if optimizer:
+            if optimizer in self.__available_optimizers:
+                conversation_prompt = getattr(Optimizers, optimizer)(
+                    conversation_prompt if conversationally else prompt
+                )
+            else:
+                raise Exception(
+                    f"Optimizer is not one of {self.__available_optimizers}"
+                )
+
+        payload = {
+            "prompt": conversation_prompt
+        }
+        def for_stream():
+            response = self.session.post(self.api_endpoint, headers=self.headers, json=payload, timeout=self.timeout)
+            if not response.ok:
+                raise RequestException(
+                    f"Failed to generate response - ({response.status_code}, {response.reason}) - {response.text}"
+                )
+
+            resp = response.text
+            self.last_response.update(dict(text=resp))
+            self.conversation.update_chat_history(
+                prompt, self.get_message(self.last_response)
+            )
+            return self.last_response
+        
+        def for_non_stream():
+            for _ in for_stream():
+                pass
+            return self.last_response
+
+        return for_stream() if stream else for_non_stream()
+
+    def chat(
+        self,
+        prompt: str,
+        stream: bool = False,
+        optimizer: str = None,
+        conversationally: bool = False,
+    ) -> str:
+        """Generate response `str`
+        Args:
+            prompt (str): Prompt to be send.
+            stream (bool, optional): Flag for streaming response. Defaults to False.
+            optimizer (str, optional): Prompt optimizer name - `[code, shell_command]`. Defaults to None.
+            conversationally (bool, optional): Chat conversationally when using optimizer. Defaults to False.
+        Returns:
+            str: Response generated
+        """
+
+        def for_stream():
+            for response in self.ask(
+                prompt, True, optimizer=optimizer, conversationally=conversationally
+            ):
+                yield self.get_message(response)
+
+        def for_non_stream():
+            return self.get_message(
+                self.ask(
+                    prompt,
+                    False,
+                    optimizer=optimizer,
+                    conversationally=conversationally,
+                )
+            )
+
+        return for_stream() if stream else for_non_stream()
+
+    def get_message(self, response: dict) -> str:
+        """Retrieves message only from response
+
+        Args:
+            response (dict): Response generated by `self.ask`
+
+        Returns:
+            str: Message extracted
+        """
+        assert isinstance(response, dict), "Response should be of dict data-type only"
+        return response["text"]
+
+
+if __name__ == '__main__':
+    from rich import print
+    ai = TeachAnything()
+    response = ai.chat(input(">>> "))
+    for chunk in response:
+        print(chunk, end="", flush=True)

webscout/Provider/__init__.py

@@ -43,6 +43,10 @@ from .Youchat import *
 from .yep import *
 from .Cloudflare import *
 from .turboseek import *
+from .NetFly import *
+from .EDITEE import *
+from .TeachAnything import *
+from .AI21 import *
 __all__ = [
     'ThinkAnyAI',
     'Farfalle',
@@ -89,4 +93,8 @@ __all__ = [
     'YEPCHAT',
     'Cloudflare',
     'TurboSeek',
+    'NetFly',
+    'Editee',
+    'TeachAnything',
+    'AI21',
 ]

webscout/__init__.py

@@ -11,6 +11,8 @@ from .Bing_search import *
 import g4f
 from .YTdownloader import *
 from .Provider import *
+from .Provider.TTI import *
+from .Provider.TTS import *
 from .Extra import gguf
 from .Extra import autollama
 from .Extra import weather_ascii, weather

webscout/version.py

@@ -1,2 +1,2 @@
-__version__ = "4.8"
+__version__ = "5.2"
 __prog__ = "webscout"