File size: 959 Bytes
28b2eff
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
const express = require('express');
const bodyParser = require('body-parser');
const axios = require('axios');
const { MongoClient } = require('mongodb');

const app = express();
app.use(bodyParser.json());

const uri = 'your-mongodb-connection-string';
const client = new MongoClient(uri, { useNewUrlParser: true, useUnifiedTopology: true });

client.connect(err => {
  if (err) throw err;
  const collection = client.db("code-generation").collection("snippets");

  app.post('/generate-code', async (req, res) => {
    const { prompt } = req.body;

    try {
      const response = await axios.post('http://localhost:8000/generate-code', { prompt });
      const code = response.data.code;
      await collection.insertOne({ prompt, code });
      res.json(response.data);
    } catch (error) {
      res.status(500).json({ error: 'Error generating code' });
    }
  });

  app.listen(5000, () => {
    console.log('Server running on port 5000');
  });
});