make_aibo_dairy: support --test-response

jsk-ros-pkg · Apr 26, 2024 · c67cdc1 · c67cdc1
1 parent 3afe66e
commit c67cdc1
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/database_talker/scripts/make_aibo_diary.py b/database_talker/scripts/make_aibo_diary.py
@@ -145,7 +145,8 @@ def make_diary(self, language="Japanese"):
 
 if __name__ == '__main__':
     parser = argparse.ArgumentParser()
-    parser.add_argument('--test', action='store_true')
+    parser.add_argument('--test-diary', '--test', action='store_true')
+    parser.add_argument('--test-response', type=str, default=None)
     parser.add_argument('--prompt-type', default='basic', choices=['basic','personality'])
 
     args = parser.parse_args(rospy.myargv()[1:])
@@ -155,10 +156,15 @@ def make_diary(self, language="Japanese"):
     logger = logging.getLogger('rosout')
     logger.setLevel(rospy.impl.rosout._rospy_to_logging_levels[rospy.DEBUG])
 
-    ml = MessageListener(wait_for_chat_server=not args.test, prompt_type=args.prompt_type)
-    if args.test:
+    ml = MessageListener(wait_for_chat_server=not (args.test_diary or args.test_response), prompt_type=args.prompt_type)
+    if args.test_diary:
         ret = ml.make_diary()
         if 'filename' in ret:
             rospy.loginfo("image is saved at {}".format(ret['filename']))
         sys.exit(0)
+    elif args.test_response:
+        ret = ml.make_response(args.test_response)
+        if 'filename' in ret:
+            rospy.loginfo("image is saved at {}".format(ret['filename']))
+        sys.exit(0)
     rospy.spin()