aljaz/Auto-GPT
1
0
Fork 0
mirror of https://github.com/aljazceru/Auto-GPT.git synced 2025-12-19 15:04:26 +01:00
Code Issues Packages Projects Releases Wiki Activity

Enhance SkillTreeView with Asynchronous Initialization and Node Mapping

Browse Source 
This commit refactors the SkillTreeView class to include asynchronous initialization through FutureBuilder. The new version also replaces the integer-based node IDs with string-based IDs, aligning better with the SkillTreeNode model. The code now clears previous graph nodes and edges before adding new ones, preventing duplication. Additionally, the TreeNodeView component is now populated dynamically with data from the SkillTreeNode model, making the tree view more robust and integrated.
This commit is contained in:
hunteraraujo
2023-09-13 17:50:53 -07:00
parent 01b2892775
commit a7bb335752
1 changed files with 71 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
frontend/lib/views/skill_tree/skill_tree_view.dart
View File

@@ -1,3 +1,4 @@
import 'package:auto_gpt_flutter_client/models/skill_tree/skill_tree_node.dart';
import 'package:auto_gpt_flutter_client/viewmodels/skill_tree_viewmodel.dart'; import 'package:auto_gpt_flutter_client/viewmodels/skill_tree_viewmodel.dart';
import 'package:auto_gpt_flutter_client/views/skill_tree/tree_node_view.dart'; import 'package:auto_gpt_flutter_client/views/skill_tree/tree_node_view.dart';
import 'package:flutter/material.dart'; import 'package:flutter/material.dart';
@@ -9,18 +10,35 @@ class SkillTreeView extends StatefulWidget {
const SkillTreeView({Key? key, required this.viewModel}) : super(key: key); const SkillTreeView({Key? key, required this.viewModel}) : super(key: key);
@override @override
_TreeViewPageState createState() => _TreeViewPageState(); _SkillTreeViewState createState() => _SkillTreeViewState();
} }
class _TreeViewPageState extends State<SkillTreeView> { class _SkillTreeViewState extends State<SkillTreeView> {
Future<void>? initialization;
@override @override
void initState() { void initState() {
super.initState(); super.initState();
initialization = widget.viewModel.initializeSkillTree();
}
widget.viewModel.initializeSkillTree(); @override
Widget build(BuildContext context) {
return FutureBuilder<void>(
future: initialization,
builder: (context, snapshot) {
widget.viewModel.graph.nodes.clear();
widget.viewModel.graph.edges.clear();
if (snapshot.connectionState == ConnectionState.waiting) {
return const CircularProgressIndicator();
}
if (snapshot.hasError) {
return const Text("An error occurred");
}
// Create Node and Edge objects for GraphView // Create Node and Edge objects for GraphView
final Map<int, Node> nodeMap = {}; final Map<String, Node> nodeMap = {};
for (var skillTreeNode in widget.viewModel.skillTreeNodes) { for (var skillTreeNode in widget.viewModel.skillTreeNodes) {
final node = Node.Id(skillTreeNode.id); final node = Node.Id(skillTreeNode.id);
widget.viewModel.graph.addNode(node); widget.viewModel.graph.addNode(node);
@@ -28,14 +46,13 @@ class _TreeViewPageState extends State<SkillTreeView> {
} }
for (var skillTreeEdge in widget.viewModel.skillTreeEdges) { for (var skillTreeEdge in widget.viewModel.skillTreeEdges) {
final fromNode = nodeMap[int.parse(skillTreeEdge.from)]; final fromNode = nodeMap[skillTreeEdge.from];
final toNode = nodeMap[int.parse(skillTreeEdge.to)]; final toNode = nodeMap[skillTreeEdge.to];
widget.viewModel.graph.addEdge(fromNode!, toNode!); if (fromNode != null && toNode != null) {
widget.viewModel.graph.addEdge(fromNode, toNode);
} }
} }
@override
Widget build(BuildContext context) {
return Scaffold( return Scaffold(
body: Column( body: Column(
mainAxisSize: MainAxisSize.max, mainAxisSize: MainAxisSize.max,
@@ -43,7 +60,7 @@ class _TreeViewPageState extends State<SkillTreeView> {
Expanded( Expanded(
child: InteractiveViewer( child: InteractiveViewer(
constrained: false, constrained: false,
boundaryMargin: EdgeInsets.all(100), boundaryMargin: const EdgeInsets.all(100),
minScale: 0.01, minScale: 0.01,
maxScale: 5.6, maxScale: 5.6,
child: GraphView( child: GraphView(
@@ -55,10 +72,17 @@ class _TreeViewPageState extends State<SkillTreeView> {
..strokeWidth = 1 ..strokeWidth = 1
..style = PaintingStyle.stroke, ..style = PaintingStyle.stroke,
builder: (Node node) { builder: (Node node) {
int nodeId = node.key?.value as int; String nodeId = node.key?.value as String;
SkillTreeNode? skillTreeNode =
widget.viewModel.getNodeById(nodeId);
if (skillTreeNode != null) {
return TreeNodeView( return TreeNodeView(
nodeId: nodeId, node: skillTreeNode,
selected: nodeId == widget.viewModel.selectedNode?.id); selected:
nodeId == widget.viewModel.selectedNode?.id);
} else {
return const SizedBox(); // Return an empty widget if the node is not found
}
}, },
), ),
), ),
@@ -66,5 +90,7 @@ class _TreeViewPageState extends State<SkillTreeView> {
], ],
), ),
); );
},
);
} }
} }