add tag filter

app.py

@@ -289,7 +289,7 @@ class Style(str, Enum):
 
 
 @ app.get("/api/models")
-def get_page(page: int = 1, sort: Sort = Sort.trending, style: Style = Style.all):
+def get_page(page: int = 1, sort: Sort = Sort.trending, style: Style = Style.all, tag: str = None):
     page = page if page > 0 else 1
     if sort == Sort.trending:
         sort_query = "likes / MYPOWER((JULIANDAY('now') - JULIANDAY(datetime(json_extract(data, '$.lastModified')))) + 2, 2) DESC"
@@ -312,16 +312,24 @@ def get_page(page: int = 1, sort: Sort = Sort.trending, style: Style = Style.all
     with database.get_db() as db:
         cursor = db.cursor()
         cursor.execute(f"""
-            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 > 3 AND {style_query}
-            ORDER BY {sort_query}
-            LIMIT {MAX_PAGE_SIZE} OFFSET {(page - 1) * MAX_PAGE_SIZE}
-        """)
+                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
+                    ) AS subquery
+                    WHERE (? IS NULL AND likes > 3 OR ? IS NOT NULL)
+                    AND {style_query}
+                    AND (? IS NULL OR EXISTS (
+                            SELECT 1
+                            FROM json_each(json_extract(data, '$.meta.tags'))
+                            WHERE json_each.value = ?
+                        ))
+                    ORDER BY {sort_query}
+                    LIMIT {MAX_PAGE_SIZE} OFFSET {(page - 1) * MAX_PAGE_SIZE};
+        """, (tag, tag, tag, tag))
         results = cursor.fetchall()
         total = results[0]['total'] if results else 0
         total_pages = (total + MAX_PAGE_SIZE - 1) // MAX_PAGE_SIZE