Upload 16 files

Dockerfile

@@ -1,11 +1,16 @@
-FROM node:20
+FROM node:20-alpine
 
+# Set the working directory inside the container
 WORKDIR /app
 
-COPY . .
+# Copy package files first (better for Docker caching)
+COPY package*.json ./
 
+# Install dependencies
 RUN npm install
 
-EXPOSE 7860
+# Copy the rest of your code (server.js, etc.)
+COPY . .
 
-CMD ["node", "server.js"]
+# Run the app
+CMD ["npm", "start"]

package.json

@@ -1,10 +1,13 @@
 {
-  "name": "hf-web",
+  "name": "hf-gradio-docker-app",
   "version": "1.0.0",
+  "description": "Connecting to Gradio with HF token",
+  "main": "server.js",
+  "type": "module",
+  "scripts": {
+    "start": "node server.js"
+  },
   "dependencies": {
-    "@gradio/client": "^1.9.0",
-    "express": "^4.18.2",
-    "dotenv": "^16.4.5",
-    "node-fetch": "^3.3.2"
+    "@gradio/client": "^1.0.0"
   }
 }

server.js

@@ -18,12 +18,22 @@ if (!HF_TOKEN) {
 app.use(express.json({ limit: "10mb" }));
 
 app.post("/api/proxy", async (req, res) => {
-  const { fn, args } = req.body;
+  let { fn, args } = req.body;
+  
   if (!fn) return res.status(400).json({ error: "Missing 'fn' in request body" });
 
+  // THE FIX: Scrub the 'args' object clean before passing it to Gradio.
+  // If your frontend accidentally sent 'read_token' in the payload, this deletes it.
+  if (args && typeof args === 'object' && !Array.isArray(args)) {
+    delete args.read_token;
+    delete args.hf_token;
+  }
+
   try {
+    // Auth is handled cleanly right here
     const client = await Client.connect("pockit-cloud/main", { hf_token: HF_TOKEN });
-    // Pass args directly to predict
+    
+    // Now 'args' only contains the actual model inputs
     const result = await client.predict(fn, args);
     res.json({ data: result.data });
   } catch (err) {
@@ -42,11 +52,14 @@ app.get("/api/download", async (req, res) => {
     const response = await fetch(url, {
       headers: { Authorization: `Bearer ${HF_TOKEN}` },
     });
+    
     if (!response.ok) {
       return res.status(response.status).json({ error: "File fetch failed" });
     }
+    
     res.setHeader("Content-Disposition", `attachment; filename=\"${file.split("/").pop()}\"`);
     res.setHeader("Content-Type", response.headers.get("content-type") || "application/octet-stream");
+    
     response.body.pipe(res);
   } catch (err) {
     console.error("Download error:", err);