Update server.js

server.js

@@ -1,80 +1,76 @@
 const express = require('express');
-const { spawn } = require('child_process');
-const { createProxyMiddleware } = require('http-proxy-middleware');
-const path = require('path');
+const path    = require('path');
 
-const app = express();
+const app  = express();
 const PORT = 7860;
-const PROXY_PORT = 3000;
 
-// ─── Lance le proxy free-claude-code en arrière-plan ───────────────────────
-const proxy = spawn('node', ['/app/proxy/index.js'], {
-  env: {
-    ...process.env,
-    PORT: String(PROXY_PORT),
-    NVIDIA_API_KEY: process.env.NVIDIA_API_KEY || ''
-  },
-  stdio: 'inherit'
-});
+app.use(express.json({ limit: '10mb' }));
+app.use(express.static(path.join(__dirname, 'public')));
 
-proxy.on('error', (err) => console.error('Proxy error:', err));
-proxy.on('exit', (code) => console.log('Proxy exited with code:', code));
+const MODEL_MAP = {
+  'kimi-k2':   'moonshotai/kimi-k2-instruct',
+  'glm-4':     'thudm/glm-4-9b-chat',
+  'minimax-m2':'minimax/minimax-m2',
+  'devstral':  'mistralai/devstral-small-2505',
+  'default':   'moonshotai/kimi-k2-instruct',
+};
 
-// Attendre que le proxy démarre
-setTimeout(() => {
-  console.log(`Proxy lancé sur le port ${PROXY_PORT}`);
-}, 2000);
+function resolveModel(name) {
+  if (!name) return MODEL_MAP['default'];
+  const key = Object.keys(MODEL_MAP).find(k => name.toLowerCase().includes(k));
+  return key ? MODEL_MAP[key] : MODEL_MAP['default'];
+}
 
-// ─── Servir l'interface web ─────────────────────────────────────────────────
-app.use(express.static(path.join(__dirname, 'public')));
-app.use(express.json());
+app.get('/api/status', (_req, res) => {
+  const key = process.env.NVIDIA_API_KEY || '';
+  res.json({
+    status: 'ok',
+    nvidia_key_set: key.length > 10,
+    models: Object.keys(MODEL_MAP).filter(k => k !== 'default'),
+  });
+});
 
-// ─── Route API — relaie vers le proxy Claude ────────────────────────────────
 app.post('/api/chat', async (req, res) => {
+  const apiKey = process.env.NVIDIA_API_KEY;
+  if (!apiKey || apiKey.length < 10) {
+    return res.status(500).json({ error: 'NVIDIA_API_KEY manquante. Ajoute-la dans Settings > Secrets.' });
+  }
+
+  const { messages = [], model = 'kimi-k2', max_tokens = 2048 } = req.body;
+  const nvidiaModel = resolveModel(model);
+
   try {
-    const response = await fetch(`http://localhost:${PROXY_PORT}/v1/messages`, {
+    const response = await fetch('https://integrate.api.nvidia.com/v1/chat/completions', {
       method: 'POST',
       headers: {
-        'Content-Type': 'application/json',
-        'x-api-key': 'fake-key',
-        'anthropic-version': '2023-06-01'
+        'Content-Type':  'application/json',
+        'Authorization': `Bearer ${apiKey}`,
       },
       body: JSON.stringify({
-        model: req.body.model || 'claude-sonnet-4-20250514',
-        max_tokens: req.body.max_tokens || 2048,
-        messages: req.body.messages,
-        stream: req.body.stream || false
-      })
+        model:       nvidiaModel,
+        messages,
+        max_tokens,
+        temperature: 0.7,
+        stream:      false,
+      }),
     });
 
-    if (req.body.stream) {
-      res.setHeader('Content-Type', 'text/event-stream');
-      res.setHeader('Cache-Control', 'no-cache');
-      res.setHeader('Connection', 'keep-alive');
-      response.body.pipeTo(new WritableStream({
-        write(chunk) { res.write(chunk); }
-      })).finally(() => res.end());
-    } else {
-      const data = await response.json();
-      res.json(data);
+    if (!response.ok) {
+      const err = await response.text();
+      return res.status(response.status).json({ error: `NVIDIA API ${response.status}: ${err}` });
     }
+
+    const data  = await response.json();
+    const reply = data?.choices?.[0]?.message?.content || '';
+    res.json({ content: [{ type: 'text', text: reply }] });
+
   } catch (err) {
-    console.error('API error:', err);
     res.status(500).json({ error: err.message });
   }
 });
 
-// ─── Status endpoint ────────────────────────────────────────────────────────
-app.get('/api/status', (req, res) => {
-  const key = process.env.NVIDIA_API_KEY;
-  res.json({
-    status: 'ok',
-    proxy_port: PROXY_PORT,
-    nvidia_key_set: !!(key && key.length > 10),
-    models: ['Kimi K2', 'GLM 4.7', 'MiniMax M2', 'Devstral']
-  });
-});
-
 app.listen(PORT, '0.0.0.0', () => {
-  console.log(`✅ Serveur démarré sur http://0.0.0.0:${PORT}`);
-});
+  const key = process.env.NVIDIA_API_KEY || '';
+  console.log(`Server on port ${PORT}`);
+  console.log(`NVIDIA key: ${key.length > 10 ? 'OK' : 'MISSING'}`);
+});