Update 05- sample, sticking with generators here because they give us a lot more control over the order that things get queued.
This commit is contained in:
@@ -9,7 +9,7 @@ from dailyai.services.ai_services import TTSService
|
||||
|
||||
|
||||
class ElevenLabsTTSService(TTSService):
|
||||
def __init__(self, input_queue, output_queue, api_key=None, voice_id=None):
|
||||
def __init__(self, input_queue=None, output_queue=None, api_key=None, voice_id=None):
|
||||
super().__init__(input_queue, output_queue)
|
||||
|
||||
self.api_key = api_key or os.getenv("ELEVENLABS_API_KEY")
|
||||
|
||||
@@ -66,9 +66,14 @@ class OpenAIImageGenService(ImageGenService):
|
||||
size=size
|
||||
)
|
||||
image_url = image.data[0].url
|
||||
response = requests.get(image_url)
|
||||
if not image_url:
|
||||
raise Exception("No image provided in response", image)
|
||||
|
||||
dalle_stream = io.BytesIO(response.content)
|
||||
dalle_im = Image.open(dalle_stream)
|
||||
# Load the image from the url
|
||||
async with aiohttp.ClientSession() as session:
|
||||
async with session.get(image_url) as response:
|
||||
image_stream = io.BytesIO(await response.content.read())
|
||||
image = Image.open(image_stream)
|
||||
return (image_url, image.tobytes())
|
||||
|
||||
return (image_url, dalle_im.tobytes())
|
||||
|
||||
Reference in New Issue
Block a user