add filtering options

app.py

@@ -1,3 +1,4 @@
+from enum import Enum
 import os
 import re
 import aiohttp
@@ -176,12 +177,12 @@ async def sync_data():
                                "class": classifier
                            })])
             db.commit()
-    
-    if(len(models) > 0):
+
+    if (len(models) > 0):
         print("Updating repository")
         subprocess.Popen(
             "git add . && git commit --amend -m 'update' && git push --force", cwd=DB_FOLDER, shell=True)
-    else:  
+    else:
         print("No new models found")
 
 app = FastAPI()
@@ -203,18 +204,31 @@ app.add_middleware(
 MAX_PAGE_SIZE = 30
 
 
+class Sort(str, Enum):
+    trending = "trending"
+    recent = "recent"
+    likes = "likes"
+
+
 @ app.get("/api/models")
-def get_page(page: int = 1):
+def get_page(page: int = 1, sort: Sort = Sort.trending):
     page = page if page > 0 else 1
+    if sort == Sort.trending:
+        sort_query = "json_extract(data, '$.likes') / POWER((JULIANDAY('now') - JULIANDAY(datetime(json_extract(data, '$.lastModified')))) + 2, 2) DESC"
+    elif sort == Sort.recent:
+        sort_query = "datetime(json_extract(data, '$.lastModified')) DESC"
+    elif sort == Sort.likes:
+        sort_query = "json_extract(data, '$.likes') DESC"
+
     with database.get_db() as db:
         cursor = db.cursor()
-        cursor.execute("""
+        cursor.execute(f"""
             SELECT *, COUNT(*) OVER() AS total
             FROM models
             WHERE json_extract(data, '$.likes') > 4
-            ORDER BY datetime(json_extract(data, '$.lastModified')) DESC, json_extract(data, '$.likes') DESC
-            LIMIT ? OFFSET ?
-        """, (MAX_PAGE_SIZE, (page - 1) * MAX_PAGE_SIZE))
+            ORDER BY {sort_query}
+            LIMIT {MAX_PAGE_SIZE} OFFSET {(page - 1) * MAX_PAGE_SIZE}
+        """)
         results = cursor.fetchall()
         total = results[0]['total'] if results else 0
         total_pages = (total + MAX_PAGE_SIZE - 1) // MAX_PAGE_SIZE
@@ -230,8 +244,8 @@ def read_root():
     return "Just a bot to sync data from diffusers gallery"
 
 
-@app.on_event("startup")
-@repeat_every(seconds=60 * 60 * 24, wait_first=False)
-async def repeat_sync():
-    await sync_data()
-    return "Synced data to huggingface datasets"
+# @app.on_event("startup")
+# @repeat_every(seconds=60 * 60 * 24, wait_first=True)
+# async def repeat_sync():
+#     await sync_data()
+#     return "Synced data to huggingface datasets"