script1.js

precision = 5;

function toggleCustomModel() {
    var modelDropdown = document.getElementById("model-dropdown");
    var modelInput = document.getElementById("model-input");
    console.log(modelDropdown.value);
    
    //if the model dropdown is set to custom, change the html to show the input box
    if (modelDropdown.value === "custom") {
        document.getElementById("params-custom").disabled = false;
        document.getElementById("custom-model-row-params").style.display = "";
        document.getElementById("params-tokens").disabled = false;
        document.getElementById("custom-model-row-tokens").style.display = "";
        // change the placeholder text
        document.getElementById("params-custom").placeholder = "Enter custom parameters";
        
    } else { 
        document.getElementById("params-custom").disabled = true;
        document.getElementById("custom-model-row-params").style.display = "none";
        document.getElementById("params-tokens").disabled = true;
        document.getElementById("custom-model-row-tokens").style.display = "none";
    }
    // run checkvars
    checkVars();


}



function toggleCustomCompute(){
    var computeDropdown = document.getElementById("compute-dropdown");
    var computeInput = document.getElementById("compute-input");

    //if the compute dropdown is set to custom, change the html to show the input box
    if (computeDropdown.value === "custom") {
        
        document.getElementById("compute-custom-input").disabled = false;
        document.getElementById("custom-compute-row").style.display = "";

    } else {
        document.getElementById("compute-custom-input").disabled = true;
        document.getElementById("custom-compute-row").style.display = "none";
    }
    checkVars();

}
var toUpdate = "";
var shouldUpdateNewField = true;
function checkVars(){
    var compute_params;
    var compute_tokens;
    var compute_flops;
    var compute_chips;
    var compute_utilization;
    var compute_time;

    console.log("checkVars ran");
    if (document.getElementById("model-dropdown").value != "custom") {
        if (document.getElementById("model-dropdown").value == "gpt-3") {
            compute_params = 175000000000;
            compute_tokens = 500000000000;
        } else if (document.getElementById("model-dropdown").value == "PaLM") {
            compute_params = 540000000000;
            compute_tokens = 780000000000;
        } else if (document.getElementById("model-dropdown").value == "gpt-2") {
            compute_params = 1500000000;
            compute_tokens = 400000000000;
        }
    } else {
        compute_params = (document.getElementById("params-custom").value == "") ? null : document.getElementById("params-custom").value;
        compute_tokens = (document.getElementById("params-tokens").value == "") ? null : document.getElementById("params-tokens").value;
    }
    if (document.getElementById("compute-dropdown").value != "custom") {
        if (document.getElementById("compute-dropdown").value == "a100") {
            compute_flops = 3.12e14;
        } else if (document.getElementById("compute-dropdown").value == "v100") {
            compute_flops = 1.3e14;
        } else if (document.getElementById("compute-dropdown").value == "rtx-3090") {
            compute_flops = .7e14;
        }
    } else {
        compute_flops = (document.getElementById("compute-custom-input").value == "") ? null : document.getElementById("compute-custom-input").value;
    }  
    compute_chips  = (document.getElementById("compute-chips").value == "") ? null : document.getElementById("compute-chips").value;
    compute_utilization = (document.getElementById("compute-utilization").value == "") ? 30 : document.getElementById("compute-utilization").value;
    compute_time = (document.getElementById("compute-time").value == "") ? null : document.getElementById("compute-time").value;

    

    // create array of the vars that are null
    var nullVars = [];
    if (compute_params == null) {
        nullVars.push("compute_params");
    }
    if (compute_tokens == null) {
        nullVars.push("compute_tokens");
    }
    if (compute_flops == null) {
        nullVars.push("compute_flops");
    }
    if (compute_chips == null) {
        nullVars.push("compute_chips");
    }
    if (compute_utilization == null) {
        nullVars.push("compute_utilization");
    }
    if (compute_time == null) {
        nullVars.push("compute_time");
    }
    // check if the length of the array is 1
    if (nullVars.length == 1 && shouldUpdateNewField) {
        // if it is, then we can calculate the missing variable
        if (compute_time==null) {
            if (shouldUpdateNewField) {
                shouldUpdateNewField = false;
                toUpdate = "time";
            }
            compute_time = 600*compute_params*compute_tokens / (compute_chips * compute_flops *compute_utilization);
            var timeWithUnit = get_time_unit(compute_time);
            time = timeWithUnit[0];
            unit = timeWithUnit[1];
            document.getElementById("compute-time").value = round_to_n_sig_figs(time,precision);
            document.getElementById("time-unit").innerHTML = unit;
            document.getElementById("compute-time").readOnly = true;
            updatePowerAndCost(get_power(compute_time,compute_chips),get_cost(compute_time,compute_chips))
        }
        if (compute_flops==null) {
            if (shouldUpdateNewField) {
                shouldUpdateNewField = false;
                toUpdate = "flops";
            }
            compute_flops = 600*compute_params*compute_tokens / (compute_time * compute_chips * compute_utilization);
            document.getElementById("compute-custom-input").value = round_to_n_sig_figs(compute_flops,precision);
            document.getElementById("compute-custom-input").readOnly = true;
            updatePowerAndCost(get_power(compute_time,compute_chips),get_cost(compute_time,compute_chips))
        }
        if (compute_chips==null) {
            if (shouldUpdateNewField) {
                shouldUpdateNewField = false;
                toUpdate = "chips";
            }
            compute_chips = 600*compute_params*compute_tokens / (compute_time * compute_flops * compute_utilization);
            document.getElementById("compute-chips").value = round_to_n_sig_figs(compute_chips,precision);
            document.getElementById("compute-chips").readOnly = true;
            updatePowerAndCost(get_power(compute_time,compute_chips),get_cost(compute_time,compute_chips))
        }
        if (compute_utilization==null) {
            if (shouldUpdateNewField) {
                toUpdate = "utilization";
                shouldUpdateNewField = false;
            }
            compute_utilization = 600*compute_params*compute_tokens / (compute_time * compute_flops * compute_chips);
            document.getElementById("compute-utilization").value = round_to_n_sig_figs(compute_utilization,precision);
            document.getElementById("compute-utilization").readOnly = true;
            updatePowerAndCost(get_power(compute_time,compute_chips),get_cost(compute_time,compute_chips))

        }
        if (compute_params==null) {
            if (shouldUpdateNewField) {
                toUpdate = "params";
                shouldUpdateNewField = false
            }
            compute_params = compute_tokens*compute_chips*compute_utilization*compute_time / (600*compute_tokens);
            document.getElementById("params-custom").value = round_to_n_sig_figs(compute_params,precision);
            document.getElementById("params-custom").readOnly = true;
            updatePowerAndCost(get_power(compute_time,compute_chips),get_cost(compute_time,compute_chips))

        }
        if (compute_tokens==null) {
            if (shouldUpdateNewField) {
                toUpdate = "tokens";
                shouldUpdateNewField = false
            }
            compute_tokens = compute_params*compute_chips*compute_utilization*compute_time / (600*compute_params);
            document.getElementById("params-tokens").value = round_to_n_sig_figs(compute_tokens,precision);
            document.getElementById("params-tokens").readOnly = true;
            updatePowerAndCost(get_power(compute_time,compute_chips),get_cost(compute_time,compute_chips))
        }
    }
    if (nullVars.length == 0) {
        if (toUpdate == "time") {
            compute_time = 600*compute_params*compute_tokens / (compute_chips * compute_flops *compute_utilization);
            // pass time into the function to convert it and unpack
            var timeWithUnit = get_time_unit(compute_time);
            time = timeWithUnit[0];
            unit = timeWithUnit[1];
            console.log(time);
            console.log(unit);
            document.getElementById("compute-time").value = round_to_n_sig_figs(time,precision);
            document.getElementById("time-unit").innerHTML = unit;
            updatePowerAndCost(get_power(compute_time,compute_chips),get_cost(compute_time,compute_chips))
        }
        if (toUpdate == "flops") {
            compute_flops = 600*compute_params*compute_tokens / (compute_time * compute_chips * compute_utilization);
            document.getElementById("compute-custom-input").value = round_to_n_sig_figs(compute_flops,precision);
            updatePowerAndCost(get_power(compute_time,compute_chips),get_cost(compute_time,compute_chips))
        }
        if (toUpdate == "chips") {
            compute_chips = 600*compute_params*compute_tokens / (compute_time * compute_flops * compute_utilization);
            document.getElementById("compute-chips").value = round_to_n_sig_figs(compute_chips,precision);
            updatePowerAndCost(get_power(compute_time,compute_chips),get_cost(compute_time,compute_chips))
        }
        if (toUpdate == "utilization") {
            compute_utilization = 600*compute_params*compute_tokens / (compute_time * compute_flops * compute_chips);
            document.getElementById("compute-utilization").value = round_to_n_sig_figs(compute_utilization,precision);
            updatePowerAndCost(get_power(compute_time,compute_chips),get_cost(compute_time,compute_chips))
        }
        if (toUpdate == "params") {
            compute_params = compute_tokens*compute_chips*compute_utilization*compute_time / (600*compute_tokens);
            document.getElementById("params-custom").value = round_to_n_sig_figs(compute_params,precision);
            updatePowerAndCost(get_power(compute_time,compute_chips),get_cost(compute_time,compute_chips))
        }
        if (toUpdate == "tokens") {
            compute_tokens = compute_params*compute_chips*compute_utilization*compute_time / (600*compute_params);
            document.getElementById("params-tokens").value = round_to_n_sig_figs(compute_tokens,precision);
            updatePowerAndCost(get_power(compute_time,compute_chips),get_cost(compute_time,compute_chips))
        }

    }
}
//function to round to n sig figs with no scientific notation (always write out number)
function round_to_n_sig_figs(x,n) {
    val = Number.parseFloat(x).toPrecision(n);
    // return val without scientific notation
    return Number(val);
}

