keys/sortpending.py

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)