Write a Java program that calculates and displays the balance of an investment over several years, using a do-while loop. The program should prompt the user for the initial investment amount.

import javax.swing.JOptionPane;

public class BankBalanceDoWhile {

    public static void main(String[] args) {
        double investmentAmount = 0.0;
        double annualInterestRate = 0.05; // Assuming a fixed annual interest rate of 5%
        int year = 0;
        int choice = 0;
        boolean validInput;

        // Step 1: Get the initial investment amount
        do {
            String input = JOptionPane.showInputDialog("Enter your initial investment amount:");
            try {
                investmentAmount = Double.parseDouble(input);
                if (investmentAmount < 0) {
                    JOptionPane.showMessageDialog(null, "Investment amount cannot be negative. Please enter a positive value.");
                    validInput = false;
                } else {
                    validInput = true;
                }
            } catch (NumberFormatException e) {
                JOptionPane.showMessageDialog(null, "Invalid input. Please enter a numeric value for the investment amount.");
                validInput = false;
            }
        } while (!validInput);

        double currentBalance = investmentAmount;

        // Step 2: Loop to display annual balances using a do-while loop
        do {
            year++;
            currentBalance *= (1 + annualInterestRate); // Calculate balance for the current year

            // Display the balance for the current year
            JOptionPane.showMessageDialog(null, String.format("Balance after year %d: R%.2f", year, currentBalance));

            // Ask the user if they wish to see the balance for another year
            do {
                String choiceInput = JOptionPane.showInputDialog(
                    "Do you wish to see the balance for another year?\n" +
                    "Enter 1 to view the next balance.\n" +
                    "Enter 0 to exit."
                );
                try {
                    choice = Integer.parseInt(choiceInput);
                    if (choice == 1 || choice == 0) {
                        validInput = true;
                    } else {
                        JOptionPane.showMessageDialog(null, "Invalid input. You entered " + choice + ". Please enter 1 to continue or 0 to exit.");
                        validInput = false;
                    }
                } catch (NumberFormatException e) {
                    JOptionPane.showMessageDialog(null, "Invalid input. Please enter an integer (1 or 0).");
                    validInput = false;
                }
            } while (!validInput);

        } while (choice == 1);

        JOptionPane.showMessageDialog(null, "Thank you for using the Bank Balance Calculator.");
    }
}