| from diffusers import pipeline |
| import uuid |
| import os |
|
|
| class TextToImageTool: |
| def __init__(self, model_name="stabilityai/stable-diffusion-2"): |
| self.pipe = pipeline("text-to-image", model=model_name) |
| self.output_dir = "generated_images" |
| os.makedirs(self.output_dir, exist_ok=True) |
|
|
| def __call__(self, prompt: str) -> str: |
| """Generates image from text and returns local file path""" |
| result = self.pipe(prompt) |
| image_path = os.path.join(self.output_dir, f"{uuid.uuid4()}.png") |
| result.images[0].save(image_path) |
| return image_path |
