diff --git a/lib/pages/map_page.dart b/lib/pages/map_page.dart index 1561c47..5a779c7 100644 --- a/lib/pages/map_page.dart +++ b/lib/pages/map_page.dart @@ -22,14 +22,28 @@ class MapPage extends StatefulWidget { class _MapPageState extends State { final _mapController = MapController(); late FloodStationProvider _floodStationProvider; + bool _loading = false; @override Widget build(BuildContext context) { _floodStationProvider = context.watch(); - if (_floodStationProvider.allStations.isEmpty) { + if (_loading == true) { + return Center( + child: CircularProgressIndicator(), + ); + } else if (_floodStationProvider.allStations.isEmpty) { return Center( child: ElevatedButton( - onPressed: _floodStationProvider.loadAllStations, + onPressed: () { + setState(() { + _loading = true; + }); + _floodStationProvider + .loadAllStations() + .whenComplete(() => setState(() { + _loading = false; + })); + }, child: Text('Load Map'), ), );