temp

local myURL = "ws://localhost:8080"
local ws = http.websocket(myURL)

if ws == nil then
    print("Failed to connect to " .. myURL)
    return
else
    print("Connected to " .. myURL)
end

function Write_File(path, data)
    fs.delete(path)

    local file = fs.open(path, "w")

    if file then
        file.write(data) -- Write content to the file
        file.close() -- Close the file
        return true
    end
    return false
end

function Chat()
    local event, url, message
    local input

    while true do

        term.setTextColor(colors.blue)

        io.write("> ")
        input = read()

        if input == "exit" then
            break
        end

        ws.send(input)

        local data

        repeat
            event, url, Response = os.pullEvent("websocket_message")
        until url == myURL and textutils.unserialiseJSON(Response).type == "chat"

        data = textutils.unserialize(Response)

        term.setTextColor(colors.lightBlue)

        textutils.slowPrint(data.text)

    end
end

function AI_Functions()
    local data

        repeat
            event, url, Response = os.pullEvent("websocket_message")
        until url == myURL and textutils.unserialiseJSON(Response).type == "file"

        data = textutils.unserialize(Response)

        if Write_File(data.name, data.code) then
            term.setTextColor(colors.green)
            print("File \"".. data.name .."\" Created")
        else
            term.setTextColor(colors.red)
            print("Failed to create file \"".. data.name .."\"")
        end

end

parallel.waitForAny(Chat, AI_Functions)

ws.close()

term.setTextColor(colors.white)
print("Disconnected from server.")