Алгоритм Прима выдает ошибку нужно сделать ввод данных с консоли

import math
 
def get_min(R, U):
    rm = (math.inf, -1, -1)
    for v in U:
        rr = min(R, key=lambda x: x[0] if (x[1] == v or x[2] == v) and (x[1] not in U or x[2] not in U) else math.inf)
        if rm[0] > rr[0]:
            rm = rr
 
    return rm
 
# список ребер графа (длина, вершина 1, вершина 2)
# первое значение возвращается, если нет минимальных ребер
R = [(math.inf, -1, -1)]
R.append(int(input())
N =int(input("Введите число вершин: "))   # число вершин в графе
U = {1}   # множество соединенных вершин
T = []    # список ребер остова
 
while len(U) < N:
    r = get_min(R, U)       # ребро с минимальным весом
    if r[0] == math.inf:    # если ребер нет, то остов построен
        break
 
    T.append(r)             # добавляем ребро в остов
    U.add(r[1])             # добавляем вершины в множество U
    U.add(r[2])
 
print(T)

import math
 
def get_min(R, U):
    rm = (math.inf, -1, -1)
    for v in U:
        rr = min(R, key=lambda x: x[0] if (x[1] == v or x[2] == v) and (x[1] not in U or x[2] not in U) else math.inf)
        if rm[0] > rr[0]:
            rm = rr
    return rm
 
# список ребер графа (длина, вершина 1, вершина 2)
# первое значение возвращается, если нет минимальных ребер
R = [(math.inf, -1, -1), (13, 1, 2), (18, 1, 3), (17, 1, 4), (14, 1, 5), (22, 1, 6),
     (26, 2, 3), (19, 2, 5), (30, 3, 4), (22, 4, 6)]
 
N = 6     # число вершин в графе
U = {1}   # множество соединенных вершин
T = []    # список ребер остова
 
while len(U) < N:
    r = get_min(R, U)       # ребро с минимальным весом
    if r[0] == math.inf:    # если ребер нет, то остов построен
        break
 
    T.append(r)             # добавляем ребро в остов
    U.add(r[1])             # добавляем вершины в множество U
    U.add(r[2])
 
print(T)

R = [(math.inf, -1, -1)]
N =int(input("Введите число вершин: "))   # число вершин в графе
M =int(input("Введите число ребер: "))   # число ребер в графе
for _ in range(M):
    R.append(tuple(map(int, input("Введите ребро: длина вершина1 вершина2:").split())))

vertices = set()
weight_of = dict()
n = int(input('Число ребер графа n = '))
for i in range(n):
    while True:
        print()
        print(f'Ребро {i + 1}:')
        v1 = input('\tвершина 1: ')
        v2 = input('\tвершина 2: ')
        if frozenset([v1, v2]) in weight_of:
            print('\tРебро с такими вершинами уже задано!')
        else:
            break
    w  = int(input('\tстоимость: '))
    weight_of[frozenset([v1, v2])] = w
    vertices.update([v1, v2])
 
tree_vertices = set()
tree_vertices.add( vertices.pop() )
 
def get_min_edge_vertices(tree_vertices, vertices):
    res_vertices = None
    if vertices:
        for A in tree_vertices:
            for B in vertices:
                cur_vertices = frozenset([A, B])
                if cur_vertices not in weight_of:
                    continue
                if not res_vertices or weight_of[cur_vertices] < weight_of[res_vertices]:
                    res_vertices = cur_vertices
        if res_vertices:
            tree_vertices.update(res_vertices)
            vertices -= res_vertices
            return set(res_vertices)
 
min_tree = []
while vertices:
    min_edge_vertices = get_min_edge_vertices(tree_vertices, vertices)
    if not min_edge_vertices:
        print()
        print('Граф несвязный.')
        break
    min_tree.append(min_edge_vertices)
else:
    print()
    print(min_tree)

vertices = set()
weight_of = dict()
n = int(input('Число ребер графа n = '))
for i in range(n):
    while True:
        print()
        print(f'Ребро {i + 1}:')
        v1 = input('\tвершина 1: ')
        v2 = input('\tвершина 2: ')
        if frozenset([v1, v2]) in weight_of:
            print('\tРебро с такими вершинами уже задано!')
        else:
            break
    w  = int(input('\tстоимость: '))
    weight_of[frozenset([v1, v2])] = w
    vertices.update([v1, v2])
 
tree_vertices = set()
tree_vertices.add( vertices.pop() )
edges = [[list(key), val] for key, val in weight_of.items() ]
edges.sort(key=lambda x: x[1])
 
def get_min_edge_vertices(tree_vertices, vertices):
    for edge in edges:
        if (
                edge[0][0] in tree_vertices
                and edge[0][1] in vertices
            or  edge[0][1] in tree_vertices
                and edge[0][0] in vertices
            ):
            res_edge_vertices = edge[0]
            tree_vertices.update(res_edge_vertices)
            vertices -= set(res_edge_vertices)
            return ''.join(sorted(res_edge_vertices))
 
min_tree = []
while vertices:
    min_edge_vertices = get_min_edge_vertices(tree_vertices, vertices)
    if not min_edge_vertices:
        print()
        print('Граф несвязный.')
        break
    min_tree.append(min_edge_vertices)
else:
    print()
    print(*min_tree)