//function that takes a number of seconds and returns a new quantity and unit that is appropriate for the number of seconds
function get_time_unit(seconds) {
    if (seconds < 60) {
        return [seconds, "seconds"];
    }
    else if (seconds < 3600) {
        return [seconds/60, "minutes"];
    }
    else if (seconds < 86400) {
        return [seconds/3600, "hours"];
    }
    else if (seconds < 604800) {
        return [seconds/86400, "days"];
    } else if (seconds < 2628000) {
        return [seconds/604800, "weeks"];
    } else if (seconds < 31536000) {
        return [seconds/2628000, "months"];
    } else if (seconds < 315360000) {
        return [seconds/31536000, "years"];
    } else if (seconds < 3153600000) {
        return [seconds/315360000, "decades"];
    } else if (seconds < 31536000000) {
        return [seconds/3153600000, "centuries"];
    } else {
        return [seconds/31536000000, "millenia"];
    }

}

    // Calculate power consumed based on time spent training and number of chips
    function get_power(time, chips) {
        // Replace this rate with actual power consumption details
        const power_rate = 0.3; // Power consumed in kW per chip per hour
        const power_consumed = compute_time * compute_chips * power_rate;
        return power_consumed;
    }

    function get_cost(time, chips) {
        // Replace this rate with actual cost details
        const cost_rate = 2; // Cost in dollars per chip per hour
        const total_cost = compute_time * compute_chips * cost_rate / 100;
        return total_cost;
    }
//

    function updatePowerAndCost(power, cost) {
        document.getElementById('power-value').innerText = power;
        document.getElementById('cost-value').innerText = cost;
        document.getElementById('power-consumed').style.display = 'block';
        document.getElementById('estimated-cost').style.display = 'block';
    }

    document.getElementById("screenshotBtn").addEventListener("click", function() {
        html2canvas(document.getElementById("table")).then(function(canvas) {
            var link = document.createElement('a');
            link.download = 'nice.png';
            link.href = canvas.toDataURL();
            link.click();
        });
    });