routers/ai/src/main.py

from .math_operations import add, subtract
from .shapes import Circle, Rectangle

def main():
    result = add(2, 3)
    print(f"2 + 3 = {result}")

    circle = Circle(radius=5)
    print(f"Circle area: {circle.area()}")

    rectangle = Rectangle(width=4, height=5)
    print(f"Rectangle area: {rectangle.area()}")

if __name__ == "__main__":
    main()