| 273 | return cv2.cvtColor(background, cv2.COLOR_BGR2RGB) |
| 274 | |
| 275 | def return_format(self, messages: List[Message]): |
| 276 | _format = [{"message": "", "sender": idx} for idx in range(len(self.agent_id))] |
| 277 | |
| 278 | for message in messages: |
| 279 | if self.task == "simulation/db_diag": |
| 280 | content_json: dict = message.content |
| 281 | content_json[ |
| 282 | "diagnose" |
| 283 | ] = f"[{message.sender}]: {content_json['diagnose']}" |
| 284 | _format[self.agent_id[message.sender]]["message"] = json.dumps( |
| 285 | content_json |
| 286 | ) |
| 287 | elif "sde" in self.task: |
| 288 | if message.sender == "code_tester": |
| 289 | pre_message, message_ = message.content.split("\n") |
| 290 | message_ = "{}\n{}".format( |
| 291 | pre_message, json.loads(message_)["feedback"] |
| 292 | ) |
| 293 | _format[self.agent_id[message.sender]][ |
| 294 | "message" |
| 295 | ] = "[{}]: {}".format(message.sender, message_) |
| 296 | else: |
| 297 | _format[self.agent_id[message.sender]][ |
| 298 | "message" |
| 299 | ] = "[{}]: {}".format(message.sender, message.content) |
| 300 | |
| 301 | else: |
| 302 | _format[self.agent_id[message.sender]]["message"] = "[{}]: {}".format( |
| 303 | message.sender, message.content |
| 304 | ) |
| 305 | |
| 306 | return _format |
| 307 | |
| 308 | def gen_output(self): |
| 309 | """ |