View source for Module:Riaa

local p = {}

function p.multi(frame)
    local type = frame.args[1] -- "platinum" or "diamond"
    local num = tonumber(frame.args[2]) or 1  -- Default to 1 if not provided
    local file = ""

    if type == "platinum" then
        file = "platinumrecord.png"
    elseif type == "diamond" then
        file = "diamondrecord.png"
    else
        return "" -- Invalid input, return nothing
    end

    local output = ""
    for i = 1, num do
        output = output .. "[[File:" .. file .. "|25px|" .. type:gsub("^%l", string.upper) .. "]]"
    end
    return output
end

return p