webscout 7.0__py3-none-any.whl → 7.2__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 (147) 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 +20 -0
  24. webscout/Litlogger/core/logger.py +123 -0
  25. webscout/Litlogger/handlers/__init__.py +12 -0
  26. webscout/Litlogger/handlers/console.py +50 -0
  27. webscout/Litlogger/handlers/file.py +143 -0
  28. webscout/Litlogger/handlers/network.py +174 -0
  29. webscout/Litlogger/styles/__init__.py +7 -0
  30. webscout/Litlogger/styles/colors.py +231 -0
  31. webscout/Litlogger/styles/formats.py +377 -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/Blackboxai.py +136 -137
  38. webscout/Provider/ChatGPTGratis.py +226 -0
  39. webscout/Provider/Cloudflare.py +91 -78
  40. webscout/Provider/DeepSeek.py +218 -0
  41. webscout/Provider/Deepinfra.py +59 -35
  42. webscout/Provider/Free2GPT.py +131 -124
  43. webscout/Provider/Gemini.py +100 -115
  44. webscout/Provider/Glider.py +74 -59
  45. webscout/Provider/Groq.py +30 -18
  46. webscout/Provider/Jadve.py +108 -77
  47. webscout/Provider/Llama3.py +117 -94
  48. webscout/Provider/Marcus.py +191 -137
  49. webscout/Provider/Netwrck.py +62 -50
  50. webscout/Provider/PI.py +79 -124
  51. webscout/Provider/PizzaGPT.py +129 -83
  52. webscout/Provider/QwenLM.py +311 -0
  53. webscout/Provider/TTI/AiForce/__init__.py +22 -22
  54. webscout/Provider/TTI/AiForce/async_aiforce.py +257 -257
  55. webscout/Provider/TTI/AiForce/sync_aiforce.py +242 -242
  56. webscout/Provider/TTI/Nexra/__init__.py +22 -22
  57. webscout/Provider/TTI/Nexra/async_nexra.py +286 -286
  58. webscout/Provider/TTI/Nexra/sync_nexra.py +258 -258
  59. webscout/Provider/TTI/PollinationsAI/__init__.py +23 -23
  60. webscout/Provider/TTI/PollinationsAI/async_pollinations.py +330 -330
  61. webscout/Provider/TTI/PollinationsAI/sync_pollinations.py +285 -285
  62. webscout/Provider/TTI/artbit/__init__.py +22 -22
  63. webscout/Provider/TTI/artbit/async_artbit.py +184 -184
  64. webscout/Provider/TTI/artbit/sync_artbit.py +176 -176
  65. webscout/Provider/TTI/blackbox/__init__.py +4 -4
  66. webscout/Provider/TTI/blackbox/async_blackbox.py +212 -212
  67. webscout/Provider/TTI/blackbox/sync_blackbox.py +199 -199
  68. webscout/Provider/TTI/deepinfra/__init__.py +4 -4
  69. webscout/Provider/TTI/deepinfra/async_deepinfra.py +227 -227
  70. webscout/Provider/TTI/deepinfra/sync_deepinfra.py +199 -199
  71. webscout/Provider/TTI/huggingface/__init__.py +22 -22
  72. webscout/Provider/TTI/huggingface/async_huggingface.py +199 -199
  73. webscout/Provider/TTI/huggingface/sync_huggingface.py +195 -195
  74. webscout/Provider/TTI/imgninza/__init__.py +4 -4
  75. webscout/Provider/TTI/imgninza/async_ninza.py +214 -214
  76. webscout/Provider/TTI/imgninza/sync_ninza.py +209 -209
  77. webscout/Provider/TTI/talkai/__init__.py +4 -4
  78. webscout/Provider/TTI/talkai/async_talkai.py +229 -229
  79. webscout/Provider/TTI/talkai/sync_talkai.py +207 -207
  80. webscout/Provider/TTS/deepgram.py +182 -182
  81. webscout/Provider/TTS/elevenlabs.py +136 -136
  82. webscout/Provider/TTS/gesserit.py +150 -150
  83. webscout/Provider/TTS/murfai.py +138 -138
  84. webscout/Provider/TTS/parler.py +133 -134
  85. webscout/Provider/TTS/streamElements.py +360 -360
  86. webscout/Provider/TTS/utils.py +280 -280
  87. webscout/Provider/TTS/voicepod.py +116 -116
  88. webscout/Provider/TextPollinationsAI.py +74 -47
  89. webscout/Provider/WiseCat.py +193 -0
  90. webscout/Provider/__init__.py +144 -136
  91. webscout/Provider/cerebras.py +242 -227
  92. webscout/Provider/chatglm.py +204 -204
  93. webscout/Provider/dgaf.py +67 -39
  94. webscout/Provider/gaurish.py +105 -66
  95. webscout/Provider/geminiapi.py +208 -208
  96. webscout/Provider/granite.py +223 -0
  97. webscout/Provider/hermes.py +218 -218
  98. webscout/Provider/llama3mitril.py +179 -179
  99. webscout/Provider/llamatutor.py +72 -62
  100. webscout/Provider/llmchat.py +60 -35
  101. webscout/Provider/meta.py +794 -794
  102. webscout/Provider/multichat.py +331 -230
  103. webscout/Provider/typegpt.py +359 -356
  104. webscout/Provider/yep.py +5 -5
  105. webscout/__main__.py +5 -5
  106. webscout/cli.py +319 -319
  107. webscout/conversation.py +241 -242
  108. webscout/exceptions.py +328 -328
  109. webscout/litagent/__init__.py +28 -28
  110. webscout/litagent/agent.py +2 -3
  111. webscout/litprinter/__init__.py +0 -58
  112. webscout/scout/__init__.py +8 -8
  113. webscout/scout/core.py +884 -884
  114. webscout/scout/element.py +459 -459
  115. webscout/scout/parsers/__init__.py +69 -69
  116. webscout/scout/parsers/html5lib_parser.py +172 -172
  117. webscout/scout/parsers/html_parser.py +236 -236
  118. webscout/scout/parsers/lxml_parser.py +178 -178
  119. webscout/scout/utils.py +38 -38
  120. webscout/swiftcli/__init__.py +811 -811
  121. webscout/update_checker.py +2 -12
  122. webscout/version.py +1 -1
  123. webscout/webscout_search.py +1142 -1140
  124. webscout/webscout_search_async.py +635 -635
  125. webscout/zeroart/__init__.py +54 -54
  126. webscout/zeroart/base.py +60 -60
  127. webscout/zeroart/effects.py +99 -99
  128. webscout/zeroart/fonts.py +816 -816
  129. {webscout-7.0.dist-info → webscout-7.2.dist-info}/METADATA +21 -28
  130. webscout-7.2.dist-info/RECORD +217 -0
  131. webstoken/__init__.py +30 -30
  132. webstoken/classifier.py +189 -189
  133. webstoken/keywords.py +216 -216
  134. webstoken/language.py +128 -128
  135. webstoken/ner.py +164 -164
  136. webstoken/normalizer.py +35 -35
  137. webstoken/processor.py +77 -77
  138. webstoken/sentiment.py +206 -206
  139. webstoken/stemmer.py +73 -73
  140. webstoken/tagger.py +60 -60
  141. webstoken/tokenizer.py +158 -158
  142. webscout/Provider/RUBIKSAI.py +0 -272
  143. webscout-7.0.dist-info/RECORD +0 -199
  144. {webscout-7.0.dist-info → webscout-7.2.dist-info}/LICENSE.md +0 -0
  145. {webscout-7.0.dist-info → webscout-7.2.dist-info}/WHEEL +0 -0
  146. {webscout-7.0.dist-info → webscout-7.2.dist-info}/entry_points.txt +0 -0
  147. {webscout-7.0.dist-info → webscout-7.2.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)