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:
Moishe Lettvin
2024-01-12 10:47:03 -05:00
parent ec1f2362c5
commit ad427bea3a
4 changed files with 30 additions and 18 deletions

View File

@@ -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")

View File

@@ -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())