microsoft--generative-ai-for-beginners
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 (javascript-typescript) (push) Has been cancelled
Security Scan / CodeQL Analysis (python) (push) Has been cancelled
Security Scan / Dependency Review (push) Has been cancelled
55 行
1.5 KiB
Python
55 行
1.5 KiB
Python
from openai import AzureOpenAI, BadRequestError
|
|
import os
|
|
import requests
|
|
from PIL import Image
|
|
import dotenv
|
|
import json
|
|
|
|
# import dotenv
|
|
dotenv.load_dotenv()
|
|
|
|
# Get endpoint and key from environment variables
|
|
client = AzureOpenAI(
|
|
api_key=os.environ['AZURE_OPENAI_API_KEY'], # this is also the default, it can be omitted
|
|
api_version = "2024-10-21", # check the Microsoft Foundry docs for the current API version required by your model
|
|
azure_endpoint=os.environ['AZURE_OPENAI_ENDPOINT']
|
|
)
|
|
|
|
model = os.environ['AZURE_OPENAI_DEPLOYMENT']
|
|
|
|
image_dir = os.path.join(os.curdir, 'images')
|
|
|
|
# Initialize the image path (note the filetype should be png)
|
|
image_path = os.path.join(image_dir, 'generated-image.png')
|
|
print(image_path)
|
|
image = Image.open(image_path)
|
|
image.show()
|
|
|
|
# ---creating variation below---
|
|
try:
|
|
print("LOG creating variation")
|
|
result = client.images.create_variation(
|
|
image=open(image_path, "rb"),
|
|
n=1,
|
|
size="1024x1024"
|
|
)
|
|
|
|
response = json.loads(result.model_dump_json())
|
|
|
|
image_path = os.path.join(image_dir, 'generated_variation.png')
|
|
|
|
image_url = response['data'][0]['url']
|
|
|
|
print("LOG downloading image")
|
|
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()
|
|
#except BadRequestError as err:
|
|
# print(err)
|
|
|
|
finally:
|
|
print("completed!") |