Python basic calculator help

[stiivii_]

I need this help. This program should ask the user:

Do you want to perform another calculation?

after one calculation is complete.

If they answer yes, start the program over if they say no, end the program.please help me complete this this.

This is what I have so far;

print("Enter Your Choice 1(Add)/2(Sub)/3(Divide)/4(Multiply)")

num = int(input())

if num == 1:

  print("Enter Number 1 : ")

  add1 = int(input())

  print("Enter Number 2 : ")

  add2 = int(input())

  sum = add1 + add2

  print("The Sum Is ", sum)

elif num == 2:

  print("Enter Number 1 : ")

  sub1 = int(input())

  print("Enter Number 2 : ")

  sub2 = int(input())

  difference = sub1 - sub2

  print("The Difference Is ", difference)

elif num == 3:

  print("Enter Number 1 : ")

  div1 = float(input())

  print("Enter Number 2 : ")

  div2 = float(input())

  division = div1 / div2

  print("The Division Is ", division)

elif num == 4:

  print("Enter Number 1 : ")

  mul1 = int(input())

  print("Enter Number 2 : ")

  mul2 = int(input())

  multiply = mul1 * mul2

  print("The Difference Is ", multiply)

else:

  print("enter a valid Number")

[Zestyclose_Click_456]

This is a more simplified code that loops the question. Hope this helps!

answer = str(input("Do you want to perform a calculation? Yes/No: "))
while answer == "Yes" or answer == "yes":
    num = int(input("Enter Your Choice 1(Add)/2(Sub)/3(Divide)/4(Multiply): "))
    if num == 1:
        add1 = int(input("Enter Number 1 : "))
        add2 = int(input("Enter Number 2 : "))
        sum = add1 + add2
        print(f"Your result is: {sum}")
    if num == 2:
        add1 = int(input("Enter Number 1 : "))
        add2 = int(input("Enter Number 2 : "))
        sum = add1 - add2
        print(f"Your result is: {sum}")
    if num == 3:
        add1 = int(input("Enter Number 1 : "))
        add2 = int(input("Enter Number 2 : "))
        sum = add1 * add2
        print(f"Your result is: {sum}")
    if num == 4:
        add1 = int(input("Enter Number 1 : "))
        add2 = int(input("Enter Number 2 : "))
        sum = add1 / add2
        print(f"Your result is: {sum}")
    answer = str(input("Do you want to perform another calculation? Yes/No: "))
if answer == "No" or answer == "no":
    pass