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