updated example.json for mabe as several things change in the codebase

Author: yeshaokai

amadeusgpt/app_utils.py

@@ -217,8 +217,6 @@ def render(self):
                         filename = save_figure_to_tempfile(fig)
                         st.image(filename, width=600)
                 elif render_key == "out_videos":
-                    print ('out_videos')
-                    print (render_value)
                     for video_path in render_value:
                         if os.path.exists(video_path):
                             st.video(video_path)

amadeusgpt/managers/event_manager.py

@@ -233,6 +233,7 @@ def get_events_from_relationship(
 
         mask = relationship.query_relationship(relation_query)
 
+        #print ('mask', mask)
         # determine whether the mask is a numpy of float or numpy of boolean
 
         if mask.dtype != bool:
@@ -323,7 +324,7 @@ def get_animals_animals_events(
                 _comparison = comparison_operator + "".join(
                     query.split(comparison_operator)[1:]
                 )
-               
+             
                 events = self.get_events_from_relationship(
                     relationship, _query, _comparison, smooth_window_size
                 )

amadeusgpt/programs/sandbox.py

@@ -549,4 +549,66 @@ def render_temp_message(query, sandbox):
     amadeus = create_amadeus(config)
     sandbox = amadeus.sandbox
 
-    render_temp_message('random', sandbox)
+    def get_chases_events(config: Config):
+        '''
+        Parameters:
+        ----------
+        config: Config
+        '''
+        # create_analysis returns an instance of AnimalBehaviorAnalysis
+        analysis = create_analysis(config)
+
+        # Get events where the closest distance between animals is less than 40 pixels
+        closest_distance_events = analysis.get_animals_animals_events(
+            cross_animal_query_list=['closest_distance<40'],
+            bodypart_names=None,
+            otheranimal_bodypart_names=None,
+            min_window=1,
+            max_window=100000
+        )
+
+        # Get events where the angle between animals is less than 30 degrees
+        angle_events = analysis.get_animals_animals_events(
+            cross_animal_query_list=['relative_angle<30'],
+            bodypart_names=None,
+            otheranimal_bodypart_names=None,
+            min_window=1,
+            max_window=100000
+        )
+
+        print ('angle_events')
+        for event in angle_events:
+            print (event)
+        return angle_events
+
+        # Get events where the animal's speed is greater than 0.2
+        speed_events = analysis.get_animals_state_events(
+            query='speed>0.2',
+            bodypart_names=None,
+            min_window=1,
+            max_window=100000
+        )
+
+        # Combine the closest distance and angle events using logical AND
+        distance_angle_events = analysis.get_composite_events(
+            events_A=closest_distance_events,
+            events_B=angle_events,
+            composition_type='logical_and',
+            max_interval_between_sequential_events=0,
+            min_window=1,
+            max_window=100000
+        )
+
+        # Combine the result with the speed events using logical AND
+        chases_events = analysis.get_composite_events(
+            events_A=distance_angle_events,
+            events_B=speed_events,
+            composition_type='logical_and',
+            max_interval_between_sequential_events=0,
+            min_window=1,
+            max_window=100000
+        )
+
+        return chases_events
+
+    get_chases_events(config)

examples/MABe/example.json

@@ -2,7 +2,7 @@
     {
         "role": "human",
         "error": NaN,
-        "query": "Define <|chases|> as a social behavior where closest distance between this animal and other animals is less than 40 pixels and the angle between this and other animals have to be less than 30 and this animal has to travel faster than 0.2."
+        "query": "Define chases as a social behavior where closest distance between this animal and other animals is less than 40 pixels and the head angle between this and other animals have to be less than 30 and this animal has to travel faster than 0.1."
     },
     {
         "role": "human",