import numpy as np

from flask import Flask
from flask import jsonify
from flask import render_template
from flask import request

app = Flask(__name__)


@app.route("/", methods=["GET", "POST"])
def home():
    result = None
    if request.method == "POST":
        # Get the comma-separated string, split by commas, and convert to integer
        s_data = np.array(
            [int(i) for i in request.form.get("s_data", "").split(",") if i]
        )
        ns_data = np.array(
            [int(i) for i in request.form.get("ns_data", "").split(",") if i]
        )

        total = sum(s_data) + sum(ns_data)

        # Calculate marginal probabilities
        marginal_prob_s = s_data / total
        marginal_prob_ns = ns_data / total

        # Calculate joint probabilities
        joint_prob = np.outer(marginal_prob_s, marginal_prob_ns)

        # Calculate likelihoods
        posterior_s = s_data / sum(s_data)
        posterior_ns = ns_data / sum(ns_data)

        # likelihood
        likelihood_s = marginal_prob_s
        likelihood_ns = marginal_prob_ns

        # Return results as dictionary
        result = {
            "JOINT PROBABILITY": joint_prob.tolist(),
            "MARGINAL PROBABILITY_S": marginal_prob_s.tolist(),
            "MARGINAL PROBABILITY_NS": marginal_prob_ns.tolist(),
            "LIKELIHOOD_S": likelihood_s.tolist(),
            "LIKELIHOOD_NS": likelihood_ns.tolist(),
            "POSTERIOR_S": posterior_s.tolist(),
            "POSTERIOR_NS": posterior_ns.tolist(),
        }

    return render_template("index.html", result=result)


if __name__ == "__main__":
    app.run(debug=True)