webscout 7.1__py3-none-any.whl → 7.3__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of webscout might be problematic. Click here for more details.

Files changed (154) hide show
  1. webscout/AIauto.py +191 -191
  2. webscout/AIbase.py +122 -122
  3. webscout/AIutel.py +440 -440
  4. webscout/Bard.py +343 -161
  5. webscout/DWEBS.py +489 -492
  6. webscout/Extra/YTToolkit/YTdownloader.py +995 -995
  7. webscout/Extra/YTToolkit/__init__.py +2 -2
  8. webscout/Extra/YTToolkit/transcriber.py +476 -479
  9. webscout/Extra/YTToolkit/ytapi/channel.py +307 -307
  10. webscout/Extra/YTToolkit/ytapi/playlist.py +58 -58
  11. webscout/Extra/YTToolkit/ytapi/pool.py +7 -7
  12. webscout/Extra/YTToolkit/ytapi/utils.py +62 -62
  13. webscout/Extra/YTToolkit/ytapi/video.py +103 -103
  14. webscout/Extra/autocoder/__init__.py +9 -9
  15. webscout/Extra/autocoder/autocoder_utiles.py +199 -199
  16. webscout/Extra/autocoder/rawdog.py +5 -7
  17. webscout/Extra/autollama.py +230 -230
  18. webscout/Extra/gguf.py +3 -3
  19. webscout/Extra/weather.py +171 -171
  20. webscout/LLM.py +442 -442
  21. webscout/Litlogger/__init__.py +67 -681
  22. webscout/Litlogger/core/__init__.py +6 -0
  23. webscout/Litlogger/core/level.py +23 -0
  24. webscout/Litlogger/core/logger.py +166 -0
  25. webscout/Litlogger/handlers/__init__.py +12 -0
  26. webscout/Litlogger/handlers/console.py +33 -0
  27. webscout/Litlogger/handlers/file.py +143 -0
  28. webscout/Litlogger/handlers/network.py +173 -0
  29. webscout/Litlogger/styles/__init__.py +7 -0
  30. webscout/Litlogger/styles/colors.py +249 -0
  31. webscout/Litlogger/styles/formats.py +460 -0
  32. webscout/Litlogger/styles/text.py +87 -0
  33. webscout/Litlogger/utils/__init__.py +6 -0
  34. webscout/Litlogger/utils/detectors.py +154 -0
  35. webscout/Litlogger/utils/formatters.py +200 -0
  36. webscout/Provider/AISEARCH/DeepFind.py +250 -250
  37. webscout/Provider/AISEARCH/ISou.py +277 -0
  38. webscout/Provider/AISEARCH/__init__.py +2 -1
  39. webscout/Provider/Blackboxai.py +3 -3
  40. webscout/Provider/ChatGPTGratis.py +226 -0
  41. webscout/Provider/Cloudflare.py +3 -4
  42. webscout/Provider/DeepSeek.py +218 -0
  43. webscout/Provider/Deepinfra.py +40 -24
  44. webscout/Provider/Free2GPT.py +131 -124
  45. webscout/Provider/Gemini.py +100 -115
  46. webscout/Provider/Glider.py +3 -3
  47. webscout/Provider/Groq.py +5 -1
  48. webscout/Provider/Jadve.py +3 -3
  49. webscout/Provider/Marcus.py +191 -192
  50. webscout/Provider/Netwrck.py +3 -3
  51. webscout/Provider/PI.py +2 -2
  52. webscout/Provider/PizzaGPT.py +2 -3
  53. webscout/Provider/QwenLM.py +311 -0
  54. webscout/Provider/TTI/AiForce/__init__.py +22 -22
  55. webscout/Provider/TTI/AiForce/async_aiforce.py +257 -257
  56. webscout/Provider/TTI/AiForce/sync_aiforce.py +242 -242
  57. webscout/Provider/TTI/FreeAIPlayground/__init__.py +9 -0
  58. webscout/Provider/TTI/FreeAIPlayground/async_freeaiplayground.py +206 -0
  59. webscout/Provider/TTI/FreeAIPlayground/sync_freeaiplayground.py +192 -0
  60. webscout/Provider/TTI/Nexra/__init__.py +22 -22
  61. webscout/Provider/TTI/Nexra/async_nexra.py +286 -286
  62. webscout/Provider/TTI/Nexra/sync_nexra.py +258 -258
  63. webscout/Provider/TTI/PollinationsAI/__init__.py +23 -23
  64. webscout/Provider/TTI/PollinationsAI/async_pollinations.py +330 -330
  65. webscout/Provider/TTI/PollinationsAI/sync_pollinations.py +285 -285
  66. webscout/Provider/TTI/__init__.py +2 -1
  67. webscout/Provider/TTI/artbit/__init__.py +22 -22
  68. webscout/Provider/TTI/artbit/async_artbit.py +184 -184
  69. webscout/Provider/TTI/artbit/sync_artbit.py +176 -176
  70. webscout/Provider/TTI/blackbox/__init__.py +4 -4
  71. webscout/Provider/TTI/blackbox/async_blackbox.py +212 -212
  72. webscout/Provider/TTI/blackbox/sync_blackbox.py +199 -199
  73. webscout/Provider/TTI/deepinfra/__init__.py +4 -4
  74. webscout/Provider/TTI/deepinfra/async_deepinfra.py +227 -227
  75. webscout/Provider/TTI/deepinfra/sync_deepinfra.py +199 -199
  76. webscout/Provider/TTI/huggingface/__init__.py +22 -22
  77. webscout/Provider/TTI/huggingface/async_huggingface.py +199 -199
  78. webscout/Provider/TTI/huggingface/sync_huggingface.py +195 -195
  79. webscout/Provider/TTI/imgninza/__init__.py +4 -4
  80. webscout/Provider/TTI/imgninza/async_ninza.py +214 -214
  81. webscout/Provider/TTI/imgninza/sync_ninza.py +209 -209
  82. webscout/Provider/TTI/talkai/__init__.py +4 -4
  83. webscout/Provider/TTI/talkai/async_talkai.py +229 -229
  84. webscout/Provider/TTI/talkai/sync_talkai.py +207 -207
  85. webscout/Provider/TTS/deepgram.py +182 -182
  86. webscout/Provider/TTS/elevenlabs.py +136 -136
  87. webscout/Provider/TTS/gesserit.py +150 -150
  88. webscout/Provider/TTS/murfai.py +138 -138
  89. webscout/Provider/TTS/parler.py +133 -134
  90. webscout/Provider/TTS/streamElements.py +360 -360
  91. webscout/Provider/TTS/utils.py +280 -280
  92. webscout/Provider/TTS/voicepod.py +116 -116
  93. webscout/Provider/TextPollinationsAI.py +28 -8
  94. webscout/Provider/WiseCat.py +193 -0
  95. webscout/Provider/__init__.py +146 -134
  96. webscout/Provider/cerebras.py +242 -227
  97. webscout/Provider/chatglm.py +204 -204
  98. webscout/Provider/dgaf.py +2 -3
  99. webscout/Provider/freeaichat.py +221 -0
  100. webscout/Provider/gaurish.py +2 -3
  101. webscout/Provider/geminiapi.py +208 -208
  102. webscout/Provider/granite.py +223 -0
  103. webscout/Provider/hermes.py +218 -218
  104. webscout/Provider/llama3mitril.py +179 -179
  105. webscout/Provider/llamatutor.py +3 -3
  106. webscout/Provider/llmchat.py +2 -3
  107. webscout/Provider/meta.py +794 -794
  108. webscout/Provider/multichat.py +331 -331
  109. webscout/Provider/typegpt.py +359 -359
  110. webscout/Provider/yep.py +3 -3
  111. webscout/__init__.py +1 -0
  112. webscout/__main__.py +5 -5
  113. webscout/cli.py +319 -319
  114. webscout/conversation.py +241 -242
  115. webscout/exceptions.py +328 -328
  116. webscout/litagent/__init__.py +28 -28
  117. webscout/litagent/agent.py +2 -3
  118. webscout/litprinter/__init__.py +0 -58
  119. webscout/scout/__init__.py +8 -8
  120. webscout/scout/core.py +884 -884
  121. webscout/scout/element.py +459 -459
  122. webscout/scout/parsers/__init__.py +69 -69
  123. webscout/scout/parsers/html5lib_parser.py +172 -172
  124. webscout/scout/parsers/html_parser.py +236 -236
  125. webscout/scout/parsers/lxml_parser.py +178 -178
  126. webscout/scout/utils.py +38 -38
  127. webscout/swiftcli/__init__.py +811 -811
  128. webscout/update_checker.py +2 -12
  129. webscout/version.py +1 -1
  130. webscout/webscout_search.py +87 -6
  131. webscout/webscout_search_async.py +58 -1
  132. webscout/yep_search.py +297 -0
  133. webscout/zeroart/__init__.py +54 -54
  134. webscout/zeroart/base.py +60 -60
  135. webscout/zeroart/effects.py +99 -99
  136. webscout/zeroart/fonts.py +816 -816
  137. {webscout-7.1.dist-info → webscout-7.3.dist-info}/METADATA +62 -22
  138. webscout-7.3.dist-info/RECORD +223 -0
  139. {webscout-7.1.dist-info → webscout-7.3.dist-info}/WHEEL +1 -1
  140. webstoken/__init__.py +30 -30
  141. webstoken/classifier.py +189 -189
  142. webstoken/keywords.py +216 -216
  143. webstoken/language.py +128 -128
  144. webstoken/ner.py +164 -164
  145. webstoken/normalizer.py +35 -35
  146. webstoken/processor.py +77 -77
  147. webstoken/sentiment.py +206 -206
  148. webstoken/stemmer.py +73 -73
  149. webstoken/tagger.py +60 -60
  150. webstoken/tokenizer.py +158 -158
  151. webscout-7.1.dist-info/RECORD +0 -198
  152. {webscout-7.1.dist-info → webscout-7.3.dist-info}/LICENSE.md +0 -0
  153. {webscout-7.1.dist-info → webscout-7.3.dist-info}/entry_points.txt +0 -0
  154. {webscout-7.1.dist-info → webscout-7.3.dist-info}/top_level.txt +0 -0
