Update server.js

server.js

@@ -1,11 +1,40 @@
-const express = require('express'); const { GoogleGenerativeAI } = require('@google/generative-ai'); const fs = require('fs'); const path = require('path');
+const express = require('express');
+const { GoogleGenerativeAI } = require('@google/generative-ai');
+const fs = require('fs');
+const path = require('path');
 
-const app = express(); const apiKey = 'AIzaSyCRpX4SnNQl__BNLcVWqgzKdbIlt06vod0'; const genAI = new GoogleGenerativeAI(apiKey); const tmpDir = path.join(require('os').tmpdir(), 'gemini_sessions'); if (!fs.existsSync(tmpDir)) fs.mkdirSync(tmpDir);
+const app = express();
+const apiKey = 'AIzaSyCRpX4SnNQl__BNLcVWqgzKdbIlt06vod0';
+const genAI = new GoogleGenerativeAI(apiKey);
+const tmpDir = path.join(require('os').tmpdir(), 'gemini_sessions');
+if (!fs.existsSync(tmpDir)) fs.mkdirSync(tmpDir);
 
-function getSessionFile(sessionId) { return path.join(tmpDir, ${sessionId}.json); }
+function getSessionFile(sessionId) {
+    return path.join(tmpDir, `${sessionId}.json`);
+}
 
-async function gemini(sessionId, prompt) { const sessionFile = getSessionFile(sessionId); let session; if (fs.existsSync(sessionFile)) { session = JSON.parse(fs.readFileSync(sessionFile, 'utf8')); } else { session = await genAI.getGenerativeModel({ model: 'gemini-2.0-flash' }).startChat(); } const result = await session.sendMessage(prompt); fs.writeFileSync(sessionFile, JSON.stringify(session)); return result.response.text(); }
+async function gemini(sessionId, prompt) {
+    const sessionFile = getSessionFile(sessionId);
+    let session;
+    if (fs.existsSync(sessionFile)) {
+        session = JSON.parse(fs.readFileSync(sessionFile, 'utf8'));
+    } else {
+        session = await genAI.getGenerativeModel({ model: 'gemini-2.0-flash' }).startChat();
+    }
+    const result = await session.sendMessage(prompt);
+    fs.writeFileSync(sessionFile, JSON.stringify(session));
+    return result.response.text();
+}
 
-app.get('/gemini', async (req, res) => { try { const { question, sessionId } = req.query; if (!question || !sessionId) return res.status(400).json({ error: 'Missing parameters' }); const reply = await gemini(sessionId, question); res.json({ reply }); } catch (error) { res.status(500).json({ error: error.message }); } });
+app.get('/gemini', async (req, res) => {
+    try {
+        const { question, sessionId } = req.query;
+        if (!question || !sessionId) return res.status(400).json({ error: 'Missing parameters' });
+        const reply = await gemini(sessionId, question);
+        res.json({ reply });
+    } catch (error) {
+        res.status(500).json({ error: error.message });
+    }
+});
 
-app.listen(7860);
+app.listen(7860);