Update start/end when splitting scene

GUI/ViewModel/SceneItem.py

@@ -16,7 +16,7 @@ def __init__(self, scene : SubtitleScene):
         self.scene_model = {
             'scene': scene.number,
             'start': scene.batches[0].txt_start,
-            'end': scene.batches[-1].srt_end,
+            'end': scene.batches[-1].txt_end,
             'duration': None,
             'gap': None,
             'summary': scene.summary
@@ -93,6 +93,7 @@ def AddBatchItem(self, batch_item : BatchItem):
             self.insertRow(batch_item.number - 1, batch_item)
 
         self.Remap()
+        self.UpdateStartAndEnd()
 
     def Update(self, update):
         """ Update the scene model with new data """
@@ -101,6 +102,21 @@ def Update(self, update):
 
         UpdateFields(self.scene_model, update, ['summary', 'start', 'end', 'duration', 'gap'])
 
+    def UpdateStartAndEnd(self):
+        """ Update the start and end times of the scene """
+        start = None
+        end = None
+        for i in range(0, self.rowCount()):
+            batch_item = self.child(i, 0)
+            if i == 0:
+                start = batch_item.start
+            end = batch_item.end
+
+        if start:
+            self.scene_model['start'] = start
+        if end:
+            self.scene_model['end'] = end
+
     def Remap(self):
         """ Rebuild the batch number map """
         batch_items = {}

GUI/ViewModel/ViewModel.py

@@ -328,6 +328,9 @@ def UpdateBatch(self, scene_number, batch_number, batch_update : dict):
     def RemoveBatch(self, scene_number, batch_number):
         logging.debug(f"Removing batch ({scene_number}, {batch_number})")
         scene_item : SceneItem = self.model.get(scene_number)
+        if not scene_item:
+            raise ViewModelError(f"Scene {scene_number} not found")
+
         if batch_number not in scene_item.batches.keys():
             raise ViewModelError(f"Scene {scene_number} batch {batch_number} does not exist")
 
@@ -342,6 +345,8 @@ def RemoveBatch(self, scene_number, batch_number):
                 logging.debug(f"Removed row {i} from scene {scene_item.number}, rowCount={scene_item.rowCount()}")
                 break
 
+        scene_item.UpdateStartAndEnd()
+
     #############################################################################
 
     def AddLine(self, scene_number, batch_number, line : SubtitleLine):