@@ -1,183 +1,183 @@
1
- import time
2
- import requests
3
- import pathlib
4
- import base64
5
- from io import BytesIO
6
- from playsound import playsound
7
- from webscout import exceptions
8
- from webscout.AIbase import TTSProvider
9
- from concurrent.futures import ThreadPoolExecutor, as_completed
10
- from webscout.Litlogger import LitLogger, LogFormat, ColorScheme
11
- from webscout.litagent import LitAgent
12
- from . import utils
13
-
14
- class DeepgramTTS(TTSProvider):
15
- """
16
- Text-to-speech provider using the DeepgramTTS API.
17
- """
18
- # Request headers
19
- headers: dict[str, str] = {
20
- "User-Agent": LitAgent().random()
21
- }
22
- cache_dir = pathlib.Path("./audio_cache")
23
- all_voices: dict[str, str] = {
24
- "Asteria": "aura-asteria-en", "Arcas": "aura-arcas-en", "Luna": "aura-luna-en",
25
- "Zeus": "aura-zeus-en", "Orpheus": "aura-orpheus-en", "Angus": "aura-angus-en",
26
- "Athena": "aura-athena-en", "Helios": "aura-helios-en", "Hera": "aura-hera-en",
27
- "Orion": "aura-orion-en", "Perseus": "aura-perseus-en", "Stella": "aura-stella-en"
28
- }
29
-
30
- def __init__(self, timeout: int = 20, proxies: dict = None):
31
- """Initializes the DeepgramTTS TTS client."""
32
- self.session = requests.Session()
33
- self.session.headers.update(self.headers)
34
- if proxies:
35
- self.session.proxies.update(proxies)
36
- self.timeout = timeout
37
- self.logger = LitLogger(
38
- name="DeepgramTTS",
39
- format=LogFormat.MODERN_EMOJI,
40
- color_scheme=ColorScheme.AURORA
41
- )
42
-
43
- def tts(self, text: str, voice: str = "Brian", verbose: bool = True) -> str:
44
- """
45
- Converts text to speech using the DeepgramTTS API and saves it to a file.
46
-
47
- Args:
48
- text (str): The text to convert to speech
49
- voice (str): The voice to use for TTS (default: "Brian")
50
- verbose (bool): Whether to print progress messages (default: True)
51
-
52
- Returns:
53
- str: Path to the generated audio file
54
-
55
- Raises:
56
- AssertionError: If the specified voice is not available
57
- requests.RequestException: If there's an error communicating with the API
58
- RuntimeError: If there's an error processing the audio
59
- """
60
- assert (
61
- voice in self.all_voices
62
- ), f"Voice '{voice}' not one of [{', '.join(self.all_voices.keys())}]"
63
-
64
- url = "https://deepgram.com/api/ttsAudioGeneration"
65
- filename = self.cache_dir / f"{int(time.time())}.mp3"
66
-
67
- # Split text into sentences using the utils module
68
- sentences = utils.split_sentences(text)
69
- if verbose:
70
- for index, sen in enumerate(sentences):
71
- self.logger.debug(f"Sentence {index}: {sen}")
72
-
73
- def generate_audio_for_chunk(part_text: str, part_number: int):
74
- """
75
- Generate audio for a single chunk of text.
76
-
77
- Args:
78
- part_text (str): The text chunk to convert
79
- part_number (int): The chunk number for ordering
80
-
81
- Returns:
82
- tuple: (part_number, audio_data)
83
-
84
- Raises:
85
- requests.RequestException: If there's an API error
86
- """
87
- max_retries = 3
88
- retry_count = 0
89
-
90
- while retry_count < max_retries:
91
- try:
92
- payload = {"text": part_text, "model": self.all_voices[voice]}
93
- response = self.session.post(
94
- url=url,
95
- headers=self.headers,
96
- json=payload,
97
- stream=True,
98
- timeout=self.timeout
99
- )
100
- response.raise_for_status()
101
-
102
- response_data = response.json().get('data')
103
- if response_data:
104
- audio_data = base64.b64decode(response_data)
105
- if verbose:
106
- self.logger.success(f"Chunk {part_number} processed successfully 🎉")
107
- return part_number, audio_data
108
-
109
- if verbose:
110
- self.logger.warning(f"No data received for chunk {part_number}. Attempt {retry_count + 1}/{max_retries} ⚠️")
111
-
112
- except requests.RequestException as e:
113
- if verbose:
114
- self.logger.error(f"Error processing chunk {part_number}: {str(e)}. Attempt {retry_count + 1}/{max_retries} 🚨")
115
- if retry_count == max_retries - 1:
116
- raise
117
-
118
- retry_count += 1
119
- time.sleep(1)
120
-
121
- raise RuntimeError(f"Failed to generate audio for chunk {part_number} after {max_retries} attempts")
122
-
123
- try:
124
- # Create the audio_cache directory if it doesn't exist
125
- self.cache_dir.mkdir(parents=True, exist_ok=True)
126
-
127
- # Using ThreadPoolExecutor to handle requests concurrently
128
- with ThreadPoolExecutor() as executor:
129
- futures = {
130
- executor.submit(generate_audio_for_chunk, sentence.strip(), chunk_num): chunk_num
131
- for chunk_num, sentence in enumerate(sentences, start=1)
132
- }
133
-
134
- # Dictionary to store results with order preserved
135
- audio_chunks = {}
136
-
137
- for future in as_completed(futures):
138
- chunk_num = futures[future]
139
- try:
140
- part_number, audio_data = future.result()
141
- audio_chunks[part_number] = audio_data
142
- except Exception as e:
143
- raise RuntimeError(f"Failed to generate audio for chunk {chunk_num}: {str(e)}")
144
-
145
- # Combine all audio chunks in order
146
- with open(filename, 'wb') as f:
147
- for chunk_num in sorted(audio_chunks.keys()):
148
- f.write(audio_chunks[chunk_num])
149
-
150
- if verbose:
151
- self.logger.success(f"Audio saved to {filename} 🎉")
152
- return str(filename)
153
-
154
- except Exception as e:
155
- self.logger.critical(f"Failed to generate audio: {str(e)} 🚨")
156
- raise RuntimeError(f"Failed to generate audio: {str(e)}")
157
-
158
- def play_audio(self, filename: str):
159
- """
160
- Plays an audio file using playsound.
161
-
162
- Args:
163
- filename (str): The path to the audio file.
164
-
165
- Raises:
166
- RuntimeError: If there is an error playing the audio.
167
- """
168
- try:
169
- playsound(filename)
170
- except Exception as e:
171
- self.logger.error(f"Failed to play audio: {str(e)} 🚨")
172
- raise RuntimeError(f"Failed to play audio: {str(e)}")
173
-
174
- # Example usage
175
- if __name__ == "__main__":
176
- deepgram = DeepgramTTS()
177
- text = "This is a test of the DeepgramTTS text-to-speech API. It supports multiple sentences. Let's see how it works!"
178
-
179
- deepgram.logger.info("Generating audio...")
180
- audio_file = deepgram.tts(text, voice="Brian")
181
-
182
- deepgram.logger.info("Playing audio...")
1
+ import time
2
+ import requests
3
+ import pathlib
4
+ import base64
5
+ from io import BytesIO
6
+ from playsound import playsound
7
+ from webscout import exceptions
8
+ from webscout.AIbase import TTSProvider
9
+ from concurrent.futures import ThreadPoolExecutor, as_completed
10
+ from webscout.Litlogger import Logger, LogFormat
11
+ from webscout.litagent import LitAgent
12
+ from . import utils
13
+
14
+ class DeepgramTTS(TTSProvider):
15
+ """
16
+ Text-to-speech provider using the DeepgramTTS API.
17
+ """
18
+ # Request headers
19
+ headers: dict[str, str] = {
20
+ "User-Agent": LitAgent().random()
21
+ }
22
+ cache_dir = pathlib.Path("./audio_cache")
23
+ all_voices: dict[str, str] = {
24
+ "Asteria": "aura-asteria-en", "Arcas": "aura-arcas-en", "Luna": "aura-luna-en",
25
+ "Zeus": "aura-zeus-en", "Orpheus": "aura-orpheus-en", "Angus": "aura-angus-en",
26
+ "Athena": "aura-athena-en", "Helios": "aura-helios-en", "Hera": "aura-hera-en",
27
+ "Orion": "aura-orion-en", "Perseus": "aura-perseus-en", "Stella": "aura-stella-en"
28
+ }
29
+
30
+ def __init__(self, timeout: int = 20, proxies: dict = None):
31
+ """Initializes the DeepgramTTS TTS client."""
32
+ self.session = requests.Session()
33
+ self.session.headers.update(self.headers)
34
+ if proxies:
35
+ self.session.proxies.update(proxies)
36
+ self.timeout = timeout
37
+ self.logger = Logger(
38
+ name="DeepgramTTS",
39
+ format=LogFormat.MODERN_EMOJI,
40
+
41
+ )
42
+
43
+ def tts(self, text: str, voice: str = "Brian", verbose: bool = True) -> str:
44
+ """
45
+ Converts text to speech using the DeepgramTTS API and saves it to a file.
46
+
47
+ Args:
48
+ text (str): The text to convert to speech
49
+ voice (str): The voice to use for TTS (default: "Brian")
50
+ verbose (bool): Whether to print progress messages (default: True)
51
+
52
+ Returns:
53
+ str: Path to the generated audio file
54
+
55
+ Raises:
56
+ AssertionError: If the specified voice is not available
57
+ requests.RequestException: If there's an error communicating with the API
58
+ RuntimeError: If there's an error processing the audio
59
+ """
60
+ assert (
61
+ voice in self.all_voices
62
+ ), f"Voice '{voice}' not one of [{', '.join(self.all_voices.keys())}]"
63
+
64
+ url = "https://deepgram.com/api/ttsAudioGeneration"
65
+ filename = self.cache_dir / f"{int(time.time())}.mp3"
66
+
67
+ # Split text into sentences using the utils module
68
+ sentences = utils.split_sentences(text)
69
+ if verbose:
70
+ for index, sen in enumerate(sentences):
71
+ self.logger.debug(f"Sentence {index}: {sen}")
72
+
73
+ def generate_audio_for_chunk(part_text: str, part_number: int):
74
+ """
75
+ Generate audio for a single chunk of text.
76
+
77
+ Args:
78
+ part_text (str): The text chunk to convert
79
+ part_number (int): The chunk number for ordering
80
+
81
+ Returns:
82
+ tuple: (part_number, audio_data)
83
+
84
+ Raises:
85
+ requests.RequestException: If there's an API error
86
+ """
87
+ max_retries = 3
88
+ retry_count = 0
89
+
90
+ while retry_count < max_retries:
91
+ try:
92
+ payload = {"text": part_text, "model": self.all_voices[voice]}
93
+ response = self.session.post(
94
+ url=url,
95
+ headers=self.headers,
96
+ json=payload,
97
+ stream=True,
98
+ timeout=self.timeout
99
+ )
100
+ response.raise_for_status()
101
+
102
+ response_data = response.json().get('data')
103
+ if response_data:
104
+ audio_data = base64.b64decode(response_data)
105
+ if verbose:
106
+ self.logger.success(f"Chunk {part_number} processed successfully 🎉")
107
+ return part_number, audio_data
108
+
109
+ if verbose:
110
+ self.logger.warning(f"No data received for chunk {part_number}. Attempt {retry_count + 1}/{max_retries} ⚠️")
111
+
112
+ except requests.RequestException as e:
113
+ if verbose:
114
+ self.logger.error(f"Error processing chunk {part_number}: {str(e)}. Attempt {retry_count + 1}/{max_retries} 🚨")
115
+ if retry_count == max_retries - 1:
116
+ raise
117
+
118
+ retry_count += 1
119
+ time.sleep(1)
120
+
121
+ raise RuntimeError(f"Failed to generate audio for chunk {part_number} after {max_retries} attempts")
122
+
123
+ try:
124
+ # Create the audio_cache directory if it doesn't exist
125
+ self.cache_dir.mkdir(parents=True, exist_ok=True)
126
+
127
+ # Using ThreadPoolExecutor to handle requests concurrently
128
+ with ThreadPoolExecutor() as executor:
129
+ futures = {
130
+ executor.submit(generate_audio_for_chunk, sentence.strip(), chunk_num): chunk_num
131
+ for chunk_num, sentence in enumerate(sentences, start=1)
132
+ }
133
+
134
+ # Dictionary to store results with order preserved
135
+ audio_chunks = {}
136
+
137
+ for future in as_completed(futures):
138
+ chunk_num = futures[future]
139
+ try:
140
+ part_number, audio_data = future.result()
141
+ audio_chunks[part_number] = audio_data
142
+ except Exception as e:
143
+ raise RuntimeError(f"Failed to generate audio for chunk {chunk_num}: {str(e)}")
144
+
145
+ # Combine all audio chunks in order
146
+ with open(filename, 'wb') as f:
147
+ for chunk_num in sorted(audio_chunks.keys()):
148
+ f.write(audio_chunks[chunk_num])
149
+
150
+ if verbose:
151
+ self.logger.success(f"Audio saved to {filename} 🎉")
152
+ return str(filename)
153
+
154
+ except Exception as e:
155
+ self.logger.critical(f"Failed to generate audio: {str(e)} 🚨")
156
+ raise RuntimeError(f"Failed to generate audio: {str(e)}")
157
+
158
+ def play_audio(self, filename: str):
159
+ """
160
+ Plays an audio file using playsound.
161
+
162
+ Args:
163
+ filename (str): The path to the audio file.
164
+
165
+ Raises:
166
+ RuntimeError: If there is an error playing the audio.
167
+ """
168
+ try:
169
+ playsound(filename)
170
+ except Exception as e:
171
+ self.logger.error(f"Failed to play audio: {str(e)} 🚨")
172
+ raise RuntimeError(f"Failed to play audio: {str(e)}")
173
+
174
+ # Example usage
175
+ if __name__ == "__main__":
176
+ deepgram = DeepgramTTS()
177
+ text = "This is a test of the DeepgramTTS text-to-speech API. It supports multiple sentences. Let's see how it works!"
178
+
179
+ deepgram.logger.info("Generating audio...")
180
+ audio_file = deepgram.tts(text, voice="Brian")
181
+
182
+ deepgram.logger.info("Playing audio...")
183
183
  deepgram.play_audio(audio_file)
