Как изменить переменную в основной функции во Flet? ⇐ Python

[Anonymous]

В начале программы я создаю случайный список узлов и линий между ними, чтобы представить граф:

Код: Выделить всё

import flet as ft
import flet.canvas as cv
import random
import numpy as np

def generateNodes(n):
a = []
for _ in range(n):
a.append(ft.Offset(random.randint(100, 400), random.randint(100, 400)))
return a

def generateEdges(nodes):
e = []
sum = 0
for i in range(len(nodes)):
for j in range(i + 1, len(nodes)):
e.extend([nodes[i], nodes[j]])
d = np.sqrt((nodes[i].x - nodes[j].x) ** 2 + (nodes[i].y - nodes[j].y) ** 2)
sum += d
return (e, d)

В моей основной функции есть Canvas и кнопка. При нажатии кнопки я хочу поменять местами два случайных узла и снова отобразить график.

Код: Выделить всё

def swap(a, i, j):
a[i], a[j] = a[j], a[i]

def main(page: ft.Page):
a = generateNodes(4)
edges, totalDistance = generateEdges(a)

page.title = "Example App"
page.window.width = 500
page.window.height = 500
page.window.resizable = False

page.vertical_alignment = ft.MainAxisAlignment.CENTER
page.horizontal_alignment = ft.CrossAxisAlignment.CENTER
node_paint = ft.Paint(
stroke_width=8, style=ft.PaintingStyle.STROKE, color=ft.colors.LIGHT_BLUE
)
edge_paint = ft.Paint(stroke_width=4, style=ft.PaintingStyle.STROKE)

cvs = cv.Canvas(
[
cv.Points(edges, point_mode=cv.PointMode.LINES, paint=edge_paint),
cv.Points(a, point_mode=cv.PointMode.POINTS, paint=node_paint),
],
width=500,
height=500,
expand=True,
)

title_text = ft.Text(
value=totalDistance, size=20, weight=ft.FontWeight.BOLD, text_align="CENTER"
)

def swap_clicked(e):
swap(a, random.randint(0, 3), random.randint(0, 3))
edges2, totalDistance2 = generateEdges(a)
title_text.value = totalDistance2
page.update()

page.add(cvs)
page.add(title_text)
page.add(ft.ElevatedButton(text="Swap", on_click=swap_clicked))

ft.app(main)

Мне удалось изменить totalDistance после того, как я поменял узлы, но не могу понять, как снова перерисовать график.
Может ли кто-нибудь помочь? Заранее спасибо.

Подробнее здесь: https://stackoverflow.com/questions/790 ... on-in-flet