| from PIL import Image |
| import io |
| import os |
| import requests |
| from crewai_tools import tool |
|
|
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
|
|
| |
| |
| |
| |
|
|
| @tool("Image Generate Tool") |
| def image_generate(context: str): |
| """This is Image generate tool""" |
| API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0" |
| hugginface_api_key = os.getenv("HUGGINGFACE_API_KEY") |
| headers = {"Authorization": f"Bearer {hugginface_api_key}"} |
|
|
| def query(payload): |
| response = requests.post(API_URL, headers=headers, json=payload) |
| return response.content |
| image_bytes = query({ |
| "inputs": context, |
| }) |
| name = context.split(" ")[0] |
| image = Image.open(io.BytesIO(image_bytes)).resize((1024, 1024)) |
| image.save(f"outputs/generate_{name}.jpg") |
| return image |
|
|
