move filter to backend

app.py

@@ -224,8 +224,16 @@ class Sort(str, Enum):
     likes = "likes"
 
 
+class Style(str, Enum):
+    all = "all"
+    anime = "anime"
+    s3D = "3d"
+    realistic = "realistic"
+    nsfw = "nsfw"
+
+
 @ app.get("/api/models")
-def get_page(page: int = 1, sort: Sort = Sort.trending):
+def get_page(page: int = 1, sort: Sort = Sort.trending, style: Style = Style.all):
     page = page if page > 0 else 1
     if sort == Sort.trending:
         sort_query = "((likes + downloads)/2) / MYPOWER((JULIANDAY('now') - JULIANDAY(datetime(json_extract(data, '$.lastModified')))) + 2, 1.5) DESC"
@@ -233,13 +241,27 @@ def get_page(page: int = 1, sort: Sort = Sort.trending):
         sort_query = "datetime(json_extract(data, '$.lastModified')) DESC"
     elif sort == Sort.likes:
         sort_query = "likes DESC"
+    if style == Style.all:
+        style_query = "1"
+    elif style == Style.anime:
+        style_query = "json_extract(data, '$.class.anime') > 0.1 AND isNFSW = false"
+    elif style == Style.s3D:
+        style_query = "json_extract(data, '$.class.3d') > 0.1 AND isNFSW = false"
+    elif style == Style.realistic:
+        style_query = "json_extract(data, '$.class.real_life') > 0.1 AND isNFSW = false"
+    elif style == Style.nsfw:
+        style_query = "isNFSW = true"
 
     with database.get_db() as db:
         cursor = db.cursor()
         cursor.execute(f"""
-            SELECT *, COUNT(*) OVER() AS total
-            FROM models
-            WHERE likes > 4
+            SELECT *, COUNT(*) OVER() AS total,  isNFSW
+            FROM (
+                SELECT * ,
+                        json_extract(data, '$.class.explicit') > 0.3 OR json_extract(data, '$.class.suggestive') > 0.3 AS isNFSW
+                FROM models
+            )
+            WHERE likes > 0 AND {style_query}
             ORDER BY {sort_query}
             LIMIT {MAX_PAGE_SIZE} OFFSET {(page - 1) * MAX_PAGE_SIZE}
         """)
@@ -248,10 +270,12 @@ def get_page(page: int = 1, sort: Sort = Sort.trending):
         total_pages = (total + MAX_PAGE_SIZE - 1) // MAX_PAGE_SIZE
         models_data = []
         for result in results:
+            print(list(result))
             data = json.loads(result['data'])
             # update downloads and likes from db table
             data['downloads'] = result['downloads']
             data['likes'] = result['likes']
+            data['isNFSW'] = bool(result['isNFSW'])
             models_data.append(data)
 
     return {