Python 16 robocampy extended

class Task:
    def __init__(self, name, deadline):
        self.name = name
        self.completed = False
        self.deadline = deadline

    def edit_name(self, new_name):
        self.name = new_name

class TaskManager:
    def __init__(self):
        self.tasks = []

    def add_task(self, name, deadline):
        task = Task(name, deadline)
        self.tasks.append(task)
        self.display_tasks()

    def remove_task(self, index):
        if 0 <= index < len(self.tasks):
            del self.tasks[index]
        self.display_tasks()

    def complete_task(self, index):
        if 0 <= index < len(self.tasks):
            self.tasks[index].completed = True
        self.display_tasks()

    def display_tasks(self, print_complete = False):
        if len(self.tasks) == 0:
            print("\nBrak zadań.")
        else:
            print("\nLista zadań:")
            for index, task in enumerate(self.tasks):
                status = "Zakończone" if task.completed else "Niezakończone"
                if print_complete and task.completed:
                    print(f"{index}. {task.name} - {status} ({task.deadline})")
                elif not print_complete and not task.completed:
                    print(f"{index}. {task.name} - {status} ({task.deadline})")

    def edit_task(self, index, new_name):
           if 0 <= index < len(self.tasks):
               self.tasks[index].edit_name(new_name)
               print("Nazwa zadania została zmieniona.")

task_manager = TaskManager()

while True:
    print("\n=============================\n")

    print("1. Dodaj nowe zadanie")
    print("2. Usuń zadanie")
    print("3. Oznacz zadanie jako zakończone")
    print("4. Wyświetl wszystkie zadania")
    print("5. Edytuj zadanie")
    print("6. Wyjdź z programu")

    print("\n=============================\n")

    choice = input("Wybierz opcję: ")

    if choice == "1":
        name = input("Podaj nazwę zadania: ")
        deadline = input("Podaj termin zadania: ")
        task_manager.add_task(name, deadline)
        print("Zadanie dodane.")

    elif choice == "2":
        index = int(input("Podaj indeks zadania do usunięcia: "))
        if index < 0 or index >= len(task_manager.tasks):
            print("Niepoprawne dane.")
            continue
        else:
            task_manager.remove_task(index)
            print("Zadanie usunięte.")

    elif choice == "3":
        index = int(input("Podaj indeks zadania do oznaczenia jako zakończone: "))
        if index < 0 or index >= len(task_manager.tasks):
            print("Niepoprawne dane.")
            continue
        else:
            task_manager.complete_task(index)
            print("Zadanie oznaczone jako zakończone.")

    elif choice == "4":
        print_complete = input("Drukować zadania zakończone (1) czy niezakończone (0)? ")
        if print_complete == "1":
            task_manager.display_tasks(True)
        else:
            task_manager.display_tasks(False)

    elif choice == "5":
        index = int(input("Podaj indeks zadania do edycji: "))
        if index < 0 or index >= len(task_manager.tasks):
            print("Niepoprawne dane.")
            continue
        else:
            new_name = input("Podaj nową nazwę zadania: ")
            task_manager.edit_task(index, new_name)

    elif choice == "6":
        break



    else:
        print("Nieznana opcja.")