72 lines
2.3 KiB
Python
72 lines
2.3 KiB
Python
import base64
|
|
import os
|
|
import shutil
|
|
|
|
from openai import OpenAI
|
|
from wand.image import Image
|
|
|
|
client = OpenAI()
|
|
|
|
# List all files and directories in the specified path
|
|
path = 'data/RAW/SC1/BR1'
|
|
pending = os.path.join(path, 'pending')
|
|
|
|
bitting = None
|
|
|
|
for file in sorted(os.listdir(pending)):
|
|
if file.startswith('.'):
|
|
continue
|
|
|
|
print(file)
|
|
filename = os.path.join(pending, file)
|
|
|
|
with Image(filename=filename) as img:
|
|
img.format = 'jpg'
|
|
img.resize(512, 384)
|
|
blob = img.make_blob()
|
|
blob_b64 = base64.b64encode(blob).decode('utf-8')
|
|
|
|
chat_completion = client.chat.completions.create(
|
|
messages=[
|
|
{
|
|
'role': 'user',
|
|
'content': [
|
|
{
|
|
'type': 'text',
|
|
'text': 'If this is a picture of a white card with black lettering on it, respond with just the 5 digit number written on the card. If this is a picture of a key, respond with "front" if you can see the engravings "SC1" & "USA", and "back" if you cannot.'
|
|
},
|
|
{
|
|
'type': 'image_url',
|
|
'image_url': {
|
|
'url': f'data:image/jpeg;base64,{blob_b64}'
|
|
}
|
|
}
|
|
]
|
|
}
|
|
],
|
|
model='gpt-4o',
|
|
)
|
|
|
|
content = chat_completion.choices[0].message.content
|
|
|
|
print(content)
|
|
|
|
match content:
|
|
case 'front':
|
|
assert bitting is not None, 'key image before card image'
|
|
dest = os.path.join(path, bitting, 'front.orf')
|
|
assert not os.path.exists(dest), 'key image already exists'
|
|
shutil.move(filename, dest)
|
|
|
|
case 'back':
|
|
assert bitting is not None, 'key image before card image'
|
|
dest = os.path.join(path, bitting, 'back.orf')
|
|
assert not os.path.exists(dest), 'key image already exists'
|
|
shutil.move(filename, dest)
|
|
|
|
case _:
|
|
bitting = content
|
|
os.makedirs(os.path.join(path, bitting), exist_ok=True)
|
|
dest = os.path.join(path, bitting, 'card.orf')
|
|
assert not os.path.exists(dest), 'card image already exists'
|
|
shutil.move(filename, dest) |