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