class CalculatorActivity : AppCompatActivity() {
    private lateinit var binding: ActivityCalculatorBinding
    private var isGenderSelected = false
    private var previousSelectedPosition = 0
    private var selectedGenderValue = 0 // 0 for Female, 1 for Male

    override fun onCreate(savedInstanceState: Bundle?) {
        super.onCreate(savedInstanceState)
        binding = ActivityCalculatorBinding.inflate(layoutInflater)
        setContentView(binding.root)

        val textInputEditTextAge = binding.tedAge
        val placeholderAge = "... Years"
        textInputEditTextAge.inputType = InputType.TYPE_CLASS_NUMBER
        textInputEditTextAge.hint = placeholderAge

        val textInputEditTextWeight = binding.tedWeight
        val placeholderWeight = "... KG"
        textInputEditTextWeight.inputType = InputType.TYPE_CLASS_NUMBER
        textInputEditTextWeight.hint = placeholderWeight

        val textInputEditTextHeight = binding.tedHeight
        val placeholderHeight = "... CM"
        textInputEditTextHeight.inputType = InputType.TYPE_CLASS_NUMBER
        textInputEditTextHeight.hint = placeholderHeight

        val textInputEditTextHours = binding.tedHours
        val placeholderHours = "... Hours"
        textInputEditTextHours.inputType = InputType.TYPE_CLASS_NUMBER
        textInputEditTextHours.hint = placeholderHours

        val textInputEditTextYears = binding.tedYears
        val placeholderYears = "... Years"
        textInputEditTextYears.inputType = InputType.TYPE_CLASS_NUMBER
        textInputEditTextYears.hint = placeholderYears

        val gender = resources.getStringArray(R.array.Gender)

        val spinner = binding.genderSpinner
        if (spinner != null) {
            val adapter = ArrayAdapter(this,
                android.R.layout.simple_spinner_dropdown_item, gender)
            spinner.adapter = adapter

            spinner.onItemSelectedListener = object :
                AdapterView.OnItemSelectedListener {
                override fun onItemSelected(parent: AdapterView<*>, view: View, position: Int, id: Long) {
                    if (isGenderSelected) {
                        // A gender other than "-- Select your Gender --" has already been selected
                        if (position == 0) {
                            spinner.setSelection(previousSelectedPosition) // Set the spinner to the previous selected position
                        } else {
                            previousSelectedPosition = position // Update the previous selected position
                            selectedGenderValue = if (position == 1) 1 else 0 // Convert position to numeric value (0 or 1)
                        }
                    } else {
                        isGenderSelected = true
                        previousSelectedPosition = position // Set the initial selected position
                        selectedGenderValue = if (position == 1) 1 else 0 // Convert position to numeric value (0 or 1)
                    }
                }

                override fun onNothingSelected(parent: AdapterView<*>) {
                    // write code to perform some action
                }
            }
        }

        binding.btnPredict.setOnClickListener {
            val gender = selectedGenderValue.toFloat()
            val age = textInputEditTextAge.text.toString().toFloatOrNull()
            val weight = textInputEditTextWeight.text.toString().toFloatOrNull()
            val height = textInputEditTextHeight.text.toString().toFloatOrNull()
            val hours = textInputEditTextHours.text.toString().toFloatOrNull()
            val years = textInputEditTextYears.text.toString().toFloatOrNull()
            val bmi = (weight?.div(((height?.div(100))?.times((height?.div(100)!!))!!)))
            if (gender !=null && age != null && weight != null && height != null && hours != null && years != null && bmi != null){
                val ortEnvironment = OrtEnvironment.getEnvironment()
                val ortSession = createORTSession(ortEnvironment)
                val output = runPrediction(
                    gender, age, weight, height, hours, years, bmi,
                    ortSession, ortEnvironment
                )
                showOutputPopup(output)
            } else {
                Toast.makeText(this, "Please fill in all the inputs", Toast.LENGTH_LONG).show()
            }
        }
    }

    private fun createORTSession( ortEnvironment: OrtEnvironment) : OrtSession {
        val modelBytes = resources.openRawResource( R.raw.model ).readBytes()
        return ortEnvironment.createSession( modelBytes )
    }

    private fun runPrediction( genders: Float , age: Float , weight: Float , height: Float , hours: Float , years: Float , bmi : Float , ortSession: OrtSession , ortEnvironment: OrtEnvironment ) : Float {
        // Get the name of the input node
        val inputName = ortSession.inputNames?.iterator()?.next()
        // Make a FloatBuffer of the inputs
        val floatBufferInputs = FloatBuffer.wrap( floatArrayOf( genders, age, weight, height, hours, years, bmi ) )
        // Create input tensor with floatBufferInputs of shape ( 1 , 1 )
        val inputTensor = OnnxTensor.createTensor( ortEnvironment , floatBufferInputs , longArrayOf( 1, 1 ) )
        // Run the model
        val results = ortSession.run( mapOf( inputName to inputTensor ) )
        // Fetch and return the results
        val output = results[0].value as Array<FloatArray>
        return output[0][0]
    }

    fun showOutputPopup(output: Float) {
        // Inflate the custom layout for the popup
        val inflater = layoutInflater
        val popupView = inflater.inflate(R.layout.popup_output, null)

        // Find views within the custom layout
        val tvOutput = popupView.findViewById<TextView>(R.id.tvOutput)
        val btnClose = popupView.findViewById<Button>(R.id.btnClose)

        // Set the output text
        tvOutput.text = "Output is $output"

        // Create the dialog builder
        val builder = AlertDialog.Builder(this)
        builder.setView(popupView)

        // Create and show the dialog
        val dialog = builder.create()
        dialog.show()

        // Handle button click
        btnClose.setOnClickListener {
            dialog.dismiss() // Close the dialog when the button is clicked
        }
    }
}