const express = require('express');
const axios = require('axios');
const bodyParser = require('body-parser');

const app = express();
const port = 3000;

app.use(bodyParser.json());

const CIVITAI_API_KEY = 'YOUR_CIVITAI_API_KEY'; // Substitua pelo seu API Key da Civitai

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

    try {
        const response = await axios.post('https://api.civitai.com/v1/images/generate', {
            prompt,
            api_key: CIVITAI_API_KEY,
        });

        res.json({
            imageUrl: response.data.image_url
        });
    } catch (error) {
        console.error(error);
        res.status(500).send('Erro ao gerar a imagem.');
    }
});

app.listen(port, () => {
    console.log(`Servidor rodando na porta ${port}`);
});