From b6763fb8847df5a5678f37137e7a702569e5c925 Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Mon, 5 Sep 2022 23:08:06 +0300
Subject: added random artist button added a setting for padding when doing
 inpaint at original resolution

---
 modules/artists.py | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)
 create mode 100644 modules/artists.py

(limited to 'modules/artists.py')

diff --git a/modules/artists.py b/modules/artists.py
new file mode 100644
index 00000000..3612758b
--- /dev/null
+++ b/modules/artists.py
@@ -0,0 +1,25 @@
+import os.path
+import csv
+from collections import namedtuple
+
+Artist = namedtuple("Artist", ['name', 'weight', 'category'])
+
+
+class ArtistsDatabase:
+    def __init__(self, filename):
+        self.cats = set()
+        self.artists = []
+
+        if not os.path.exists(filename):
+            return
+
+        with open(filename, "r", newline='', encoding="utf8") as file:
+            reader = csv.DictReader(file)
+
+            for row in reader:
+                artist = Artist(row["artist"], float(row["score"]), row["category"])
+                self.artists.append(artist)
+                self.cats.add(artist.category)
+
+    def categories(self):
+        return sorted(self.cats)
-- 
cgit v1.2.3