const axios = require('axios');

async function gpt(prompt, model = 'chatgpt4') {
    try {
        const model_list = {
            chatgpt4: {
                api: 'https://stablediffusion.fr/gpt4/predict2',
                referer: 'https://stablediffusion.fr/chatgpt4'
            },
            chatgpt3: {
                api: 'https://stablediffusion.fr/gpt3/predict',
                referer: 'https://stablediffusion.fr/chatgpt3'
            }
        };
        if (!prompt) throw new Error('Prompt is required');
        if (!model_list[model]) throw new Error(`List available models: ${Object.keys(model_list).join(', ')}`);
        
        const hmm = await axios.get(model_list[model].referer);
        const { data } = await axios.post(model_list[model].api, {
            prompt: prompt
        }, {
            headers: {
                accept: '*/*',
                'content-type': 'application/json',
                origin: 'https://stablediffusion.fr',
                referer: model_list[model].referer,
                cookie: hmm.headers['set-cookie'].join('; '),
                'user-agent': 'Mozilla/5.0 (Linux; Android 10; K) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/136.0.0.0 Mobile Safari/537.36'
            }
        });
        
        return data.message;
    } catch (error) {
        throw new Error(error.message);
    }
}

// Usage:
const resp = await gpt('hi, how are you?', 'chatgpt4');
console.log(resp);