Как передать значение , выбранное пользователем ,в функцию рисования фигуры?

Задание : добавить цвет в функцию рисования фигуры; запросить у пользователя цвет фигуры посредством выбора из существующих: вывести список всех цветов с номерами и ждать ввода номера желаемого цвета. Потом нарисовать все фигуры этим цветом.

# -*- coding: utf-8 -*-
import simple_draw as sd

# словарь цветов

    colors = {
        '1': (sd.COLOR_RED, "red"),
        '2': (sd.COLOR_ORANGE, "orange"),
        '3': (sd.COLOR_YELLOW, "yellow"),
        '4': (sd.COLOR_GREEN, "green"),
        '5': (sd.COLOR_CYAN, "cyan"),
        '6': (sd.COLOR_BLUE, "blue"),
        '7': (sd.COLOR_PURPLE, "purple")
    }

# запрос ввести номер цвета и проверка на корректность введённого числа

    print('Возможные цвета:')
    for key in color_number.keys():
        print(key, ':', color_number[key])
    
    user_input = input("Введите желаемый цвет:")
    color = int(user_input)
    if color in color_number:
        print('Введённый цвет', color_number[color])
    elif color >= 7:
        print('Введённый номер не корректен')
    elif color <= 0:
        print('Введённый номер не корректен')

# функция рисования и ,в качестве примера, рисование треугольника

    color_1 = color
    
    
    def figure(point, length, angle, quantity_of_side):
        delta_angle = 360 // quantity_of_side
        point_1 = point
        for v in range(quantity_of_side - 1):
            angle += delta_angle
            v = sd.get_vector(start_point=point, angle=angle, length=length, width=3)
            v.draw(color=color_1)
            point = v.end_point
    
        sd.line(start_point=point, end_point=point_1, color=color_1, width=3)
    
    
    figure(point=sd.get_point(-100, 100), length=100, angle=0, quantity_of_side=6)

# 1.ТРЕУГОЛЬНИК

    def triangle(point, angle, length):
        figure(point, length, angle, quantity_of_side=3)
    
    
    triangle(point=sd.get_point(200, 100), angle=0, length=100)
    sd.pause()


# Вывод должен быть вида:

    1:red
    2:orange
    3:yellow
    4:cean
    5:blue
    6:purple
    Введите желаемый цвет:

# фигуру рисует-но постоянно чёрным цветом

как прописать чтоб выбранный цвет передавался в функцию рисования??