add content embedder

1 files changed, 15 insertions, 0 deletions

diff --git a/src/imgbeddings-api.py b/src/imgbeddings-api.py
new file mode 100644
index 0000000..795c625
--- /dev/null
+++ b/src/imgbeddings-api.py
@@ -0,0 +1,15 @@
+from PIL import Image
+from imgbeddings import imgbeddings
+import sys
+import json as j
+#from itertools import batched # TODO das hier ab python 3.12
+
+b = imgbeddings()
+
+ems = []
+
+for f in sys.argv[1:]: # TODO this should be batched for faster ai stuff
+    im = Image.open(open(f, "rb"))
+    ems += [b.to_embeddings(im)]
+
+print(j.dumps([em[0].tolist() for em in ems]))