Spaces:
Sleeping
Sleeping
File size: 1,918 Bytes
b713a83 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 | import logging
import os
from openai import OpenAI
from dotenv import load_dotenv
# Load environment variables
load_dotenv()
# Configure logging
logging.basicConfig(level=logging.INFO, format='%(levelname)s: %(message)s')
class VoiceToImageAgent:
def __init__(self):
self.client = OpenAI()
def transcribe(self, audio_path):
logging.info("Audio received")
logging.info(f"Transcribing audio from {audio_path}")
with open(audio_path, "rb") as f:
text = self.client.audio.transcriptions.create(
file=f,
model="whisper-1"
)
logging.info(f"Transcription: \"{text.text}\"")
return text.text
def text_to_prompt(self, transcript):
logging.info("Generating image prompt")
response = self.client.chat.completions.create(
model="gpt-4o-mini",
messages=[
{"role": "system", "content": "Convert user intent into a detailed image description. Keep it descriptive and visual."},
{"role": "user", "content": transcript}
]
)
prompt = response.choices[0].message.content
logging.info(f"Image prompt generated: \"{prompt}\"")
return prompt
def generate_image(self, prompt):
logging.info("Generating image")
try:
# Using dall-e-3 as it's the current standard.
# Note: dall-e-3 requires 1024x1024.
result = self.client.images.generate(
model="dall-e-3",
prompt=prompt,
size="1024x1024",
quality="standard",
n=1,
)
image_url = result.data[0].url
logging.info("Image generation completed")
return image_url
except Exception as e:
logging.error(f"Error generating image: {e}")
raise e
|