app.py

import sys

from modules.report import export_report_csv, generate_class_report_console
from modules.student import (add_student, search_student, show_all_students,
                             update_subjects_marks)
from utils.menu_decorator import heading, horizontal_line


def main():
    while True:
        heading("Smart Student Management System")
        print("1. Add student")
        print("2. Search student")
        print("3. Update subject or mark")
        print("4. Show all students")
        print("5. Generate class report")
        print("6. Export report (csv)")
        print("7. Exit")
        horizontal_line()

        option = input("\nEnter option: ").strip()
        if option == "1":
            add_student()
        elif option == "2":
            search_student()
        elif option == "3":
            update_subjects_marks()
        elif option == "4":
            show_all_students()
        elif option == "5":
            generate_class_report_console()
        elif option == "6":
            export_report_csv()
        elif option == "7":
            print("\n:) Thanks for using App! See you next time.")
            break
        else:
            print("\n:( Invalid option. Try again.")


if __name__ == "__main__":
    try:
        main()
    except KeyboardInterrupt:
        print("\n:( Keyboard Interrupted.")
        sys.exit(0)