From 546966ccfa600bc9f3d0fd098d9f9985eb85150d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C3=81lvaro=20Gonz=C3=A1lez?= <alvaro@asus>
Date: Fri, 6 Mar 2026 08:52:23 +0100
Subject: [PATCH] The formula is updated when edited and saved

---
 lib/ai/formula_list.dart   | 5 +++++
 lib/ai/formula_screen.dart | 5 +++--
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/lib/ai/formula_list.dart b/lib/ai/formula_list.dart
index 9689384..a9648e6 100644
--- a/lib/ai/formula_list.dart
+++ b/lib/ai/formula_list.dart
@@ -81,6 +81,11 @@ class _FormulaListState extends State<FormulaList> {
         builder: (context) => FormulaEditor(
           formula: formula,
           corpus: widget.corpus,
+          onSave: (updatedFormula){
+            setState((){
+              // THIS UPDATES THE FORMULA LIST
+            });
+          }
         ),
       ),
     );
diff --git a/lib/ai/formula_screen.dart b/lib/ai/formula_screen.dart
index 56758b6..9e92252 100644
--- a/lib/ai/formula_screen.dart
+++ b/lib/ai/formula_screen.dart
@@ -11,10 +11,10 @@ import 'unit_dropdown.dart';
 import 'formula_editor.dart';
 
 class FormulaScreen extends StatefulWidget {
-  final Formula formula;
+  Formula formula;
   final Corpus corpus;
 
-  const FormulaScreen({super.key, required this.formula, required this.corpus});
+  FormulaScreen({super.key, required this.formula, required this.corpus});
 
   @override
   State<FormulaScreen> createState() => _FormulaScreenState();
@@ -218,6 +218,7 @@ class _FormulaScreenState extends State<FormulaScreen> {
                       // Refresh the screen after saving
                       setState(() {
                         // The corpus has been updated, refresh the displayed formula
+                        widget.formula = updatedFormula;
                       });
                     },
                   ),