-
Notifications
You must be signed in to change notification settings - Fork 0
/
server.mjs
53 lines (43 loc) · 1.59 KB
/
server.mjs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
import dotenv from 'dotenv';
import express from 'express';
import fetch from 'node-fetch';
import path from 'path';
dotenv.config();
const app = express();
const PORT = process.env.PORT || 3001;
app.use(express.json());
// Serve static files from 'public' directory
app.use(express.static('public'));
// Fallback route handler for the root to serve 'index.html'
app.get('/', (req, res) => {
res.sendFile(path.join(__dirname, 'public', 'index.html'));
});
app.post('/generate-text', async (req, res) => {
const textPrompt = req.body.prompt;
const apiUrl = `https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent?key=${process.env.API_KEY}`;
const requestBody = {
"contents": [{
"parts": [{
"text": textPrompt
}]
}]
};
try {
const response = await fetch(apiUrl, {
method: 'POST',
headers: {'Content-Type': 'application/json'},
body: JSON.stringify(requestBody)
});
const data = await response.json();
console.log("API Response:", JSON.stringify(data, null, 2)); // Debug output
// Correctly extracting text based on the structure shown in your API response
const generatedText = data.candidates[0].content.parts.map(part => part.text).join('\n');
res.json({ generatedText });
} catch (error) {
console.error('Error calling Google Gemini API:', error);
res.status(500).send('Failed to generate text');
}
});
app.listen(PORT, () => {
console.log(`Server running on http://localhost:${PORT}`);
});