webscout 8.2.2__py3-none-any.whl → 8.2.7__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.
- webscout/AIauto.py +112 -22
- webscout/AIbase.py +144 -7
- webscout/AIutel.py +249 -131
- webscout/Bard.py +579 -206
- webscout/DWEBS.py +78 -35
- webscout/__init__.py +0 -1
- webscout/cli.py +256 -0
- webscout/conversation.py +307 -436
- webscout/exceptions.py +23 -0
- webscout/prompt_manager.py +56 -42
- webscout/version.py +1 -1
- webscout/webscout_search.py +65 -47
- webscout/webscout_search_async.py +81 -126
- webscout/yep_search.py +93 -43
- {webscout-8.2.2.dist-info → webscout-8.2.7.dist-info}/METADATA +172 -52
- webscout-8.2.7.dist-info/RECORD +26 -0
- {webscout-8.2.2.dist-info → webscout-8.2.7.dist-info}/WHEEL +1 -1
- webscout-8.2.7.dist-info/entry_points.txt +3 -0
- webscout-8.2.7.dist-info/top_level.txt +1 -0
- inferno/__init__.py +0 -6
- inferno/__main__.py +0 -9
- inferno/cli.py +0 -6
- webscout/Extra/GitToolkit/__init__.py +0 -10
- webscout/Extra/GitToolkit/gitapi/__init__.py +0 -12
- webscout/Extra/GitToolkit/gitapi/repository.py +0 -195
- webscout/Extra/GitToolkit/gitapi/user.py +0 -96
- webscout/Extra/GitToolkit/gitapi/utils.py +0 -62
- webscout/Extra/YTToolkit/YTdownloader.py +0 -957
- webscout/Extra/YTToolkit/__init__.py +0 -3
- webscout/Extra/YTToolkit/transcriber.py +0 -476
- webscout/Extra/YTToolkit/ytapi/__init__.py +0 -6
- webscout/Extra/YTToolkit/ytapi/channel.py +0 -307
- webscout/Extra/YTToolkit/ytapi/errors.py +0 -13
- webscout/Extra/YTToolkit/ytapi/extras.py +0 -45
- webscout/Extra/YTToolkit/ytapi/https.py +0 -88
- webscout/Extra/YTToolkit/ytapi/patterns.py +0 -61
- webscout/Extra/YTToolkit/ytapi/playlist.py +0 -59
- webscout/Extra/YTToolkit/ytapi/pool.py +0 -8
- webscout/Extra/YTToolkit/ytapi/query.py +0 -40
- webscout/Extra/YTToolkit/ytapi/stream.py +0 -63
- webscout/Extra/YTToolkit/ytapi/utils.py +0 -62
- webscout/Extra/YTToolkit/ytapi/video.py +0 -232
- webscout/Extra/__init__.py +0 -7
- webscout/Extra/autocoder/__init__.py +0 -9
- webscout/Extra/autocoder/autocoder.py +0 -849
- webscout/Extra/autocoder/autocoder_utiles.py +0 -332
- webscout/Extra/gguf.py +0 -682
- webscout/Extra/tempmail/__init__.py +0 -28
- webscout/Extra/tempmail/async_utils.py +0 -141
- webscout/Extra/tempmail/base.py +0 -161
- webscout/Extra/tempmail/cli.py +0 -187
- webscout/Extra/tempmail/emailnator.py +0 -84
- webscout/Extra/tempmail/mail_tm.py +0 -361
- webscout/Extra/tempmail/temp_mail_io.py +0 -292
- webscout/Extra/weather.py +0 -194
- webscout/Extra/weather_ascii.py +0 -76
- webscout/LLM.py +0 -442
- webscout/Litlogger/__init__.py +0 -67
- webscout/Litlogger/core/__init__.py +0 -6
- webscout/Litlogger/core/level.py +0 -23
- webscout/Litlogger/core/logger.py +0 -165
- webscout/Litlogger/handlers/__init__.py +0 -12
- webscout/Litlogger/handlers/console.py +0 -33
- webscout/Litlogger/handlers/file.py +0 -143
- webscout/Litlogger/handlers/network.py +0 -173
- webscout/Litlogger/styles/__init__.py +0 -7
- webscout/Litlogger/styles/colors.py +0 -249
- webscout/Litlogger/styles/formats.py +0 -458
- webscout/Litlogger/styles/text.py +0 -87
- webscout/Litlogger/utils/__init__.py +0 -6
- webscout/Litlogger/utils/detectors.py +0 -153
- webscout/Litlogger/utils/formatters.py +0 -200
- webscout/Local/__init__.py +0 -12
- webscout/Local/__main__.py +0 -9
- webscout/Local/api.py +0 -576
- webscout/Local/cli.py +0 -516
- webscout/Local/config.py +0 -75
- webscout/Local/llm.py +0 -287
- webscout/Local/model_manager.py +0 -253
- webscout/Local/server.py +0 -721
- webscout/Local/utils.py +0 -93
- webscout/Provider/AI21.py +0 -177
- webscout/Provider/AISEARCH/DeepFind.py +0 -250
- webscout/Provider/AISEARCH/ISou.py +0 -256
- webscout/Provider/AISEARCH/Perplexity.py +0 -359
- webscout/Provider/AISEARCH/__init__.py +0 -10
- webscout/Provider/AISEARCH/felo_search.py +0 -228
- webscout/Provider/AISEARCH/genspark_search.py +0 -208
- webscout/Provider/AISEARCH/hika_search.py +0 -194
- webscout/Provider/AISEARCH/iask_search.py +0 -436
- webscout/Provider/AISEARCH/monica_search.py +0 -246
- webscout/Provider/AISEARCH/scira_search.py +0 -324
- webscout/Provider/AISEARCH/webpilotai_search.py +0 -281
- webscout/Provider/Aitopia.py +0 -292
- webscout/Provider/AllenAI.py +0 -413
- webscout/Provider/Andi.py +0 -228
- webscout/Provider/Blackboxai.py +0 -229
- webscout/Provider/C4ai.py +0 -432
- webscout/Provider/ChatGPTClone.py +0 -226
- webscout/Provider/ChatGPTES.py +0 -237
- webscout/Provider/ChatGPTGratis.py +0 -194
- webscout/Provider/Chatify.py +0 -175
- webscout/Provider/Cloudflare.py +0 -273
- webscout/Provider/Cohere.py +0 -208
- webscout/Provider/DeepSeek.py +0 -196
- webscout/Provider/Deepinfra.py +0 -297
- webscout/Provider/ElectronHub.py +0 -709
- webscout/Provider/ExaAI.py +0 -261
- webscout/Provider/ExaChat.py +0 -342
- webscout/Provider/Free2GPT.py +0 -241
- webscout/Provider/GPTWeb.py +0 -193
- webscout/Provider/Gemini.py +0 -169
- webscout/Provider/GithubChat.py +0 -367
- webscout/Provider/Glider.py +0 -211
- webscout/Provider/Groq.py +0 -670
- webscout/Provider/HF_space/__init__.py +0 -0
- webscout/Provider/HF_space/qwen_qwen2.py +0 -206
- webscout/Provider/HeckAI.py +0 -233
- webscout/Provider/HuggingFaceChat.py +0 -462
- webscout/Provider/Hunyuan.py +0 -272
- webscout/Provider/Jadve.py +0 -266
- webscout/Provider/Koboldai.py +0 -381
- webscout/Provider/LambdaChat.py +0 -392
- webscout/Provider/Llama.py +0 -200
- webscout/Provider/Llama3.py +0 -204
- webscout/Provider/Marcus.py +0 -148
- webscout/Provider/Netwrck.py +0 -228
- webscout/Provider/OLLAMA.py +0 -396
- webscout/Provider/OPENAI/__init__.py +0 -25
- webscout/Provider/OPENAI/base.py +0 -46
- webscout/Provider/OPENAI/c4ai.py +0 -367
- webscout/Provider/OPENAI/chatgpt.py +0 -549
- webscout/Provider/OPENAI/chatgptclone.py +0 -460
- webscout/Provider/OPENAI/deepinfra.py +0 -272
- webscout/Provider/OPENAI/e2b.py +0 -1350
- webscout/Provider/OPENAI/exaai.py +0 -404
- webscout/Provider/OPENAI/exachat.py +0 -433
- webscout/Provider/OPENAI/freeaichat.py +0 -352
- webscout/Provider/OPENAI/glider.py +0 -316
- webscout/Provider/OPENAI/heckai.py +0 -337
- webscout/Provider/OPENAI/llmchatco.py +0 -327
- webscout/Provider/OPENAI/netwrck.py +0 -348
- webscout/Provider/OPENAI/opkfc.py +0 -488
- webscout/Provider/OPENAI/scirachat.py +0 -463
- webscout/Provider/OPENAI/sonus.py +0 -294
- webscout/Provider/OPENAI/standardinput.py +0 -425
- webscout/Provider/OPENAI/textpollinations.py +0 -285
- webscout/Provider/OPENAI/toolbaz.py +0 -405
- webscout/Provider/OPENAI/typegpt.py +0 -346
- webscout/Provider/OPENAI/uncovrAI.py +0 -455
- webscout/Provider/OPENAI/utils.py +0 -211
- webscout/Provider/OPENAI/venice.py +0 -413
- webscout/Provider/OPENAI/wisecat.py +0 -381
- webscout/Provider/OPENAI/writecream.py +0 -156
- webscout/Provider/OPENAI/x0gpt.py +0 -371
- webscout/Provider/OPENAI/yep.py +0 -327
- webscout/Provider/OpenGPT.py +0 -199
- webscout/Provider/Openai.py +0 -496
- webscout/Provider/PI.py +0 -344
- webscout/Provider/Perplexitylabs.py +0 -415
- webscout/Provider/Phind.py +0 -535
- webscout/Provider/PizzaGPT.py +0 -198
- webscout/Provider/QwenLM.py +0 -254
- webscout/Provider/Reka.py +0 -214
- webscout/Provider/StandardInput.py +0 -278
- webscout/Provider/TTI/AiForce/__init__.py +0 -22
- webscout/Provider/TTI/AiForce/async_aiforce.py +0 -224
- webscout/Provider/TTI/AiForce/sync_aiforce.py +0 -245
- webscout/Provider/TTI/FreeAIPlayground/__init__.py +0 -9
- webscout/Provider/TTI/FreeAIPlayground/async_freeaiplayground.py +0 -181
- webscout/Provider/TTI/FreeAIPlayground/sync_freeaiplayground.py +0 -180
- webscout/Provider/TTI/ImgSys/__init__.py +0 -23
- webscout/Provider/TTI/ImgSys/async_imgsys.py +0 -202
- webscout/Provider/TTI/ImgSys/sync_imgsys.py +0 -195
- webscout/Provider/TTI/MagicStudio/__init__.py +0 -2
- webscout/Provider/TTI/MagicStudio/async_magicstudio.py +0 -111
- webscout/Provider/TTI/MagicStudio/sync_magicstudio.py +0 -109
- webscout/Provider/TTI/Nexra/__init__.py +0 -22
- webscout/Provider/TTI/Nexra/async_nexra.py +0 -286
- webscout/Provider/TTI/Nexra/sync_nexra.py +0 -258
- webscout/Provider/TTI/PollinationsAI/__init__.py +0 -23
- webscout/Provider/TTI/PollinationsAI/async_pollinations.py +0 -311
- webscout/Provider/TTI/PollinationsAI/sync_pollinations.py +0 -265
- webscout/Provider/TTI/__init__.py +0 -12
- webscout/Provider/TTI/aiarta/__init__.py +0 -2
- webscout/Provider/TTI/aiarta/async_aiarta.py +0 -482
- webscout/Provider/TTI/aiarta/sync_aiarta.py +0 -440
- webscout/Provider/TTI/artbit/__init__.py +0 -22
- webscout/Provider/TTI/artbit/async_artbit.py +0 -155
- webscout/Provider/TTI/artbit/sync_artbit.py +0 -148
- webscout/Provider/TTI/fastflux/__init__.py +0 -22
- webscout/Provider/TTI/fastflux/async_fastflux.py +0 -261
- webscout/Provider/TTI/fastflux/sync_fastflux.py +0 -252
- webscout/Provider/TTI/huggingface/__init__.py +0 -22
- webscout/Provider/TTI/huggingface/async_huggingface.py +0 -199
- webscout/Provider/TTI/huggingface/sync_huggingface.py +0 -195
- webscout/Provider/TTI/piclumen/__init__.py +0 -23
- webscout/Provider/TTI/piclumen/async_piclumen.py +0 -268
- webscout/Provider/TTI/piclumen/sync_piclumen.py +0 -233
- webscout/Provider/TTI/pixelmuse/__init__.py +0 -4
- webscout/Provider/TTI/pixelmuse/async_pixelmuse.py +0 -249
- webscout/Provider/TTI/pixelmuse/sync_pixelmuse.py +0 -182
- webscout/Provider/TTI/talkai/__init__.py +0 -4
- webscout/Provider/TTI/talkai/async_talkai.py +0 -229
- webscout/Provider/TTI/talkai/sync_talkai.py +0 -207
- webscout/Provider/TTS/__init__.py +0 -7
- webscout/Provider/TTS/deepgram.py +0 -156
- webscout/Provider/TTS/elevenlabs.py +0 -111
- webscout/Provider/TTS/gesserit.py +0 -127
- webscout/Provider/TTS/murfai.py +0 -113
- webscout/Provider/TTS/parler.py +0 -111
- webscout/Provider/TTS/speechma.py +0 -180
- webscout/Provider/TTS/streamElements.py +0 -333
- webscout/Provider/TTS/utils.py +0 -280
- webscout/Provider/TeachAnything.py +0 -187
- webscout/Provider/TextPollinationsAI.py +0 -231
- webscout/Provider/TwoAI.py +0 -199
- webscout/Provider/Venice.py +0 -219
- webscout/Provider/VercelAI.py +0 -234
- webscout/Provider/WebSim.py +0 -228
- webscout/Provider/WiseCat.py +0 -196
- webscout/Provider/Writecream.py +0 -211
- webscout/Provider/WritingMate.py +0 -197
- webscout/Provider/Youchat.py +0 -330
- webscout/Provider/__init__.py +0 -198
- webscout/Provider/ai4chat.py +0 -202
- webscout/Provider/aimathgpt.py +0 -189
- webscout/Provider/akashgpt.py +0 -342
- webscout/Provider/askmyai.py +0 -158
- webscout/Provider/asksteve.py +0 -203
- webscout/Provider/bagoodex.py +0 -145
- webscout/Provider/cerebras.py +0 -242
- webscout/Provider/chatglm.py +0 -205
- webscout/Provider/cleeai.py +0 -213
- webscout/Provider/copilot.py +0 -428
- webscout/Provider/elmo.py +0 -234
- webscout/Provider/freeaichat.py +0 -271
- webscout/Provider/gaurish.py +0 -244
- webscout/Provider/geminiapi.py +0 -208
- webscout/Provider/geminiprorealtime.py +0 -160
- webscout/Provider/granite.py +0 -187
- webscout/Provider/hermes.py +0 -219
- webscout/Provider/julius.py +0 -223
- webscout/Provider/koala.py +0 -268
- webscout/Provider/labyrinth.py +0 -340
- webscout/Provider/learnfastai.py +0 -266
- webscout/Provider/lepton.py +0 -194
- webscout/Provider/llama3mitril.py +0 -180
- webscout/Provider/llamatutor.py +0 -192
- webscout/Provider/llmchat.py +0 -213
- webscout/Provider/llmchatco.py +0 -311
- webscout/Provider/meta.py +0 -794
- webscout/Provider/multichat.py +0 -325
- webscout/Provider/promptrefine.py +0 -193
- webscout/Provider/scira_chat.py +0 -277
- webscout/Provider/scnet.py +0 -187
- webscout/Provider/searchchat.py +0 -293
- webscout/Provider/sonus.py +0 -208
- webscout/Provider/talkai.py +0 -194
- webscout/Provider/toolbaz.py +0 -320
- webscout/Provider/turboseek.py +0 -219
- webscout/Provider/tutorai.py +0 -252
- webscout/Provider/typefully.py +0 -280
- webscout/Provider/typegpt.py +0 -232
- webscout/Provider/uncovr.py +0 -312
- webscout/Provider/x0gpt.py +0 -256
- webscout/Provider/yep.py +0 -376
- webscout/litagent/__init__.py +0 -29
- webscout/litagent/agent.py +0 -455
- webscout/litagent/constants.py +0 -60
- webscout/litprinter/__init__.py +0 -59
- webscout/scout/__init__.py +0 -8
- webscout/scout/core/__init__.py +0 -7
- webscout/scout/core/crawler.py +0 -140
- webscout/scout/core/scout.py +0 -568
- webscout/scout/core/search_result.py +0 -96
- webscout/scout/core/text_analyzer.py +0 -63
- webscout/scout/core/text_utils.py +0 -277
- webscout/scout/core/web_analyzer.py +0 -52
- webscout/scout/core.py +0 -881
- webscout/scout/element.py +0 -460
- webscout/scout/parsers/__init__.py +0 -69
- webscout/scout/parsers/html5lib_parser.py +0 -172
- webscout/scout/parsers/html_parser.py +0 -236
- webscout/scout/parsers/lxml_parser.py +0 -178
- webscout/scout/utils.py +0 -37
- webscout/swiftcli/__init__.py +0 -809
- webscout/zeroart/__init__.py +0 -55
- webscout/zeroart/base.py +0 -60
- webscout/zeroart/effects.py +0 -99
- webscout/zeroart/fonts.py +0 -816
- webscout-8.2.2.dist-info/RECORD +0 -309
- webscout-8.2.2.dist-info/entry_points.txt +0 -5
- webscout-8.2.2.dist-info/top_level.txt +0 -3
- webstoken/__init__.py +0 -30
- webstoken/classifier.py +0 -189
- webstoken/keywords.py +0 -216
- webstoken/language.py +0 -128
- webstoken/ner.py +0 -164
- webstoken/normalizer.py +0 -35
- webstoken/processor.py +0 -77
- webstoken/sentiment.py +0 -206
- webstoken/stemmer.py +0 -73
- webstoken/tagger.py +0 -60
- webstoken/tokenizer.py +0 -158
- {webscout-8.2.2.dist-info → webscout-8.2.7.dist-info/licenses}/LICENSE.md +0 -0
|
@@ -1,207 +0,0 @@
|
|
|
1
|
-
import uuid
|
|
2
|
-
import requests
|
|
3
|
-
import json
|
|
4
|
-
import os
|
|
5
|
-
from typing import Union, Any, Dict, List, Optional
|
|
6
|
-
|
|
7
|
-
from webscout.AIbase import ImageProvider
|
|
8
|
-
from webscout import exceptions
|
|
9
|
-
from webscout.litagent import agent # Import our fire user agent generator 🔥
|
|
10
|
-
from webscout.Litlogger import Logger # For that cyberpunk logging swag ⚡
|
|
11
|
-
|
|
12
|
-
# Initialize our fire logger 🚀
|
|
13
|
-
logger = Logger("TalkaiImager")
|
|
14
|
-
|
|
15
|
-
class TalkaiImager(ImageProvider):
|
|
16
|
-
"""
|
|
17
|
-
TalkAI Image Provider - Your go-to for fire AI art! 🎨
|
|
18
|
-
|
|
19
|
-
>>> # Generate some fire art! 🔥
|
|
20
|
-
>>> imager = TalkaiImager(logging=True)
|
|
21
|
-
>>> images = imager.generate("Epic dragon breathing fire", amount=2)
|
|
22
|
-
>>> paths = imager.save(images)
|
|
23
|
-
>>> print(paths)
|
|
24
|
-
['epic_dragon_0.png', 'epic_dragon_1.png']
|
|
25
|
-
|
|
26
|
-
>>> # Turn off logging for stealth mode 🥷
|
|
27
|
-
>>> quiet_imager = TalkaiImager(logging=False)
|
|
28
|
-
>>> images = quiet_imager.generate("Cyberpunk city at night")
|
|
29
|
-
>>> paths = quiet_imager.save(images)
|
|
30
|
-
"""
|
|
31
|
-
|
|
32
|
-
def __init__(self, timeout: int = 60, proxies: dict = {}, logging: bool = True):
|
|
33
|
-
"""Initialize your TalkAI provider with custom settings! ⚙️
|
|
34
|
-
|
|
35
|
-
Args:
|
|
36
|
-
timeout (int): Request timeout in seconds (default: 60)
|
|
37
|
-
proxies (dict): Proxy settings for requests (default: {})
|
|
38
|
-
logging (bool): Enable fire logging (default: True)
|
|
39
|
-
"""
|
|
40
|
-
self.api_endpoint = "https://talkai.info/chat/send/"
|
|
41
|
-
self.headers = {
|
|
42
|
-
'accept': 'application/json',
|
|
43
|
-
'accept-language': 'en-US,en;q=0.9',
|
|
44
|
-
'content-type': 'application/json',
|
|
45
|
-
'origin': 'https://talkai.info',
|
|
46
|
-
'referer': 'https://talkai.info/image/',
|
|
47
|
-
'user-agent': agent.random(), # Using our fire random agent! 🔥
|
|
48
|
-
}
|
|
49
|
-
self.session = requests.Session()
|
|
50
|
-
self.session.headers.update(self.headers)
|
|
51
|
-
self.session.proxies.update(proxies)
|
|
52
|
-
self.timeout = timeout
|
|
53
|
-
self.prompt: str = "AI-generated image - webscout"
|
|
54
|
-
self.image_extension: str = "png"
|
|
55
|
-
self.logging = logging
|
|
56
|
-
if self.logging:
|
|
57
|
-
logger.info("TalkaiImager initialized! Ready to create some fire art! 🚀")
|
|
58
|
-
|
|
59
|
-
def generate(
|
|
60
|
-
self, prompt: str, amount: int = 1,
|
|
61
|
-
max_retries: int = 3, retry_delay: int = 5
|
|
62
|
-
) -> List[str]:
|
|
63
|
-
"""Generate some fire images from your prompt! 🎨
|
|
64
|
-
|
|
65
|
-
Args:
|
|
66
|
-
prompt (str): Your creative prompt
|
|
67
|
-
amount (int): How many images to generate
|
|
68
|
-
max_retries (int): Max retry attempts if generation fails
|
|
69
|
-
retry_delay (int): Seconds to wait between retries
|
|
70
|
-
|
|
71
|
-
Returns:
|
|
72
|
-
List[str]: List of image URLs
|
|
73
|
-
"""
|
|
74
|
-
assert bool(prompt), "Prompt cannot be empty."
|
|
75
|
-
assert isinstance(amount, int) and amount > 0, "Amount must be a positive integer."
|
|
76
|
-
|
|
77
|
-
self.prompt = prompt
|
|
78
|
-
image_urls = []
|
|
79
|
-
|
|
80
|
-
if self.logging:
|
|
81
|
-
logger.info(f"Generating {amount} images... 🎨")
|
|
82
|
-
|
|
83
|
-
for _ in range(amount):
|
|
84
|
-
for attempt in range(max_retries):
|
|
85
|
-
try:
|
|
86
|
-
with self.session.post(
|
|
87
|
-
self.api_endpoint,
|
|
88
|
-
json=self._create_payload(prompt),
|
|
89
|
-
timeout=self.timeout
|
|
90
|
-
) as response:
|
|
91
|
-
response.raise_for_status()
|
|
92
|
-
data = response.json()
|
|
93
|
-
|
|
94
|
-
if 'data' in data and len(data['data']) > 0 and 'url' in data['data'][0]:
|
|
95
|
-
image_urls.append(data['data'][0]['url'])
|
|
96
|
-
if self.logging:
|
|
97
|
-
logger.success(f"Generated image {len(image_urls)}/{amount}! 🎨")
|
|
98
|
-
break
|
|
99
|
-
else:
|
|
100
|
-
raise exceptions.InvalidResponseError("No image URL found in API response.")
|
|
101
|
-
|
|
102
|
-
except requests.exceptions.RequestException as e:
|
|
103
|
-
if attempt == max_retries - 1:
|
|
104
|
-
if self.logging:
|
|
105
|
-
logger.error(f"Error making API request: {e} 😢")
|
|
106
|
-
raise exceptions.APIConnectionError(f"Error making API request: {e}") from e
|
|
107
|
-
else:
|
|
108
|
-
if self.logging:
|
|
109
|
-
logger.warning(f"Attempt {attempt + 1} failed. Retrying in {retry_delay} seconds... 🔄")
|
|
110
|
-
import time
|
|
111
|
-
time.sleep(retry_delay)
|
|
112
|
-
except json.JSONDecodeError as e:
|
|
113
|
-
if self.logging:
|
|
114
|
-
logger.error(f"Invalid JSON response: {e} 😢")
|
|
115
|
-
raise exceptions.InvalidResponseError(f"Invalid JSON response: {e}") from e
|
|
116
|
-
except Exception as e:
|
|
117
|
-
if self.logging:
|
|
118
|
-
logger.error(f"An unexpected error occurred: {e} 😢")
|
|
119
|
-
raise exceptions.FailedToGenerateResponseError(f"An unexpected error occurred: {e}") from e
|
|
120
|
-
|
|
121
|
-
if self.logging:
|
|
122
|
-
logger.success("All images generated successfully! 🎉")
|
|
123
|
-
return image_urls
|
|
124
|
-
|
|
125
|
-
def _create_payload(self, prompt: str) -> Dict[str, Any]:
|
|
126
|
-
"""Create the API request payload 📦
|
|
127
|
-
|
|
128
|
-
Args:
|
|
129
|
-
prompt (str): The image generation prompt
|
|
130
|
-
|
|
131
|
-
Returns:
|
|
132
|
-
Dict[str, Any]: API request payload
|
|
133
|
-
"""
|
|
134
|
-
return {
|
|
135
|
-
"type": "image",
|
|
136
|
-
"messagesHistory": [
|
|
137
|
-
{
|
|
138
|
-
"id": str(uuid.uuid4()),
|
|
139
|
-
"from": "you",
|
|
140
|
-
"content": prompt
|
|
141
|
-
}
|
|
142
|
-
],
|
|
143
|
-
"settings": {
|
|
144
|
-
"model": "gpt-4o-mini" # Or another suitable model if available
|
|
145
|
-
}
|
|
146
|
-
}
|
|
147
|
-
|
|
148
|
-
def save(
|
|
149
|
-
self,
|
|
150
|
-
response: List[str],
|
|
151
|
-
name: str = None,
|
|
152
|
-
dir: str = os.getcwd(),
|
|
153
|
-
filenames_prefix: str = "",
|
|
154
|
-
) -> List[str]:
|
|
155
|
-
"""Save your fire images! 💾
|
|
156
|
-
|
|
157
|
-
Args:
|
|
158
|
-
response (List[str]): List of image URLs
|
|
159
|
-
name (str, optional): Base name for saved files
|
|
160
|
-
dir (str, optional): Where to save the images
|
|
161
|
-
filenames_prefix (str, optional): Prefix for filenames
|
|
162
|
-
|
|
163
|
-
Returns:
|
|
164
|
-
List[str]: List of saved filenames
|
|
165
|
-
"""
|
|
166
|
-
assert isinstance(response, list), f"Response should be a list, not {type(response)}"
|
|
167
|
-
name = self.prompt if name is None else name
|
|
168
|
-
|
|
169
|
-
if not os.path.exists(dir):
|
|
170
|
-
os.makedirs(dir)
|
|
171
|
-
if self.logging:
|
|
172
|
-
logger.info(f"Created directory: {dir} 📁")
|
|
173
|
-
|
|
174
|
-
if self.logging:
|
|
175
|
-
logger.info(f"Saving {len(response)} images... 💾")
|
|
176
|
-
|
|
177
|
-
filenames = []
|
|
178
|
-
for i, url in enumerate(response):
|
|
179
|
-
try:
|
|
180
|
-
with self.session.get(url, stream=True, timeout=self.timeout) as r:
|
|
181
|
-
r.raise_for_status()
|
|
182
|
-
filename = f"{filenames_prefix}{name}_{i}.{self.image_extension}"
|
|
183
|
-
filepath = os.path.join(dir, filename)
|
|
184
|
-
with open(filepath, 'wb') as f:
|
|
185
|
-
for chunk in r.iter_content(chunk_size=8192):
|
|
186
|
-
f.write(chunk)
|
|
187
|
-
filenames.append(filename)
|
|
188
|
-
if self.logging:
|
|
189
|
-
logger.success(f"Saved image to: {filepath} 💾")
|
|
190
|
-
except requests.exceptions.RequestException as e:
|
|
191
|
-
if self.logging:
|
|
192
|
-
logger.error(f"Error downloading image from {url}: {e} 😢")
|
|
193
|
-
filenames.append(None) # Indicate failure to download
|
|
194
|
-
|
|
195
|
-
if self.logging:
|
|
196
|
-
logger.success(f"All images saved successfully! Check {dir} 🎉")
|
|
197
|
-
return filenames
|
|
198
|
-
|
|
199
|
-
|
|
200
|
-
if __name__ == "__main__":
|
|
201
|
-
bot = TalkaiImager()
|
|
202
|
-
try:
|
|
203
|
-
resp = bot.generate("A shiny red sports car speeding down a scenic mountain road", 1)
|
|
204
|
-
print(bot.save(resp))
|
|
205
|
-
except Exception as e:
|
|
206
|
-
if bot.logging:
|
|
207
|
-
logger.error(f"An error occurred: {e} 😢")
|
|
@@ -1,156 +0,0 @@
|
|
|
1
|
-
import time
|
|
2
|
-
import requests
|
|
3
|
-
import pathlib
|
|
4
|
-
import base64
|
|
5
|
-
import tempfile
|
|
6
|
-
from io import BytesIO
|
|
7
|
-
from webscout import exceptions
|
|
8
|
-
from webscout.AIbase import TTSProvider
|
|
9
|
-
from concurrent.futures import ThreadPoolExecutor, as_completed
|
|
10
|
-
from webscout.litagent import LitAgent
|
|
11
|
-
from . import utils
|
|
12
|
-
|
|
13
|
-
class DeepgramTTS(TTSProvider):
|
|
14
|
-
"""
|
|
15
|
-
Text-to-speech provider using the DeepgramTTS API.
|
|
16
|
-
"""
|
|
17
|
-
# Request headers
|
|
18
|
-
headers: dict[str, str] = {
|
|
19
|
-
"User-Agent": LitAgent().random()
|
|
20
|
-
}
|
|
21
|
-
all_voices: dict[str, str] = {
|
|
22
|
-
"Asteria": "aura-asteria-en", "Arcas": "aura-arcas-en", "Luna": "aura-luna-en",
|
|
23
|
-
"Zeus": "aura-zeus-en", "Orpheus": "aura-orpheus-en", "Angus": "aura-angus-en",
|
|
24
|
-
"Athena": "aura-athena-en", "Helios": "aura-helios-en", "Hera": "aura-hera-en",
|
|
25
|
-
"Orion": "aura-orion-en", "Perseus": "aura-perseus-en", "Stella": "aura-stella-en"
|
|
26
|
-
}
|
|
27
|
-
|
|
28
|
-
def __init__(self, timeout: int = 20, proxies: dict = None):
|
|
29
|
-
"""Initializes the DeepgramTTS TTS client."""
|
|
30
|
-
self.session = requests.Session()
|
|
31
|
-
self.session.headers.update(self.headers)
|
|
32
|
-
if proxies:
|
|
33
|
-
self.session.proxies.update(proxies)
|
|
34
|
-
self.timeout = timeout
|
|
35
|
-
self.temp_dir = tempfile.mkdtemp(prefix="webscout_tts_")
|
|
36
|
-
|
|
37
|
-
def tts(self, text: str, voice: str = "Brian", verbose: bool = True) -> str:
|
|
38
|
-
"""
|
|
39
|
-
Converts text to speech using the DeepgramTTS API and saves it to a file.
|
|
40
|
-
|
|
41
|
-
Args:
|
|
42
|
-
text (str): The text to convert to speech
|
|
43
|
-
voice (str): The voice to use for TTS (default: "Brian")
|
|
44
|
-
verbose (bool): Whether to print progress messages (default: True)
|
|
45
|
-
|
|
46
|
-
Returns:
|
|
47
|
-
str: Path to the generated audio file
|
|
48
|
-
|
|
49
|
-
Raises:
|
|
50
|
-
AssertionError: If the specified voice is not available
|
|
51
|
-
requests.RequestException: If there's an error communicating with the API
|
|
52
|
-
RuntimeError: If there's an error processing the audio
|
|
53
|
-
"""
|
|
54
|
-
assert (
|
|
55
|
-
voice in self.all_voices
|
|
56
|
-
), f"Voice '{voice}' not one of [{', '.join(self.all_voices.keys())}]"
|
|
57
|
-
|
|
58
|
-
url = "https://deepgram.com/api/ttsAudioGeneration"
|
|
59
|
-
filename = pathlib.Path(tempfile.mktemp(suffix=".mp3", dir=self.temp_dir))
|
|
60
|
-
|
|
61
|
-
# Split text into sentences using the utils module
|
|
62
|
-
sentences = utils.split_sentences(text)
|
|
63
|
-
if verbose:
|
|
64
|
-
for index, sen in enumerate(sentences):
|
|
65
|
-
print(f"[debug] Sentence {index}: {sen}")
|
|
66
|
-
|
|
67
|
-
def generate_audio_for_chunk(part_text: str, part_number: int):
|
|
68
|
-
"""
|
|
69
|
-
Generate audio for a single chunk of text.
|
|
70
|
-
|
|
71
|
-
Args:
|
|
72
|
-
part_text (str): The text chunk to convert
|
|
73
|
-
part_number (int): The chunk number for ordering
|
|
74
|
-
|
|
75
|
-
Returns:
|
|
76
|
-
tuple: (part_number, audio_data)
|
|
77
|
-
|
|
78
|
-
Raises:
|
|
79
|
-
requests.RequestException: If there's an API error
|
|
80
|
-
"""
|
|
81
|
-
max_retries = 3
|
|
82
|
-
retry_count = 0
|
|
83
|
-
|
|
84
|
-
while retry_count < max_retries:
|
|
85
|
-
try:
|
|
86
|
-
payload = {"text": part_text, "model": self.all_voices[voice]}
|
|
87
|
-
response = self.session.post(
|
|
88
|
-
url=url,
|
|
89
|
-
headers=self.headers,
|
|
90
|
-
json=payload,
|
|
91
|
-
stream=True,
|
|
92
|
-
timeout=self.timeout
|
|
93
|
-
)
|
|
94
|
-
response.raise_for_status()
|
|
95
|
-
|
|
96
|
-
response_data = response.json().get('data')
|
|
97
|
-
if response_data:
|
|
98
|
-
audio_data = base64.b64decode(response_data)
|
|
99
|
-
if verbose:
|
|
100
|
-
print(f"[debug] Chunk {part_number} processed successfully")
|
|
101
|
-
return part_number, audio_data
|
|
102
|
-
|
|
103
|
-
if verbose:
|
|
104
|
-
print(f"[debug] No data received for chunk {part_number}. Attempt {retry_count + 1}/{max_retries}")
|
|
105
|
-
|
|
106
|
-
except requests.RequestException as e:
|
|
107
|
-
if verbose:
|
|
108
|
-
print(f"[debug] Error processing chunk {part_number}: {str(e)}. Attempt {retry_count + 1}/{max_retries}")
|
|
109
|
-
if retry_count == max_retries - 1:
|
|
110
|
-
raise
|
|
111
|
-
|
|
112
|
-
retry_count += 1
|
|
113
|
-
time.sleep(1)
|
|
114
|
-
|
|
115
|
-
raise RuntimeError(f"Failed to generate audio for chunk {part_number} after {max_retries} attempts")
|
|
116
|
-
|
|
117
|
-
try:
|
|
118
|
-
# Using ThreadPoolExecutor to handle requests concurrently
|
|
119
|
-
with ThreadPoolExecutor() as executor:
|
|
120
|
-
futures = {
|
|
121
|
-
executor.submit(generate_audio_for_chunk, sentence.strip(), chunk_num): chunk_num
|
|
122
|
-
for chunk_num, sentence in enumerate(sentences, start=1)
|
|
123
|
-
}
|
|
124
|
-
|
|
125
|
-
# Dictionary to store results with order preserved
|
|
126
|
-
audio_chunks = {}
|
|
127
|
-
|
|
128
|
-
for future in as_completed(futures):
|
|
129
|
-
chunk_num = futures[future]
|
|
130
|
-
try:
|
|
131
|
-
part_number, audio_data = future.result()
|
|
132
|
-
audio_chunks[part_number] = audio_data
|
|
133
|
-
except Exception as e:
|
|
134
|
-
raise RuntimeError(f"Failed to generate audio for chunk {chunk_num}: {str(e)}")
|
|
135
|
-
|
|
136
|
-
# Combine all audio chunks in order
|
|
137
|
-
with open(filename, 'wb') as f:
|
|
138
|
-
for chunk_num in sorted(audio_chunks.keys()):
|
|
139
|
-
f.write(audio_chunks[chunk_num])
|
|
140
|
-
|
|
141
|
-
if verbose:
|
|
142
|
-
print(f"[debug] Audio saved to {filename}")
|
|
143
|
-
return str(filename)
|
|
144
|
-
|
|
145
|
-
except Exception as e:
|
|
146
|
-
print(f"[debug] Failed to generate audio: {str(e)}") if verbose else None
|
|
147
|
-
raise RuntimeError(f"Failed to generate audio: {str(e)}")
|
|
148
|
-
|
|
149
|
-
# Example usage
|
|
150
|
-
if __name__ == "__main__":
|
|
151
|
-
deepgram = DeepgramTTS()
|
|
152
|
-
text = "This is a test of the DeepgramTTS text-to-speech API. It supports multiple sentences. Let's see how it works!"
|
|
153
|
-
|
|
154
|
-
print("[debug] Generating audio...")
|
|
155
|
-
audio_file = deepgram.tts(text, voice="Asteria")
|
|
156
|
-
print(f"Audio saved to: {audio_file}")
|
|
@@ -1,111 +0,0 @@
|
|
|
1
|
-
import time
|
|
2
|
-
import requests
|
|
3
|
-
import pathlib
|
|
4
|
-
import tempfile
|
|
5
|
-
from io import BytesIO
|
|
6
|
-
from webscout import exceptions
|
|
7
|
-
from webscout.AIbase import TTSProvider
|
|
8
|
-
from webscout.litagent import LitAgent
|
|
9
|
-
from concurrent.futures import ThreadPoolExecutor, as_completed
|
|
10
|
-
from . import utils
|
|
11
|
-
|
|
12
|
-
class ElevenlabsTTS(TTSProvider):
|
|
13
|
-
"""
|
|
14
|
-
Text-to-speech provider using the ElevenlabsTTS API.
|
|
15
|
-
"""
|
|
16
|
-
# Request headers
|
|
17
|
-
headers: dict[str, str] = {
|
|
18
|
-
"User-Agent": LitAgent().random()
|
|
19
|
-
}
|
|
20
|
-
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"}
|
|
21
|
-
|
|
22
|
-
def __init__(self, timeout: int = 20, proxies: dict = None):
|
|
23
|
-
"""Initializes the ElevenlabsTTS TTS client."""
|
|
24
|
-
self.session = requests.Session()
|
|
25
|
-
self.session.headers.update(self.headers)
|
|
26
|
-
if proxies:
|
|
27
|
-
self.session.proxies.update(proxies)
|
|
28
|
-
self.timeout = timeout
|
|
29
|
-
self.params = {'allow_unauthenticated': '1'}
|
|
30
|
-
self.temp_dir = tempfile.mkdtemp(prefix="webscout_tts_")
|
|
31
|
-
|
|
32
|
-
def tts(self, text: str, voice: str = "Brian", verbose:bool = True) -> str:
|
|
33
|
-
"""
|
|
34
|
-
Converts text to speech using the ElevenlabsTTS API and saves it to a file.
|
|
35
|
-
"""
|
|
36
|
-
assert (
|
|
37
|
-
voice in self.all_voices
|
|
38
|
-
), f"Voice '{voice}' not one of [{', '.join(self.all_voices.keys())}]"
|
|
39
|
-
|
|
40
|
-
filename = pathlib.Path(tempfile.mktemp(suffix=".mp3", dir=self.temp_dir))
|
|
41
|
-
|
|
42
|
-
# Split text into sentences
|
|
43
|
-
sentences = utils.split_sentences(text)
|
|
44
|
-
|
|
45
|
-
# Function to request audio for each chunk
|
|
46
|
-
def generate_audio_for_chunk(part_text: str, part_number: int):
|
|
47
|
-
while True:
|
|
48
|
-
try:
|
|
49
|
-
json_data = {'text': part_text, 'model_id': 'eleven_multilingual_v2'}
|
|
50
|
-
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)
|
|
51
|
-
response.raise_for_status()
|
|
52
|
-
|
|
53
|
-
# Check if the request was successful
|
|
54
|
-
if response.ok and response.status_code == 200:
|
|
55
|
-
if verbose:
|
|
56
|
-
print(f"[debug] Chunk {part_number} processed successfully")
|
|
57
|
-
return part_number, response.content
|
|
58
|
-
else:
|
|
59
|
-
if verbose:
|
|
60
|
-
print(f"[debug] No data received for chunk {part_number}. Retrying...")
|
|
61
|
-
except requests.RequestException as e:
|
|
62
|
-
if verbose:
|
|
63
|
-
print(f"[debug] Error for chunk {part_number}: {e}. Retrying...")
|
|
64
|
-
time.sleep(1)
|
|
65
|
-
try:
|
|
66
|
-
# Using ThreadPoolExecutor to handle requests concurrently
|
|
67
|
-
with ThreadPoolExecutor() as executor:
|
|
68
|
-
futures = {executor.submit(generate_audio_for_chunk, sentence.strip(), chunk_num): chunk_num
|
|
69
|
-
for chunk_num, sentence in enumerate(sentences, start=1)}
|
|
70
|
-
|
|
71
|
-
# Dictionary to store results with order preserved
|
|
72
|
-
audio_chunks = {}
|
|
73
|
-
|
|
74
|
-
for future in as_completed(futures):
|
|
75
|
-
chunk_num = futures[future]
|
|
76
|
-
try:
|
|
77
|
-
part_number, audio_data = future.result()
|
|
78
|
-
audio_chunks[part_number] = audio_data # Store the audio data in correct sequence
|
|
79
|
-
except Exception as e:
|
|
80
|
-
if verbose:
|
|
81
|
-
print(f"[debug] Failed to generate audio for chunk {chunk_num}: {e}")
|
|
82
|
-
|
|
83
|
-
# Combine audio chunks in the correct sequence
|
|
84
|
-
combined_audio = BytesIO()
|
|
85
|
-
for part_number in sorted(audio_chunks.keys()):
|
|
86
|
-
combined_audio.write(audio_chunks[part_number])
|
|
87
|
-
if verbose:
|
|
88
|
-
print(f"[debug] Added chunk {part_number} to the combined file.")
|
|
89
|
-
|
|
90
|
-
# Save the combined audio data to a single file
|
|
91
|
-
with open(filename, 'wb') as f:
|
|
92
|
-
f.write(combined_audio.getvalue())
|
|
93
|
-
if verbose:
|
|
94
|
-
print(f"[debug] Final Audio Saved as {filename}")
|
|
95
|
-
return filename.as_posix()
|
|
96
|
-
|
|
97
|
-
except requests.exceptions.RequestException as e:
|
|
98
|
-
if verbose:
|
|
99
|
-
print(f"[debug] Failed to perform the operation: {e}")
|
|
100
|
-
raise exceptions.FailedToGenerateResponseError(
|
|
101
|
-
f"Failed to perform the operation: {e}"
|
|
102
|
-
)
|
|
103
|
-
|
|
104
|
-
# Example usage
|
|
105
|
-
if __name__ == "__main__":
|
|
106
|
-
elevenlabs = ElevenlabsTTS()
|
|
107
|
-
text = "This is a test of the ElevenlabsTTS text-to-speech API. It supports multiple sentences and advanced logging."
|
|
108
|
-
|
|
109
|
-
print("[debug] Generating audio...")
|
|
110
|
-
audio_file = elevenlabs.tts(text, voice="Brian")
|
|
111
|
-
print(f"Audio saved to: {audio_file}")
|
|
@@ -1,127 +0,0 @@
|
|
|
1
|
-
import time
|
|
2
|
-
import requests
|
|
3
|
-
import pathlib
|
|
4
|
-
import base64
|
|
5
|
-
from io import BytesIO
|
|
6
|
-
from webscout import exceptions
|
|
7
|
-
from webscout.AIbase import TTSProvider
|
|
8
|
-
from webscout.litagent import LitAgent
|
|
9
|
-
from concurrent.futures import ThreadPoolExecutor, as_completed
|
|
10
|
-
from . import utils
|
|
11
|
-
|
|
12
|
-
class GesseritTTS(TTSProvider):
|
|
13
|
-
"""Text-to-speech provider using the GesseritTTS API."""
|
|
14
|
-
# Request headers
|
|
15
|
-
headers: dict[str, str] = {
|
|
16
|
-
"User-Agent": LitAgent().random()
|
|
17
|
-
}
|
|
18
|
-
cache_dir = pathlib.Path("./audio_cache")
|
|
19
|
-
all_voices: dict[str, str] = {
|
|
20
|
-
"Emma": "en_us_001", # Female Voice
|
|
21
|
-
"Liam": "en_us_006", # Male Voice
|
|
22
|
-
"Noah": "en_us_007", # Male Voice
|
|
23
|
-
"Oliver": "en_us_009", # Male Voice
|
|
24
|
-
"Elijah": "en_us_010", # Male Voice
|
|
25
|
-
"James": "en_male_narration", # Male Voice
|
|
26
|
-
"Charlie": "en_male_funny", # Male Voice
|
|
27
|
-
"Sophia": "en_female_emotional", # Female Voice
|
|
28
|
-
"Cody": "en_male_cody", # Male Voice
|
|
29
|
-
}
|
|
30
|
-
|
|
31
|
-
def __init__(self, timeout: int = 20, proxies: dict = None):
|
|
32
|
-
"""Initializes the GesseritTTS TTS client."""
|
|
33
|
-
self.session = requests.Session()
|
|
34
|
-
self.session.headers.update(self.headers)
|
|
35
|
-
if proxies:
|
|
36
|
-
self.session.proxies.update(proxies)
|
|
37
|
-
self.timeout = timeout
|
|
38
|
-
|
|
39
|
-
def tts(self, text: str, voice: str = "Oliver", verbose:bool = True) -> str:
|
|
40
|
-
"""Converts text to speech using the GesseritTTS API and saves it to a file."""
|
|
41
|
-
assert (
|
|
42
|
-
voice in self.all_voices
|
|
43
|
-
), f"Voice '{voice}' not one of [{', '.join(self.all_voices.keys())}]"
|
|
44
|
-
|
|
45
|
-
filename = self.cache_dir / f"{int(time.time())}.mp3"
|
|
46
|
-
|
|
47
|
-
voice_id = self.all_voices[voice]
|
|
48
|
-
|
|
49
|
-
# Split text into sentences
|
|
50
|
-
sentences = utils.split_sentences(text)
|
|
51
|
-
|
|
52
|
-
# Function to request audio for each chunk
|
|
53
|
-
def generate_audio_for_chunk(part_text: str, part_number: int):
|
|
54
|
-
while True:
|
|
55
|
-
try:
|
|
56
|
-
payload = {
|
|
57
|
-
"text": part_text,
|
|
58
|
-
"voice": voice_id
|
|
59
|
-
}
|
|
60
|
-
response = self.session.post('https://gesserit.co/api/tiktok-tts', headers=self.headers, json=payload, timeout=self.timeout)
|
|
61
|
-
response.raise_for_status()
|
|
62
|
-
|
|
63
|
-
# Create the audio_cache directory if it doesn't exist
|
|
64
|
-
self.cache_dir.mkdir(parents=True, exist_ok=True)
|
|
65
|
-
|
|
66
|
-
# Check if the request was successful
|
|
67
|
-
if response.ok and response.status_code == 200:
|
|
68
|
-
data = response.json()
|
|
69
|
-
audio_base64 = data["audioUrl"].split(",")[1]
|
|
70
|
-
audio_data = base64.b64decode(audio_base64)
|
|
71
|
-
if verbose:
|
|
72
|
-
print(f"[debug] Chunk {part_number} processed successfully")
|
|
73
|
-
return part_number, audio_data
|
|
74
|
-
else:
|
|
75
|
-
if verbose:
|
|
76
|
-
print(f"[debug] No data received for chunk {part_number}. Retrying...")
|
|
77
|
-
except requests.RequestException as e:
|
|
78
|
-
if verbose:
|
|
79
|
-
print(f"[debug] Error for chunk {part_number}: {e}. Retrying...")
|
|
80
|
-
time.sleep(1)
|
|
81
|
-
try:
|
|
82
|
-
# Using ThreadPoolExecutor to handle requests concurrently
|
|
83
|
-
with ThreadPoolExecutor() as executor:
|
|
84
|
-
futures = {executor.submit(generate_audio_for_chunk, sentence.strip(), chunk_num): chunk_num
|
|
85
|
-
for chunk_num, sentence in enumerate(sentences, start=1)}
|
|
86
|
-
|
|
87
|
-
# Dictionary to store results with order preserved
|
|
88
|
-
audio_chunks = {}
|
|
89
|
-
|
|
90
|
-
for future in as_completed(futures):
|
|
91
|
-
chunk_num = futures[future]
|
|
92
|
-
try:
|
|
93
|
-
part_number, audio_data = future.result()
|
|
94
|
-
audio_chunks[part_number] = audio_data # Store the audio data in correct sequence
|
|
95
|
-
except Exception as e:
|
|
96
|
-
if verbose:
|
|
97
|
-
print(f"[debug] Failed to generate audio for chunk {chunk_num}: {e}")
|
|
98
|
-
|
|
99
|
-
# Combine audio chunks in the correct sequence
|
|
100
|
-
combined_audio = BytesIO()
|
|
101
|
-
for part_number in sorted(audio_chunks.keys()):
|
|
102
|
-
combined_audio.write(audio_chunks[part_number])
|
|
103
|
-
if verbose:
|
|
104
|
-
print(f"[debug] Added chunk {part_number} to the combined file.")
|
|
105
|
-
|
|
106
|
-
# Save the combined audio data to a single file
|
|
107
|
-
with open(filename, 'wb') as f:
|
|
108
|
-
f.write(combined_audio.getvalue())
|
|
109
|
-
if verbose:
|
|
110
|
-
print(f"[debug] Final Audio Saved as {filename}")
|
|
111
|
-
return filename.as_posix()
|
|
112
|
-
|
|
113
|
-
except requests.exceptions.RequestException as e:
|
|
114
|
-
if verbose:
|
|
115
|
-
print(f"[debug] Failed to perform the operation: {e}")
|
|
116
|
-
raise exceptions.FailedToGenerateResponseError(
|
|
117
|
-
f"Failed to perform the operation: {e}"
|
|
118
|
-
)
|
|
119
|
-
|
|
120
|
-
# Example usage
|
|
121
|
-
if __name__ == "__main__":
|
|
122
|
-
gesserit = GesseritTTS()
|
|
123
|
-
text = "This is a test of the GesseritTTS text-to-speech API. It supports multiple sentences and advanced logging."
|
|
124
|
-
|
|
125
|
-
print("[debug] Generating audio...")
|
|
126
|
-
audio_file = gesserit.tts(text, voice="Oliver")
|
|
127
|
-
print(f"Audio saved to: {audio_file}")
|