AI Coach fixed

server/src/routes/ai.ts

@@ -5,7 +5,7 @@ import jwt from 'jsonwebtoken';
 const router = express.Router();
 const JWT_SECRET = process.env.JWT_SECRET || 'secret';
 const API_KEY = process.env.API_KEY;
-const MODEL_ID = 'gemini-1.5-flash';
+const MODEL_ID = 'gemini-2.0-flash';
 
 const authenticate = (req: any, res: any, next: any) => {
     const token = req.headers.authorization?.split(' ')[1];
@@ -35,11 +35,11 @@ router.post('/chat', async (req, res) => {
         const { systemInstruction, userMessage } = req.body;
 
         const model = ai.getGenerativeModel({
-            model: MODEL_ID,
-            systemInstruction
+            model: MODEL_ID
         });
 
-        const result = await model.generateContent(userMessage);
+        const prompt = `${systemInstruction}\n\nUser: ${userMessage}`;
+        const result = await model.generateContent(prompt);
         const response = result.response.text();
 
         res.json({ response });