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 == '+':
-    print("Sum : ", add(numb1, numb2))
-elif operator == '-':
-    print("Subtract : ", subtract(numb1, numb2))
-elif operator == '*':
-    print("Product : ", multiply(numb1, numb2))
-elif operator == '/':
-    print("Division : ", divide(numb1, numb2))
-elif operator == '%':
-    print("Remainder : ", remainder(numb1, numb2))
-else:
-    print("Invalid 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 == '-':
+        print("Subtract : ", subtract(numb1, numb2))
+    elif operator == '*':
+        print("Product : ", multiply(numb1, numb2))
+    elif operator == '/':
+        print("Division : ", divide(numb1, numb2))
+    elif operator == '%':
+        print("Remainder : ", remainder(numb1, numb2))
+    else:
+        print("Invalid Operator")
+
+    print("Do you want to do another calculation (1/0)")
+    if int(input()):
+        break