main.py

#write by yash-pagar

def add_numbers(x, y):
    return x + y
def subtract_numbers(x, y):
    return x - y
def multiply_numbers(x, y):
    return x * y
def divide_numbers(x, y):
    if y == 0:
        return "Error! Division by zero."
    else:
        return x / y


print("Select operation:")
print("1. Add")
print("2. Subtract")
print("3. Multiply")
print("4. Divide")
choice = input("Enter choice (1/2/3/4): ")

a = int(input("Enter the first number: "))
b = int(input("Enter the second number: "))

if choice == '1':
    print("Result:", add_numbers(a, b))
elif choice == '2':
    print("Result:", subtract_numbers(a, b))
elif choice == '3':
    print("Result:", multiply_numbers(a, b))
elif choice == '4':
    print("Result:", divide_numbers(a, b))
else:
    print("Invalid input")