Update server.js

server.js

@@ -1,11 +1,13 @@
 const express = require('express');
-const { GoogleGenerativeAI } = require('@google/generative-ai');
+const axios = require('axios');
 const fs = require('fs');
 const path = require('path');
 
 const app = express();
+app.set('json spaces', 2);
+
 const apiKey = 'AIzaSyCRpX4SnNQl__BNLcVWqgzKdbIlt06vod0';
-const genAI = new GoogleGenerativeAI(apiKey);
+const apiUrl = 'https://generativelanguage.googleapis.com/v1/models/gemini-2.0-flash:generateContent';
 const tmpDir = path.join(require('os').tmpdir(), 'gemini_sessions');
 if (!fs.existsSync(tmpDir)) fs.mkdirSync(tmpDir);
 
@@ -15,17 +17,20 @@ function getSessionFile(sessionId) {
 
 async function gemini(sessionId, prompt) {
     const sessionFile = getSessionFile(sessionId);
-    let chatSession;
+    let history = [];
     if (fs.existsSync(sessionFile)) {
-        chatSession = JSON.parse(fs.readFileSync(sessionFile, 'utf8'));
-    } else {
-        chatSession = await genAI.getGenerativeModel({ model: 'gemini-2.0-flash' }).startChat();
+        history = JSON.parse(fs.readFileSync(sessionFile, 'utf8'));
     }
+    history.push({ role: 'user', content: prompt });
+
+    const response = await axios.post(`${apiUrl}?key=${apiKey}`, {
+        prompt: { parts: [{ text: prompt }] }
+    });
     
-    const result = await chatSession.sendMessage(prompt);
-    
-    fs.writeFileSync(sessionFile, JSON.stringify(chatSession.history, null, 2));
-    return result.response.text();
+    const reply = response.data.candidates?.[0]?.content?.parts?.[0]?.text || 'No response';
+    history.push({ role: 'assistant', content: reply });
+    fs.writeFileSync(sessionFile, JSON.stringify(history, null, 2));
+    return reply;
 }
 
 app.get('/gemini', async (req, res) => {
@@ -33,9 +38,9 @@ app.get('/gemini', async (req, res) => {
         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 }, null, 2);
+        res.json({ reply });
     } catch (error) {
-        res.status(500).json({ error: error.message }, null, 2);
+        res.status(500).json({ error: error.message });
     }
 });