wehub-resource-sync
97e91a83f3
Ruff / Ruff (push) Has been cancelled
Test / Core Tests (push) Has been cancelled
Test / Offline Coverage Tests (Python 3.10) (push) Has been cancelled
Test / Offline Coverage Tests (Python 3.11) (push) Has been cancelled
Test / Offline Coverage Tests (Python 3.12) (push) Has been cancelled
Test / Offline Coverage Tests (Python 3.13) (push) Has been cancelled
Test / Offline Coverage Tests (Python 3.9) (push) Has been cancelled
Test / Full Coverage (Python 3.11) (push) Has been cancelled
Test / Core Provider Tests (OpenAI) (push) Has been cancelled
Test / Core Provider Tests (Anthropic) (push) Has been cancelled
Test / Core Provider Tests (Google) (push) Has been cancelled
Test / Core Provider Tests (Other) (push) Has been cancelled
Test / Anthropic Tests (push) Has been cancelled
Test / Gemini Tests (push) Has been cancelled
Test / Google GenAI Tests (push) Has been cancelled
Test / Vertex AI Tests (push) Has been cancelled
Test / OpenAI Tests (push) Has been cancelled
Test / Writer Tests (push) Has been cancelled
Test / Auto Client Tests (push) Has been cancelled
ty / type-check (push) Has been cancelled
66 lines
1.8 KiB
Python
66 lines
1.8 KiB
Python
import instructor
|
|
from openai import OpenAI
|
|
from pydantic import BaseModel
|
|
import base64
|
|
|
|
client = instructor.from_openai(OpenAI(), mode=instructor.Mode.MD_JSON)
|
|
|
|
|
|
class Circle(BaseModel):
|
|
x: int
|
|
y: int
|
|
color: str
|
|
|
|
|
|
def encode_image(image_path):
|
|
with open(image_path, "rb") as image_file:
|
|
return base64.b64encode(image_file.read()).decode("utf-8")
|
|
|
|
|
|
def draw_circle(image_size, num_circles, path):
|
|
from PIL import Image, ImageDraw
|
|
import random
|
|
|
|
image = Image.new("RGB", image_size, "white")
|
|
|
|
draw = ImageDraw.Draw(image)
|
|
for _ in range(num_circles):
|
|
# Randomize the circle properties
|
|
radius = 100 # random.randint(10, min(image_size)//5) # Radius between 10 and 1/5th of the smallest dimension
|
|
x = random.randint(radius, image_size[0] - radius)
|
|
y = random.randint(radius, image_size[1] - radius)
|
|
color = ["red", "black", "blue", "green"][random.randint(0, 3)]
|
|
|
|
circle_position = (x - radius, y - radius, x + radius, y + radius)
|
|
print(f"Generating circle at {x, y} with color {color}")
|
|
draw.ellipse(circle_position, fill=color, outline="black")
|
|
|
|
image.save(path)
|
|
|
|
|
|
img_path = "circle.jpg"
|
|
draw_circle((1024, 1024), 1, img_path)
|
|
base64_image = encode_image(img_path)
|
|
|
|
response = client.chat.completions.create(
|
|
model="gpt-4-vision-preview",
|
|
max_tokens=1800,
|
|
response_model=Circle,
|
|
messages=[
|
|
{
|
|
"role": "user",
|
|
"content": [
|
|
{"type": "text", "text": "find the circle"},
|
|
{
|
|
"type": "image_url",
|
|
"image_url": {"url": f"data:image/jpeg;base64,{base64_image}"},
|
|
},
|
|
],
|
|
}
|
|
],
|
|
)
|
|
|
|
print(
|
|
f"Found circle with center at x: {response.x}, y: {response.y} and color: {response.color}"
|
|
)
|