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');
  });
});