keys/sortpending.py

import os
import base64
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)