@@ -1,137 +1,137 @@
1
- import time
2
- import requests
3
- import pathlib
4
- from io import BytesIO
5
- from playsound import playsound
6
- from webscout import exceptions
7
- from webscout.AIbase import TTSProvider
8
- from webscout.Litlogger import LitLogger, LogFormat, ColorScheme
9
- from webscout.litagent import LitAgent
10
- from concurrent.futures import ThreadPoolExecutor, as_completed
11
- from . import utils
12
-
13
- class ElevenlabsTTS(TTSProvider):
14
- """
15
- Text-to-speech provider using the ElevenlabsTTS API.
16
- """
17
- # Request headers
18
- headers: dict[str, str] = {
19
- "User-Agent": LitAgent().random()
20
- }
21
- cache_dir = pathlib.Path("./audio_cache")
22
- all_voices: dict[str, str] = {"Brian": "nPczCjzI2devNBz1zQrb", "Alice":"Xb7hH8MSUJpSbSDYk0k2", "Bill":"pqHfZKP75CvOlQylNhV4", "Callum":"N2lVS1w4EtoT3dr4eOWO", "Charlie":"IKne3meq5aSn9XLyUdCD", "Charlotte":"XB0fDUnXU5powFXDhCwa", "Chris":"iP95p4xoKVk53GoZ742B", "Daniel":"onwK4e9ZLuTAKqWW03F9", "Eric":"cjVigY5qzO86Huf0OWal", "George":"JBFqnCBsd6RMkjVDRZzb", "Jessica":"cgSgspJ2msm6clMCkdW9", "Laura":"FGY2WhTYpPnrIDTdsKH5", "Liam":"TX3LPaxmHKxFdv7VOQHJ", "Lily":"pFZP5JQG7iQjIQuC4Bku", "Matilda":"XrExE9yKIg1WjnnlVkGX", "Sarah":"EXAVITQu4vr4xnSDxMaL", "Will":"bIHbv24MWmeRgasZH58o"}
23
-
24
- def __init__(self, timeout: int = 20, proxies: dict = None):
25
- """Initializes the ElevenlabsTTS TTS client."""
26
- self.session = requests.Session()
27
- self.session.headers.update(self.headers)
28
- if proxies:
29
- self.session.proxies.update(proxies)
30
- self.timeout = timeout
31
- self.params = {'allow_unauthenticated': '1'}
32
- self.logger = LitLogger(
33
- name="ElevenlabsTTS",
34
- format=LogFormat.MODERN_EMOJI,
35
- color_scheme=ColorScheme.AURORA
36
- )
37
-
38
- def tts(self, text: str, voice: str = "Brian", verbose:bool = True) -> str:
39
- """
40
- Converts text to speech using the ElevenlabsTTS API and saves it to a file.
41
- """
42
- assert (
43
- voice in self.all_voices
44
- ), f"Voice '{voice}' not one of [{', '.join(self.all_voices.keys())}]"
45
-
46
- filename = self.cache_dir / f"{int(time.time())}.mp3"
47
-
48
- # Split text into sentences
49
- sentences = utils.split_sentences(text)
50
-
51
- # Function to request audio for each chunk
52
- def generate_audio_for_chunk(part_text: str, part_number: int):
53
- while True:
54
- try:
55
- json_data = {'text': part_text, 'model_id': 'eleven_multilingual_v2'}
56
- response = self.session.post(f'https://api.elevenlabs.io/v1/text-to-speech/{self.all_voices[voice]}',params=self.params, headers=self.headers, json=json_data, timeout=self.timeout)
57
- response.raise_for_status()
58
-
59
- # Create the audio_cache directory if it doesn't exist
60
- self.cache_dir.mkdir(parents=True, exist_ok=True)
61
-
62
- # Check if the request was successful
63
- if response.ok and response.status_code == 200:
64
- if verbose:
65
- self.logger.success(f"Chunk {part_number} processed successfully 🎉")
66
- return part_number, response.content
67
- else:
68
- if verbose:
69
- self.logger.warning(f"No data received for chunk {part_number}. Retrying...")
70
- except requests.RequestException as e:
71
- if verbose:
72
- self.logger.error(f"Error for chunk {part_number}: {e}. Retrying... 🔄")
73
- time.sleep(1)
74
- try:
75
- # Using ThreadPoolExecutor to handle requests concurrently
76
- with ThreadPoolExecutor() as executor:
77
- futures = {executor.submit(generate_audio_for_chunk, sentence.strip(), chunk_num): chunk_num
78
- for chunk_num, sentence in enumerate(sentences, start=1)}
79
-
80
- # Dictionary to store results with order preserved
81
- audio_chunks = {}
82
-
83
- for future in as_completed(futures):
84
- chunk_num = futures[future]
85
- try:
86
- part_number, audio_data = future.result()
87
- audio_chunks[part_number] = audio_data # Store the audio data in correct sequence
88
- except Exception as e:
89
- if verbose:
90
- self.logger.error(f"Failed to generate audio for chunk {chunk_num}: {e} 🚨")
91
-
92
- # Combine audio chunks in the correct sequence
93
- combined_audio = BytesIO()
94
- for part_number in sorted(audio_chunks.keys()):
95
- combined_audio.write(audio_chunks[part_number])
96
- if verbose:
97
- self.logger.debug(f"Added chunk {part_number} to the combined file.")
98
-
99
- # Save the combined audio data to a single file
100
- with open(filename, 'wb') as f:
101
- f.write(combined_audio.getvalue())
102
- if verbose:
103
- self.logger.info(f"Final Audio Saved as {filename} 🔊")
104
- return filename.as_posix()
105
-
106
- except requests.exceptions.RequestException as e:
107
- self.logger.critical(f"Failed to perform the operation: {e} 🚨")
108
- raise exceptions.FailedToGenerateResponseError(
109
- f"Failed to perform the operation: {e}"
110
- )
111
-
112
- def play_audio(self, filename: str):
113
- """
114
- Plays an audio file using playsound.
115
-
116
- Args:
117
- filename (str): The path to the audio file.
118
-
119
- Raises:
120
- RuntimeError: If there is an error playing the audio.
121
- """
122
- try:
123
- playsound(filename)
124
- except Exception as e:
125
- self.logger.error(f"Error playing audio: {e} 🔇")
126
- raise RuntimeError(f"Error playing audio: {e}")
127
-
128
- # Example usage
129
- if __name__ == "__main__":
130
- elevenlabs = ElevenlabsTTS()
131
- text = "This is a test of the ElevenlabsTTS text-to-speech API. It supports multiple sentences and advanced logging."
132
-
133
- elevenlabs.logger.info("Generating audio...")
134
- audio_file = elevenlabs.tts(text, voice="Brian")
135
-
136
- elevenlabs.logger.info("Playing audio...")
1
+ import time
2
+ import requests
3
+ import pathlib
4
+ from io import BytesIO
5
+ from playsound import playsound
6
+ from webscout import exceptions
7
+ from webscout.AIbase import TTSProvider
8
+ from webscout.Litlogger import Logger, LogFormat
9
+ from webscout.litagent import LitAgent
10
+ from concurrent.futures import ThreadPoolExecutor, as_completed
11
+ from . import utils
12
+
13
+ class ElevenlabsTTS(TTSProvider):
14
+ """
15
+ Text-to-speech provider using the ElevenlabsTTS API.
16
+ """
17
+ # Request headers
18
+ headers: dict[str, str] = {
19
+ "User-Agent": LitAgent().random()
20
+ }
21
+ cache_dir = pathlib.Path("./audio_cache")
22
+ all_voices: dict[str, str] = {"Brian": "nPczCjzI2devNBz1zQrb", "Alice":"Xb7hH8MSUJpSbSDYk0k2", "Bill":"pqHfZKP75CvOlQylNhV4", "Callum":"N2lVS1w4EtoT3dr4eOWO", "Charlie":"IKne3meq5aSn9XLyUdCD", "Charlotte":"XB0fDUnXU5powFXDhCwa", "Chris":"iP95p4xoKVk53GoZ742B", "Daniel":"onwK4e9ZLuTAKqWW03F9", "Eric":"cjVigY5qzO86Huf0OWal", "George":"JBFqnCBsd6RMkjVDRZzb", "Jessica":"cgSgspJ2msm6clMCkdW9", "Laura":"FGY2WhTYpPnrIDTdsKH5", "Liam":"TX3LPaxmHKxFdv7VOQHJ", "Lily":"pFZP5JQG7iQjIQuC4Bku", "Matilda":"XrExE9yKIg1WjnnlVkGX", "Sarah":"EXAVITQu4vr4xnSDxMaL", "Will":"bIHbv24MWmeRgasZH58o"}
23
+
24
+ def __init__(self, timeout: int = 20, proxies: dict = None):
25
+ """Initializes the ElevenlabsTTS TTS client."""
26
+ self.session = requests.Session()
27
+ self.session.headers.update(self.headers)
28
+ if proxies:
29
+ self.session.proxies.update(proxies)
30
+ self.timeout = timeout
31
+ self.params = {'allow_unauthenticated': '1'}
32
+ self.logger = Logger(
33
+ name="ElevenlabsTTS",
34
+ format=LogFormat.MODERN_EMOJI,
35
+
36
+ )
37
+
38
+ def tts(self, text: str, voice: str = "Brian", verbose:bool = True) -> str:
39
+ """
40
+ Converts text to speech using the ElevenlabsTTS API and saves it to a file.
41
+ """
42
+ assert (
43
+ voice in self.all_voices
44
+ ), f"Voice '{voice}' not one of [{', '.join(self.all_voices.keys())}]"
45
+
46
+ filename = self.cache_dir / f"{int(time.time())}.mp3"
47
+
48
+ # Split text into sentences
49
+ sentences = utils.split_sentences(text)
50
+
51
+ # Function to request audio for each chunk
52
+ def generate_audio_for_chunk(part_text: str, part_number: int):
53
+ while True:
54
+ try:
55
+ json_data = {'text': part_text, 'model_id': 'eleven_multilingual_v2'}
56
+ response = self.session.post(f'https://api.elevenlabs.io/v1/text-to-speech/{self.all_voices[voice]}',params=self.params, headers=self.headers, json=json_data, timeout=self.timeout)
57
+ response.raise_for_status()
58
+
59
+ # Create the audio_cache directory if it doesn't exist
60
+ self.cache_dir.mkdir(parents=True, exist_ok=True)
61
+
62
+ # Check if the request was successful
63
+ if response.ok and response.status_code == 200:
64
+ if verbose:
65
+ self.logger.success(f"Chunk {part_number} processed successfully 🎉")
66
+ return part_number, response.content
67
+ else:
68
+ if verbose:
69
+ self.logger.warning(f"No data received for chunk {part_number}. Retrying...")
70
+ except requests.RequestException as e:
71
+ if verbose:
72
+ self.logger.error(f"Error for chunk {part_number}: {e}. Retrying... 🔄")
73
+ time.sleep(1)
74
+ try:
75
+ # Using ThreadPoolExecutor to handle requests concurrently
76
+ with ThreadPoolExecutor() as executor:
77
+ futures = {executor.submit(generate_audio_for_chunk, sentence.strip(), chunk_num): chunk_num
78
+ for chunk_num, sentence in enumerate(sentences, start=1)}
79
+
80
+ # Dictionary to store results with order preserved
81
+ audio_chunks = {}
82
+
83
+ for future in as_completed(futures):
84
+ chunk_num = futures[future]
85
+ try:
86
+ part_number, audio_data = future.result()
87
+ audio_chunks[part_number] = audio_data # Store the audio data in correct sequence
88
+ except Exception as e:
89
+ if verbose:
90
+ self.logger.error(f"Failed to generate audio for chunk {chunk_num}: {e} 🚨")
91
+
92
+ # Combine audio chunks in the correct sequence
93
+ combined_audio = BytesIO()
94
+ for part_number in sorted(audio_chunks.keys()):
95
+ combined_audio.write(audio_chunks[part_number])
96
+ if verbose:
97
+ self.logger.debug(f"Added chunk {part_number} to the combined file.")
98
+
99
+ # Save the combined audio data to a single file
100
+ with open(filename, 'wb') as f:
101
+ f.write(combined_audio.getvalue())
102
+ if verbose:
103
+ self.logger.info(f"Final Audio Saved as {filename} 🔊")
104
+ return filename.as_posix()
105
+
106
+ except requests.exceptions.RequestException as e:
107
+ self.logger.critical(f"Failed to perform the operation: {e} 🚨")
108
+ raise exceptions.FailedToGenerateResponseError(
109
+ f"Failed to perform the operation: {e}"
110
+ )
111
+
112
+ def play_audio(self, filename: str):
113
+ """
114
+ Plays an audio file using playsound.
115
+
116
+ Args:
117
+ filename (str): The path to the audio file.
118
+
119
+ Raises:
120
+ RuntimeError: If there is an error playing the audio.
121
+ """
122
+ try:
123
+ playsound(filename)
124
+ except Exception as e:
125
+ self.logger.error(f"Error playing audio: {e} 🔇")
126
+ raise RuntimeError(f"Error playing audio: {e}")
127
+
128
+ # Example usage
129
+ if __name__ == "__main__":
130
+ elevenlabs = ElevenlabsTTS()
131
+ text = "This is a test of the ElevenlabsTTS text-to-speech API. It supports multiple sentences and advanced logging."
132
+
133
+ elevenlabs.logger.info("Generating audio...")
134
+ audio_file = elevenlabs.tts(text, voice="Brian")
135
+
136
+ elevenlabs.logger.info("Playing audio...")
137
137
  elevenlabs.play_audio(audio_file)