Reset Invalid Cells (#59)

* When the complete PEtab model is checked and passes. Reset all indices

* Also resetting indices when overwriting a table

* Running model check after upload

Author: PaulJonasJost

src/petab_gui/controllers/mother_controller.py

@@ -548,6 +548,7 @@ def open_yaml_and_load_files(self, yaml_path=None, mode="overwrite"):
                 "All files opened successfully from the YAML configuration.",
                 color="green"
             )
+            self.check_model()
             # rerun the completers
             for controller in self.controllers:
                 if controller == self.sbml_controller:
@@ -604,6 +605,8 @@ def check_model(self):
             # Log the consistency check result
             if not failed:
                 self.logger.log_message("Model is consistent.", color="green")
+                for model in self.model.pandas_models.values():
+                    model.reset_invalid_cells()
             else:
                 self.logger.log_message("Model is inconsistent.", color="red")
         finally:

src/petab_gui/controllers/table_controllers.py

@@ -144,6 +144,7 @@ def open_table(self, file_path=None, separator=None, mode="overwrite"):
             self.append_df(new_df)
         elif mode == "overwrite":
             self.overwrite_df(new_df)
+            self.model.reset_invalid_cells()
 
     def overwrite_df(self, new_df: pd.DataFrame):
         # TODO: Mother controller connects to overwritten_df signal. Set df

src/petab_gui/models/pandas_table_model.py

@@ -283,6 +283,11 @@ def check_selection(self):
         rows = set([index.row() for index in selected])
         return len(rows) > 1 and len(cols) == 1, selected
 
+    def reset_invalid_cells(self):
+        """Reset the invalid cells."""
+        self._invalid_cells = set()
+        self.layoutChanged.emit()
+
 
 
 class IndexedPandasTableModel(PandasTableModel):

src/petab_gui/models/petab_model.py

@@ -62,6 +62,25 @@ def __init__(
             sbml_model=self.problem.model,
         )
 
+    @property
+    def models(self):
+        return {
+            "measurement": self.measurement,
+            "observable": self.observable,
+            "parameter": self.parameter,
+            "condition": self.condition,
+            "sbml": self.sbml,
+        }
+
+    @property
+    def pandas_models(self):
+        return {
+            "measurement": self.measurement,
+            "observable": self.observable,
+            "parameter": self.parameter,
+            "condition": self.condition,
+        }
+
     @staticmethod
     def from_petab_yaml(
         petab_yaml_path: str,