messages.controller.js

import Message from '../models/message.model';
import Room from '../models/room.model'
import OpenAI from 'openai';
import ENV from '../config.env'

/** GET: http://localhost:3000/api/chat/roomid */
export async function getChat(req, res){
    try {
        
        const { roomid } = req.query;

        if(!roomid) return res.status(400).json({ error : "No room id present...!"});

        const messages = await Message.find({ room : roomid }, { __v: 0, room: 0})

        if(!messages) return res.status(400).json({ error : "No message found...!"});

        return res.status(200).json({ success : true, data: messages })

    } catch (error) {
        return res.status(400).json({ error });
    }
}

/** POST: http://localhost:3000/api/chat/roomid */
/** POST: http://localhost:3000/api/chat/roomid */
export async function createChat(req, res){
    const { roomid } = req.query;
    const { question } = req.body;

    if(!roomid) return res.status(400).json({ error : "No room id present...!"});
    if (!question) return res.status(400).json({ error: "Cannot get data from the user...!" });

    /** get current room */
    let rooms;
    try {
        rooms = await Room.findOne({ _id : roomid });
    } catch(err) {
        console.error(err);
        return res.status(400).json({ error : "Error finding room...!"});
    }

    if(!rooms) return res.status(400).json({ error : "No room found...!"});

    /** CONFIG OPEN AI API */
    const openai = new OpenAI({
        apiKey: ENV.OPENAI_API_KEY // This is also the default, can be omitted
    });

    let completion;
    try{
        completion = await openai.ChatCompletion.create({
            model: "gpt-3.5-turbo",
            prompt: question,
            max_tokens: 300,
        });
    } catch(err) {
        console.error(err);
        return res.status(400).json({ error: "Error generating response" });
    }

    const message = new Message({
        question,
        answer : completion.choices[0].text,
        room : roomid
    });

    try {
        await message.save();
    } catch(err) {
        console.error(err);
        return res.status(400).json({ error: "Error saving message...!" });
    }

    /** push message in the room model */
    rooms.messages.push(message._id);

    /** save data in the room model */
    try {
        await rooms.save();
    } catch(err) {
        console.error(err);
        return res.status(400).json({ error: "Error saving room...!" });
    }

    return res.status(200).json({ success : true, data: message });
}