some fixes to event negate

Author: yeshaokai

amadeusgpt/analysis_objects/event.py

@@ -163,20 +163,20 @@ def event_negate(cls, events: List[BaseEvent]) -> List[BaseEvent]:
         Get list of events that are negates of those events
         The core attributes must stay the same
         """
-        # mask = np.zeros_like(events[0].mask, dtype=bool)
         mask = np.zeros(events[0].data_length, dtype=bool)
         sender_animal_name = events[0].sender_animal_name
         for event in events:
             assert sender_animal_name == event.sender_animal_name
         video_file_path = events[0].video_file_path
         for event in events:
-            mask |= event.mask
+            mask |= event.generate_mask()
         negate_mask = ~mask
 
-        start, end = cls.get_start_end_from_mask(negate_mask)
+        
 
         negate_events = Event.mask2events(
-            start, end, video_file_path, sender_animal_name
+            negate_mask, video_file_path, sender_animal_name,
+            set(), set(), smooth_window_size = 1
         )
 
         return negate_events

amadeusgpt/managers/object_manager.py

@@ -80,9 +80,7 @@ def add_roi_object(self, object) -> None:
         self.roi_objects = self.filter_duplicates(self.roi_objects)
 
     def save_roi_objects(self, path: str) -> None:
-        roi_obects = self.get_roi_objects()
-        print("how many roi objects")
-        print(roi_obects)
+        roi_obects = self.get_roi_objects()       
         for roi in roi_obects:
             print(roi.name)
         data = {}

amadeusgpt/programs/sandbox.py

@@ -399,7 +399,7 @@ def render_qa_message(self, qa_message):
 
             else:
                 for e in function_rets:
-                    if isinstance(e, list) and isinstance(e[0], BaseEvent):
+                    if isinstance(e, list) and len(e) > 0 and isinstance(e[0], BaseEvent):
                         # here we need to understand what we do with the events
                         # we have ethogram plot, keypoint plot, head orientation plot, scene plot
                         # and animal interaction plot