From 4b421c66a5a145320c601e20ad4957a515a4dc7f Mon Sep 17 00:00:00 2001
From: ZhaiSoul <842607283@qq.com>
Date: Wed, 29 Apr 2026 14:42:29 +0800
Subject: [PATCH] docs: improve NVIDIA GPU count input

---
 .../components/NvidiaGpuConfig.tsx            | 13 ++++++------
 .../DockerComposeGenerator/generator/index.ts |  2 +-
 .../hooks/useConfigGenerator.ts               | 20 +++++++++----------
 3 files changed, 17 insertions(+), 18 deletions(-)

diff --git a/docs/src/components/DockerComposeGenerator/components/NvidiaGpuConfig.tsx b/docs/src/components/DockerComposeGenerator/components/NvidiaGpuConfig.tsx
index 5863ac386..9c9be5e6a 100644
--- a/docs/src/components/DockerComposeGenerator/components/NvidiaGpuConfig.tsx
+++ b/docs/src/components/DockerComposeGenerator/components/NvidiaGpuConfig.tsx
@@ -16,6 +16,8 @@ export default function NvidiaGpuConfig({
   onGpuCountChange,
   onGpuDeviceIdChange,
 }: Props) {
+  const showDeviceId = gpuCount !== "";
+
   return (
     <div className={styles.nvidiaConfig}>
       <div className={styles.formGroup}>
@@ -25,16 +27,15 @@ export default function NvidiaGpuConfig({
         <input
           id="dcg-gpu-count"
           type="text"
+          inputMode="numeric"
+          pattern="[0-9]*"
           className={styles.input}
           value={gpuCount}
-          placeholder="1 or all"
-          onChange={(e) => onGpuCountChange(e.target.value)}
+          placeholder="all"
+          onChange={(e) => onGpuCountChange(e.target.value.replace(/\D/g, ""))}
         />
-        <p className={styles.helpText}>
-          Enter a number (e.g. 1, 2, 3) or &quot;all&quot; to use all GPUs
-        </p>
       </div>
-      {gpuCount !== "all" && (
+      {showDeviceId && (
         <div className={styles.formGroup}>
           <label htmlFor="dcg-gpu-device-id" className={styles.label}>
             GPU device IDs (required, comma-separated):
diff --git a/docs/src/components/DockerComposeGenerator/generator/index.ts b/docs/src/components/DockerComposeGenerator/generator/index.ts
index 24b8ab263..589d9b0b6 100644
--- a/docs/src/components/DockerComposeGenerator/generator/index.ts
+++ b/docs/src/components/DockerComposeGenerator/generator/index.ts
@@ -123,7 +123,7 @@ function buildEnvironment(
 function buildDeploy(device: DeviceConfig, input: GeneratorInput): string[] {
   if (device.id === "stable-tensorrt") {
     const count = input.nvidiaGpuCount || "all";
-    const isAll = count.toLowerCase() === "all";
+    const isAll = count === "all";
     const deviceId = input.nvidiaGpuDeviceId?.trim();
 
     if (isAll) {
diff --git a/docs/src/components/DockerComposeGenerator/hooks/useConfigGenerator.ts b/docs/src/components/DockerComposeGenerator/hooks/useConfigGenerator.ts
index 72821be26..394790b27 100644
--- a/docs/src/components/DockerComposeGenerator/hooks/useConfigGenerator.ts
+++ b/docs/src/components/DockerComposeGenerator/hooks/useConfigGenerator.ts
@@ -29,7 +29,7 @@ export function useConfigGenerator() {
     return initial;
   });
 
-  const [nvidiaGpuCount, setNvidiaGpuCount] = useState("all");
+  const [nvidiaGpuCount, setNvidiaGpuCount] = useState("");
   const [nvidiaGpuDeviceId, setNvidiaGpuDeviceId] = useState("");
   const [configPath, setConfigPath] = useState("");
   const [mediaPath, setMediaPath] = useState("");
@@ -56,7 +56,7 @@ export function useConfigGenerator() {
       }
       return next;
     });
-    setNvidiaGpuCount("all");
+    setNvidiaGpuCount("");
     setNvidiaGpuDeviceId("");
     setGpuDeviceIdError(false);
   }, []);
@@ -121,15 +121,13 @@ export function useConfigGenerator() {
   );
 
   const handleNvidiaGpuCountChange = useCallback((value: string) => {
-    const lower = value.trim().toLowerCase();
-    if (lower === "all" || lower === "" || /^[0-9]+$/.test(lower)) {
-      setNvidiaGpuCount(lower || "all");
-      if (lower === "all") {
-        setNvidiaGpuDeviceId("");
-        setGpuDeviceIdError(false);
-      } else if (/^[0-9]+$/.test(lower)) {
-        setGpuDeviceIdError(false);
-      }
+    // Only allow digits
+    setNvidiaGpuCount(value);
+    if (value === "") {
+      setNvidiaGpuDeviceId("");
+      setGpuDeviceIdError(false);
+    } else {
+      setGpuDeviceIdError(false);
     }
   }, []);