main.py

import UMenu
import UProblem

def main():
    level = 1
    while True:
        if level == 1:
            # Menampilkan menu 1-10
            UMenu.display_menu()
            choice = int(input("Pilih program dari menu: "))
            if choice < 1 or choice > 10:
                level = 2
            else:
                # Menjalankan program sesuai pilihan
                UProblem.run_program(choice)

        elif level == 2:
            # Menampilkan pilihan untuk kembali ke UMenu.display_menu() dengan memasukan angka 0
            # Menampilkan Pilihan untuk kembali ke UMenu.display_menu2() dengan memasukan angka 99
            UMenu.display_menu1()
            choice = int(input("Ketik 0 untuk melanjutkan atau 99 untuk ke menu selanjutnya: "))
            if choice == 0:
                level = 1
            elif choice == 99:
                level = 3
            else:
                print("Pilihan tidak valid.")

        elif level == 3:
            # Menampilkan menu 11-20
            UMenu.display_menu2()
            choice = int(input("Pilih program dari menu: "))
            if choice >= 12 and choice <= 22:
                # Menjalankan program sesuai pilihan
                UProblem.run_program(choice)
            else:
                level = 2

if __name__ == "__main__":
    main()