wehub-resource-sync
3fbbd7970c
Code Quality / Python Lint & Format (push) Has been cancelled
Code Quality / Python Tests (push) Has been cancelled
Code Quality / JavaScript/TypeScript Lint (advisory) (push) Has been cancelled
Security Scan / CodeQL Analysis (python) (push) Has been cancelled
Security Scan / Dependency Review (push) Has been cancelled
Security Scan / CodeQL Analysis (javascript-typescript) (push) Has been cancelled
59 lines
1.7 KiB
Python
59 lines
1.7 KiB
Python
from openai import AzureOpenAI, BadRequestError
|
|
import os
|
|
import requests
|
|
from PIL import Image
|
|
import dotenv
|
|
import json
|
|
|
|
# import dotenv
|
|
dotenv.load_dotenv()
|
|
|
|
|
|
|
|
# Assign the API version (check the Microsoft Foundry docs for the current API version required by your model)
|
|
client = AzureOpenAI(
|
|
api_key=os.environ['AZURE_OPENAI_API_KEY'], # this is also the default, it can be omitted
|
|
api_version = "2024-10-21",
|
|
azure_endpoint=os.environ['AZURE_OPENAI_ENDPOINT']
|
|
)
|
|
|
|
model = os.environ['AZURE_OPENAI_DEPLOYMENT']
|
|
|
|
|
|
try:
|
|
# Create an image by using the image generation API
|
|
|
|
result = client.images.generate(
|
|
model=model,
|
|
prompt='Bunny on horse, holding a lollipop, on a foggy meadow where it grows daffodils. It says "hello"', # Enter your prompt text here
|
|
size='1024x1024',
|
|
n=1
|
|
)
|
|
|
|
generation_response = json.loads(result.model_dump_json())
|
|
# Set the directory for the stored image
|
|
image_dir = os.path.join(os.curdir, 'images')
|
|
|
|
# If the directory doesn't exist, create it
|
|
if not os.path.isdir(image_dir):
|
|
os.mkdir(image_dir)
|
|
|
|
# Initialize the image path (note the filetype should be png)
|
|
image_path = os.path.join(image_dir, 'generated-image.png')
|
|
|
|
# Retrieve the generated image
|
|
image_url = generation_response["data"][0]["url"] # extract image URL from response
|
|
generated_image = requests.get(image_url).content # download the image
|
|
with open(image_path, "wb") as image_file:
|
|
image_file.write(generated_image)
|
|
|
|
# Display the image in the default image viewer
|
|
image = Image.open(image_path)
|
|
image.show()
|
|
|
|
# catch exceptions
|
|
#except BadRequestError as err:
|
|
# print(err)
|
|
|
|
finally:
|
|
print("completed!") |