From 8f8e147a127c3f63de34dfb0952137e646a68fad Mon Sep 17 00:00:00 2001
From: Ian Gulliver <github@flamingcow.io>
Date: Sat, 10 Aug 2024 16:10:25 -0700
Subject: [PATCH] Image sorter initial commit

---
 .gitignore |  2 ++
 main.py    | 72 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 74 insertions(+)
 create mode 100644 .gitignore
 create mode 100644 main.py

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..085bbc3
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,2 @@
+data
+.idea
\ No newline at end of file
diff --git a/main.py b/main.py
new file mode 100644
index 0000000..1b9d701
--- /dev/null
+++ b/main.py
@@ -0,0 +1,72 @@
+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 engraving "SC1" 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)
\ No newline at end of file