Remove frame works

2 files changed, 11 insertions, 1 deletions

diff --git a/activity.py b/activity.py
index ff40855..6388ebd 100644
--- a/activity.py
+++ b/activity.py
@@ -137,7 +137,7 @@ class AnimateActivity(activity.Activity):
             self._toolbarbox.set_fps(options['framerate'])
 
     def _remove_frame(self, widget):
-        self._frames_list.remove_selected_frame()
+        self._frames_tray.remove_selected_frame()
         self._animation.remove_current_frame()
 
     def _add_frame(self, widget):
diff --git a/frames_tray.py b/frames_tray.py
index f048b4d..097e764 100644
--- a/frames_tray.py
+++ b/frames_tray.py
@@ -53,6 +53,16 @@ class FramesTray(HTray):
             item.set_group(self._group)
         self.add_item(item)
 
+    def get_selected_frame(self):
+        for button in self.get_children():
+            if button.get_active():
+                return button
+
+    def remove_selected_frame(self):
+        button = self.get_selected_frame()
+        self.remove_item(button)
+        del(button)
+
     def remove_frame(self):
         return