generate_img.py

import json
import requests
import os

# print(result['replicate']['items'])

def generate_image(API_KEY, text, image_count=1):
    headers = {"Authorization": f"Bearer {API_KEY}"}

    url = "https://api.edenai.run/v2/image/generation"
    payload = {
        "providers": "replicate",
        "text": text,
        "response_as_dict": True,
        "attributes_as_list": False,
        "resolution": "1024x1024",
        "num_images": image_count,
        "fallback_providers": "openai",
    }



    response = requests.post(url, json=payload, headers=headers)
    result = json.loads(response.text)
    # print(result)
    return result['replicate']['items'][0]['image_resource_url']


# image_count = random.randint(1, 1000)

# image_url = result['replicate']['items'][0]['image_resource_url']

    
# download_image(image_url, "Images/", f"image{image_count}")