From 5ece717e3143e1ae6e2147bde821657c9a677fea Mon Sep 17 00:00:00 2001
From: Your Name <your.email@example.com>
Date: Sun, 17 May 2026 00:46:37 +0800
Subject: [PATCH] fix: hide hidden models in style pro selector

---
 app/api/style-pro/generate/route.ts      | 9 ++-------
 app/api/style-pro/model-options/route.ts | 9 ++-------
 2 files changed, 4 insertions(+), 14 deletions(-)

diff --git a/app/api/style-pro/generate/route.ts b/app/api/style-pro/generate/route.ts
index 67f2279ef..c3c0be0f5 100644
--- a/app/api/style-pro/generate/route.ts
+++ b/app/api/style-pro/generate/route.ts
@@ -2,7 +2,7 @@ import { NextResponse } from "next/server";
 import { z } from "zod";
 import { auth } from "@/app/(auth)/auth";
 import { generateCanvasImage } from "@/lib/canvas/generate-image";
-import { getSystemSetting, getUserById } from "@/lib/db/queries";
+import { getSystemSetting } from "@/lib/db/queries";
 import {
   findStyleProImageModelByRef,
   getStyleProImageModelEntries,
@@ -196,13 +196,8 @@ export async function POST(request: Request) {
 
     let selectedModelRef = config.modelId || undefined;
     if (body.modelId) {
-      const currentUser = await getUserById({ id: session.user.id }).catch(
-        () => null
-      );
-      const includeHidden =
-        session.user.type === "admin" || Boolean(currentUser?.isHighVip);
       const imageModelEntries = await getStyleProImageModelEntries({
-        includeHidden,
+        includeHidden: false,
       });
       const selectedModel = findStyleProImageModelByRef(
         imageModelEntries,
diff --git a/app/api/style-pro/model-options/route.ts b/app/api/style-pro/model-options/route.ts
index 0b363ae7c..ce05a8177 100644
--- a/app/api/style-pro/model-options/route.ts
+++ b/app/api/style-pro/model-options/route.ts
@@ -1,6 +1,6 @@
 import { NextResponse } from "next/server";
 import { auth } from "@/app/(auth)/auth";
-import { getSystemSetting, getUserById } from "@/lib/db/queries";
+import { getSystemSetting } from "@/lib/db/queries";
 import {
   findStyleProImageModelByRef,
   getStyleProImageModelEntries,
@@ -23,13 +23,8 @@ export async function GET() {
       );
     }
 
-    const currentUser = await getUserById({ id: session.user.id }).catch(
-      () => null
-    );
-    const includeHidden =
-      session.user.type === "admin" || Boolean(currentUser?.isHighVip);
     const [entries, setting] = await Promise.all([
-      getStyleProImageModelEntries({ includeHidden }),
+      getStyleProImageModelEntries({ includeHidden: false }),
       getSystemSetting({ key: STYLE_PRO_CONFIG_KEY }).catch(() => null),
     ]);
     const config = normalizeStyleProConfig(setting?.value);