v3

main.py

@@ -23,21 +23,26 @@ def remainder(number1, number2):
     return number1 % number2
 
 
-# Input values
-numb1 = int(input("Enter the first number"))
-numb2 = int(input("Enter the second number"))
-operator = input("Enter the operator")
-
-# Calculate
-if operator == '+':
+while 1:
+    # Input values
+    numb1 = int(input("Enter the first number"))
+    numb2 = int(input("Enter the second number"))
+    operator = input("Enter the operator")
+
+    # Calculate
+    if operator == '+':
         print("Sum : ", add(numb1, numb2))
-elif operator == '-':
+    elif operator == '-':
         print("Subtract : ", subtract(numb1, numb2))
-elif operator == '*':
+    elif operator == '*':
         print("Product : ", multiply(numb1, numb2))
-elif operator == '/':
+    elif operator == '/':
         print("Division : ", divide(numb1, numb2))
-elif operator == '%':
+    elif operator == '%':
         print("Remainder : ", remainder(numb1, numb2))
-else:
+    else:
         print("Invalid Operator")
+
+    print("Do you want to do another calculation (1/0)")
+    if int(input()):
+        break