staff only is public

src/routes/+page.svelte

@@ -15,6 +15,9 @@
 	import Drawer from "$lib/components/models/drawer/Drawer.svelte";
 	import { onMount } from "svelte";
 	import type { ModelCard } from "$lib/type";
+	import { get } from "svelte/store";
+	import { userStore } from "$lib/stores/use-user";
+	import Add from "$lib/components/community/reactions/Add.svelte";
 
 	let data: {
 		models: ModelCard[],
@@ -23,9 +26,11 @@
 		models: [],
 		total_items: 0,
 	}
+	let user = get(userStore);
+
 	let form: Record<string, string> = {
 		filter: $page.url.searchParams.get('filter') ?? "hotest",
-		search: "",
+		search: $page.url.searchParams.get('search') ?? "",
 		page: "0"
 	}
 	let submitModelDialog = false;
@@ -85,7 +90,21 @@
 		Explore Models ({data?.total_items ?? 0})
 	</h1>
 	<div class="flex items-start sm:items-center justify-between mt-5 flex-col sm:flex-row gap-5 sm:justify-between">
-		<Radio options={MODELS_FILTER_OPTIONS} value="{form.filter}" onChange={handleChangeFilter} />
+		<Radio
+			options={[
+				...MODELS_FILTER_OPTIONS,
+				...(user?.is_admin ? [
+					{
+					label: "Staff only",
+					value: "staff_only",
+					icon: "lets-icons:view-hide-fill",
+					iconColor: "text-yellow-500"
+				}
+				] : [])
+			]}
+			value="{form.filter}"
+			onChange={handleChangeFilter}
+		/>
 		<div class="items-center justify-end gap-5 hidden lg:flex">
 			<Button href="https://huggingface.co/new/stable-diffusion-lora" target="_blank" icon="ic:round-plus" theme="dark" size="lg">Create</Button>
 			<!-- <UserIsLogged> -->

src/routes/api/models/+server.ts

@@ -30,9 +30,13 @@ export async function GET(request : RequestEvent) {
     orderBy['createdAt'] = 'desc'
   }
 
+  const only_not_public = filter === 'staff_only';
+
   const cards = await prisma.model.findMany({
     where: {
-      ...(IS_ADMIN ? {} : { isPublic: true }),
+      ...(
+        (IS_ADMIN && only_not_public) ? { isPublic: false } : !only_not_public ? {} : { isPublic: true }
+      ),
       OR: [
         { id: { contains: search } },
       ]