Module area

# CREATING MODULE IN A NEW FILE (mymodule) AND THEN IMPORTING IT IN MAIN FILE
# Importing the module created in the new file
import mymodule
print("On which geometric shape would you like to operate on?")
print("1. Rectangle")
print("2. Triangle")
print("3. Circle")
inp = int(input("Enter your choice (1-3): "))
if inp == 1:
  a = int(input('Enter the value of length: '))
  b = int(input('Enter the value of breadth: '))
  ans = mymodule.arearectangle(a, b)
  print('The area of Rectangle is: ', ans, 'units')
elif inp == 2:
  a = int(input('Enter the value of base: '))
  b = int(input('Enter the value of height: '))
  ans = mymodule.areatriangle(a, b)
  print('The area of the Triangle is: ', ans, 'units')
elif inp == 3:
  r = int(input('Enter the value of radius: '))
  ans = mymodule.areacircle(r)
  print('The area of the circle is: ', ans, 'units')
else:
  print('Invalid input')