cmdbox 0.6.4__py3-none-any.whl → 0.6.4.2__py3-none-any.whl

cmdbox/app/features/cli/cmdbox_vision_predict.py

@@ -0,0 +1,192 @@
+from cmdbox import version
+from cmdbox.app import common, client, feature
+from cmdbox.app.commons import convert, redis_client
+from cmdbox.app.features.cli import cmdbox_vision_start
+from cmdbox.app.options import Options
+from pathlib import Path
+from typing import Dict, Any, Tuple, List, Union
+import argparse
+import logging
+import json
+
+
+class VisionPredict(cmdbox_vision_start.VisionStart):
+
+    def get_mode(self) -> Union[str, List[str]]:
+        """
+        この機能のモードを返します
+
+        Returns:
+            Union[str, List[str]]: モード
+        """
+        return 'vision'
+
+    def get_cmd(self):
+        """
+        この機能のコマンドを返します
+
+        Returns:
+            str: コマンド
+        """
+        return 'predict'
+
+    def get_option(self):
+        """
+        この機能のオプションを返します
+
+        Returns:
+            Dict[str, Any]: オプション
+        """
+        return dict(
+            use_redis=self.USE_REDIS_MEIGHT, nouse_webmode=False, use_agent=True,
+            description_ja="画像/動画の推論を実行します。",
+            description_en="Executes inference on images/videos.",
+            choice=[
+                dict(opt="host", type=Options.T_STR, default=self.default_host, required=True, multi=False, hide=True, choice=None, web="mask",
+                     description_ja="Redisサーバーのサービスホストを指定します。",
+                     description_en="Specify the service host of the Redis server."),
+                dict(opt="port", type=Options.T_INT, default=self.default_port, required=True, multi=False, hide=True, choice=None, web="mask",
+                     description_ja="Redisサーバーのサービスポートを指定します。",
+                     description_en="Specify the service port of the Redis server."),
+                dict(opt="password", type=Options.T_STR, default=self.default_pass, required=True, multi=False, hide=True, choice=None, web="mask",
+                     description_ja=f"Redisサーバーのアクセスパスワード(任意)を指定します。省略時は `{self.default_pass}` を使用します。",
+                     description_en=f"Specify the access password of the Redis server (optional). If omitted, `{self.default_pass}` is used."),
+                dict(opt="svname", type=Options.T_STR, default=self.default_svname, required=True, multi=False, hide=True, choice=None, web="readonly",
+                     description_ja="サーバーのサービス名を指定します。省略時は `server` を使用します。",
+                     description_en="Specify the service name of the inference server. If omitted, `server` is used."),
+                dict(opt="retry_count", type=Options.T_INT, default=3, required=False, multi=False, hide=True, choice=None,
+                     description_ja="Redisサーバーへの再接続回数を指定します。0以下を指定すると永遠に再接続を行います。",
+                     description_en="Specifies the number of reconnections to the Redis server.If less than 0 is specified, reconnection is forever."),
+                dict(opt="retry_interval", type=Options.T_INT, default=5, required=False, multi=False, hide=True, choice=None,
+                     description_ja="Redisサーバーに再接続までの秒数を指定します。",
+                     description_en="Specifies the number of seconds before reconnecting to the Redis server."),
+                dict(opt="timeout", type=Options.T_INT, default="60", required=False, multi=False, hide=True, choice=None,
+                     description_ja="サーバーの応答が返ってくるまでの最大待ち時間を指定。",
+                     description_en="Specify the maximum waiting time until the server responds."),
+                dict(opt="vision_engine", type=Options.T_STR, default="sam2", required=True, multi=False, hide=False,
+                     choice=["sam2"],
+                     choice_show=dict(sam2=["sam2_model", "sam2_point"]),
+                     description_ja="使用するVisionエンジンを指定します。",
+                     description_en="Specify the Vision engine to use."),
+                dict(opt="sam2_model", type=Options.T_STR, default="092824/sam2.1_hiera_tiny.pt", required=True, multi=False, hide=False,
+                     choice=[k for k in self.VISION_MODEL.keys() if self.VISION_MODEL[k]['type'] == 'sam2'],
+                     choice_edit=True,
+                     description_ja="使用するSAM2モデルを指定します。",
+                     description_en="Specify the SAM2 model to use."),
+                dict(opt="sam2_point", type=Options.T_STR, default=None, required=True, multi=True, hide=False, choice=None,
+                     description_ja="使用するSAM2モデルのポイントを指定します。 `x,y` の形式で指定してください。",
+                     description_en="Specify the SAM2 model points to use. Please specify in the format `x,y`."),
+            ]
+        )
+    
+    def get_svcmd(self):
+        """
+        この機能のサーバー側のコマンドを返します
+
+        Returns:
+            str: サーバー側のコマンド
+        """
+        return 'vision_predict'
+
+    def apprun(self, logger:logging.Logger, args:argparse.Namespace, tm:float, pf:List[Dict[str, float]]=[]) -> Tuple[int, Dict[str, Any], Any]:
+        """
+        この機能の実行を行います
+
+        Args:
+            logger (logging.Logger): ロガー
+            args (argparse.Namespace): 引数
+            tm (float): 実行開始時間
+            pf (List[Dict[str, float]]): 呼出元のパフォーマンス情報
+
+        Returns:
+            Tuple[int, Dict[str, Any], Any]: 終了コード, 結果, オブジェクト
+        """
+        if args.vision_engine is None:
+            msg = dict(warn=f"Please specify the --vision_engine option.")
+            common.print_format(msg, False, tm, args.output_json, args.output_json_append, pf=pf)
+            return self.RESP_WARN, msg, None
+        if args.vision_engine == 'sam2':
+            if args.sam2_model is None:
+                msg = dict(warn=f"Please specify the --sam2_model option.")
+                common.print_format(msg, False, tm, args.output_json, args.output_json_append, pf=pf)
+                return self.RESP_WARN, msg, None
+
+            vision_engine_b64 = convert.str2b64str(args.vision_engine)
+            sam2_model_b64 = convert.str2b64str(args.sam2_model)
+            sam2_point_b64 = convert.str2b64str(common.to_str(args.sam2_point))
+
+            cl = client.Client(logger, redis_host=args.host, redis_port=args.port, redis_password=args.password, svname=args.svname)
+            ret = cl.redis_cli.send_cmd(self.get_svcmd(),
+                                        [vision_engine_b64, sam2_model_b64, sam2_point_b64],
+                                        retry_count=args.retry_count, retry_interval=args.retry_interval, timeout=args.timeout, nowait=False)
+            common.print_format(ret, False, tm, None, False, pf=pf)
+            if 'success' not in ret:
+                    return self.RESP_WARN, ret, cl
+            return self.RESP_SUCCESS, ret, cl
+
+        msg = dict(warn=f"Unsupported vision engine: {args.vision_engine}")
+        common.print_format(msg, False, tm, args.output_json, args.output_json_append, pf=pf)
+        return self.RESP_WARN, msg, None
+
+    def is_cluster_redirect(self):
+        """
+        クラスター宛のメッセージの場合、メッセージを転送するかどうかを返します
+
+        Returns:
+            bool: メッセージを転送する場合はTrue
+        """
+        return False
+
+    def svrun(self, data_dir:Path, logger:logging.Logger, redis_cli:redis_client.RedisClient, msg:List[str],
+              sessions:Dict[str, Dict[str, Any]]) -> int:
+        """
+        この機能のサーバー側の実行を行います
+
+        Args:
+            data_dir (Path): データディレクトリ
+            logger (logging.Logger): ロガー
+            redis_cli (redis_client.RedisClient): Redisクライアント
+            msg (List[str]): 受信メッセージ
+            sessions (Dict[str, Dict[str, Any]]): セッション情報
+        
+        Returns:
+            int: 終了コード
+        """
+        if logger.level == logging.DEBUG:
+            logger.debug(f"audit write svrun msg: {msg}")
+        vision_engine = convert.b64str2str(msg[2])
+        sam2_model = convert.b64str2str(msg[3])
+        sam2_point = json.loads(convert.b64str2str(msg[4]))
+        st = self.predict(msg[1], vision_engine, sam2_model, sam2_point, data_dir, logger, redis_cli, sessions)
+        return st
+
+    def predict(self, reskey:str, vision_engine:str, sam2_model:str, sam2_point:List[List[int]], data_dir:Path, logger:logging.Logger,
+                redis_cli:redis_client.RedisClient, sessions:Dict[str, Dict[str, Any]]) -> int:
+        """
+        SAM2のモデルを使用して指定したポイントのマスクを取得します
+
+        Args:
+            reskey (str): レスポンスキー
+            vision_engine (str): Visionエンジン
+            sam2_model (str): SAM2モデル
+            sam2_point (List[List[int,int]]): SAM2モデルのポイント
+            data_dir (Path): データディレクトリ
+            logger (logging.Logger): ロガー
+            redis_cli (redis_client.RedisClient): Redisクライアント
+            sessions (Dict[str, Dict[str, Any]]): セッション情報
+
+        Returns:
+            int: レスポンスコード
+        """
+        try:
+            if vision_engine == 'sam2':
+                raise NotImplementedError("SAM2 model is not implemented yet.")
+            else:
+                logger.warning(f"Unsupported vision engine: {vision_engine}")
+                redis_cli.rpush(reskey, dict(warn=f"Unsupported vision engine: {vision_engine}"))
+                return self.RESP_WARN
+
+        except Exception as e:
+            logger.warning(f"Failed to start: {e}", exc_info=True)
+            redis_cli.rpush(reskey, dict(warn=f"Failed to start: {e}"))
+            return self.RESP_WARN

cmdbox/app/features/cli/cmdbox_vision_start.py

@@ -0,0 +1,230 @@
+from cmdbox import version
+from cmdbox.app import common, client, feature
+from cmdbox.app.commons import convert, redis_client
+from cmdbox.app.options import Options
+from pathlib import Path
+from typing import Dict, Any, Tuple, List, Union
+import argparse
+import logging
+
+
+class VisionStart(feature.UnsupportEdgeFeature):
+    VISION_MODEL = dict()
+    VISION_MODEL['092824/sam2.1_hiera_tiny'] = dict(type='sam2',
+                                                    path='092824/sam2.1_hiera_tiny.pt',
+                                                    conf='configs/sam2.1/sam2.1_hiera_t.yaml',
+                                                    url='https://dl.fbaipublicfiles.com/segment_anything_2/092824/sam2.1_hiera_tiny.pt')
+    VISION_MODEL['092824/sam2.1_hiera_small'] = dict(type='sam2',
+                                                     path='092824/sam2.1_hiera_small.pt',
+                                                     conf='configs/sam2.1/sam2.1_hiera_s.yaml',
+                                                     url='https://dl.fbaipublicfiles.com/segment_anything_2/092824/sam2.1_hiera_small.pt')
+    VISION_MODEL['092824/sam2.1_hiera_base_plus'] = dict(type='sam2',
+                                                         path='092824/sam2.1_hiera_base_plus.pt',
+                                                         conf='configs/sam2.1/sam2.1_hiera_b+.yaml',
+                                                         url='https://dl.fbaipublicfiles.com/segment_anything_2/092824/sam2.1_hiera_base_plus.pt')
+    VISION_MODEL['092824/sam2.1_hiera_large'] = dict(type='sam2',
+                                                     path='092824/sam2.1_hiera_large.pt',
+                                                     conf='configs/sam2.1/sam2.1_hiera_l.yaml',
+                                                     url='https://dl.fbaipublicfiles.com/segment_anything_2/092824/sam2.1_hiera_large.pt')
+
+    def get_mode(self) -> Union[str, List[str]]:
+        """
+        この機能のモードを返します
+
+        Returns:
+            Union[str, List[str]]: モード
+        """
+        return 'vision'
+
+    def get_cmd(self):
+        """
+        この機能のコマンドを返します
+
+        Returns:
+            str: コマンド
+        """
+        return 'start'
+    
+    def get_option(self):
+        """
+        この機能のオプションを返します
+
+        Returns:
+            Dict[str, Any]: オプション
+        """
+        return dict(
+            use_redis=self.USE_REDIS_MEIGHT, nouse_webmode=False, use_agent=True,
+            description_ja="画像/動画の推論を開始します。",
+            description_en="Starts inference on images/videos.",
+            choice=[
+                dict(opt="host", type=Options.T_STR, default=self.default_host, required=True, multi=False, hide=True, choice=None, web="mask",
+                     description_ja="Redisサーバーのサービスホストを指定します。",
+                     description_en="Specify the service host of the Redis server."),
+                dict(opt="port", type=Options.T_INT, default=self.default_port, required=True, multi=False, hide=True, choice=None, web="mask",
+                     description_ja="Redisサーバーのサービスポートを指定します。",
+                     description_en="Specify the service port of the Redis server."),
+                dict(opt="password", type=Options.T_STR, default=self.default_pass, required=True, multi=False, hide=True, choice=None, web="mask",
+                     description_ja=f"Redisサーバーのアクセスパスワード(任意)を指定します。省略時は `{self.default_pass}` を使用します。",
+                     description_en=f"Specify the access password of the Redis server (optional). If omitted, `{self.default_pass}` is used."),
+                dict(opt="svname", type=Options.T_STR, default=self.default_svname, required=True, multi=False, hide=True, choice=None, web="readonly",
+                     description_ja="サーバーのサービス名を指定します。省略時は `server` を使用します。",
+                     description_en="Specify the service name of the inference server. If omitted, `server` is used."),
+                dict(opt="retry_count", type=Options.T_INT, default=3, required=False, multi=False, hide=True, choice=None,
+                     description_ja="Redisサーバーへの再接続回数を指定します。0以下を指定すると永遠に再接続を行います。",
+                     description_en="Specifies the number of reconnections to the Redis server.If less than 0 is specified, reconnection is forever."),
+                dict(opt="retry_interval", type=Options.T_INT, default=5, required=False, multi=False, hide=True, choice=None,
+                     description_ja="Redisサーバーに再接続までの秒数を指定します。",
+                     description_en="Specifies the number of seconds before reconnecting to the Redis server."),
+                dict(opt="timeout", type=Options.T_INT, default="60", required=False, multi=False, hide=True, choice=None,
+                     description_ja="サーバーの応答が返ってくるまでの最大待ち時間を指定。",
+                     description_en="Specify the maximum waiting time until the server responds."),
+                dict(opt="vision_engine", type=Options.T_STR, default="sam2", required=True, multi=False, hide=False,
+                     choice=["sam2"],
+                     choice_show=dict(sam2=["sam2_model"]),
+                     description_ja="使用するVisionエンジンを指定します。",
+                     description_en="Specify the Vision engine to use."),
+                dict(opt="sam2_model", type=Options.T_STR, default="092824/sam2.1_hiera_tiny.pt", required=True, multi=False, hide=False,
+                     choice=[k for k in self.VISION_MODEL.keys() if self.VISION_MODEL[k]['type'] == 'sam2'],
+                     choice_edit=True,
+                     description_ja="使用するSAM2モデルを指定します。",
+                     description_en="Specify the SAM2 model to use."),
+            ]
+        )
+
+    def get_svcmd(self):
+        """
+        この機能のサーバー側のコマンドを返します
+
+        Returns:
+            str: サーバー側のコマンド
+        """
+        return 'vision_start'
+
+    def apprun(self, logger:logging.Logger, args:argparse.Namespace, tm:float, pf:List[Dict[str, float]]=[]) -> Tuple[int, Dict[str, Any], Any]:
+        """
+        この機能の実行を行います
+
+        Args:
+            logger (logging.Logger): ロガー
+            args (argparse.Namespace): 引数
+            tm (float): 実行開始時間
+            pf (List[Dict[str, float]]): 呼出元のパフォーマンス情報
+
+        Returns:
+            Tuple[int, Dict[str, Any], Any]: 終了コード, 結果, オブジェクト
+        """
+        if args.vision_engine is None:
+            msg = dict(warn=f"Please specify the --vision_engine option.")
+            common.print_format(msg, False, tm, args.output_json, args.output_json_append, pf=pf)
+            return self.RESP_WARN, msg, None
+        if args.vision_engine == 'sam2':
+            if args.sam2_model is None:
+                msg = dict(warn=f"Please specify the --sam2_model option.")
+                common.print_format(msg, False, tm, args.output_json, args.output_json_append, pf=pf)
+                return self.RESP_WARN, msg, None
+
+            vision_engine_b64 = convert.str2b64str(args.vision_engine)
+            sam2_model_b64 = convert.str2b64str(args.sam2_model)
+            cl = client.Client(logger, redis_host=args.host, redis_port=args.port, redis_password=args.password, svname=args.svname)
+            ret = cl.redis_cli.send_cmd(self.get_svcmd(),
+                                        [vision_engine_b64, sam2_model_b64],
+                                        retry_count=args.retry_count, retry_interval=args.retry_interval, timeout=args.timeout, nowait=False)
+            common.print_format(ret, False, tm, None, False, pf=pf)
+            if 'success' not in ret:
+                    return self.RESP_WARN, ret, cl
+            return self.RESP_SUCCESS, ret, cl
+
+        msg = dict(warn=f"Unsupported vision engine: {args.vision_engine}")
+        common.print_format(msg, False, tm, args.output_json, args.output_json_append, pf=pf)
+        return self.RESP_WARN, msg, None
+
+    def is_cluster_redirect(self):
+        """
+        クラスター宛のメッセージの場合、メッセージを転送するかどうかを返します
+
+        Returns:
+            bool: メッセージを転送する場合はTrue
+        """
+        return True
+
+    def svrun(self, data_dir:Path, logger:logging.Logger, redis_cli:redis_client.RedisClient, msg:List[str],
+              sessions:Dict[str, Dict[str, Any]]) -> int:
+        """
+        この機能のサーバー側の実行を行います
+
+        Args:
+            data_dir (Path): データディレクトリ
+            logger (logging.Logger): ロガー
+            redis_cli (redis_client.RedisClient): Redisクライアント
+            msg (List[str]): 受信メッセージ
+            sessions (Dict[str, Dict[str, Any]]): セッション情報
+        
+        Returns:
+            int: 終了コード
+        """
+        if logger.level == logging.DEBUG:
+            logger.debug(f"audit write svrun msg: {msg}")
+        vision_engine = convert.b64str2str(msg[2])
+        sam2_model = convert.b64str2str(msg[3])
+        st = self.start(msg[1], vision_engine, sam2_model, data_dir, logger, redis_cli, sessions)
+        return st
+
+    def start(self, reskey:str, vision_engine:str, sam2_model:str, data_dir:Path, logger:logging.Logger,
+              redis_cli:redis_client.RedisClient, sessions:Dict[str, Dict[str, Any]]) -> int:
+        """
+        SAM2のモデルを開始します
+
+        Args:
+            reskey (str): レスポンスキー
+            vision_engine (str): Visionエンジン
+            sam2_model (str): SAM2モデル
+            data_dir (Path): データディレクトリ
+            logger (logging.Logger): ロガー
+            redis_cli (redis_client.RedisClient): Redisクライアント
+            sessions (Dict[str, Dict[str, Any]]): セッション情報
+
+        Returns:
+            int: レスポンスコード
+        """
+        try:
+            if vision_engine == 'sam2':
+                #===============================================================
+                # SAM2モデルの初期化
+                from sam2.build_sam import build_sam2
+                from sam2.sam2_image_predictor import SAM2ImagePredictor
+                #from sam2.sam2_video_predictor import SAM2VideoPredictor
+                import torch
+
+                sam2_dir = Path(version.__file__).parent / '.sam2' / 'model'
+                if not sam2_dir.exists():
+                    logger.error(f"Failed to start SAM2 model: sam2 directory does not exist: {sam2_dir}")
+                    redis_cli.rpush(reskey, dict(warn=f"Failed to start SAM2 model: sam2 directory does not exist: {sam2_dir}"))
+                    return self.RESP_WARN
+
+                model_file = sam2_dir / self.VISION_MODEL[sam2_model]['path']
+                config_file = self.VISION_MODEL[sam2_model]['conf']
+                if not model_file.exists():
+                    logger.error(f"Failed to start SAM2 model: model file does not exist: {model_file}")
+                    redis_cli.rpush(reskey, dict(warn=f"Failed to start SAM2 model: model file does not exist: {model_file}"))
+                    return self.RESP_WARN
+
+                device = 'cuda' if torch.cuda.is_available() else 'cpu'
+                # vvmファイルの読込み
+                predictor = SAM2ImagePredictor(build_sam2(config_file, model_file, device=device))
+                sessions[sam2_model] = dict(
+                    info=self.VISION_MODEL[sam2_model].copy(),
+                    predictor=predictor,
+                )
+                #===============================================================
+                # 成功時の処理
+                rescode, msg = (self.RESP_SUCCESS, dict(success=f'Success to start SAM2. Model: {sam2_model}'))
+                redis_cli.rpush(reskey, msg)
+                return rescode
+            else:
+                logger.warning(f"Unsupported vision engine: {vision_engine}")
+                redis_cli.rpush(reskey, dict(warn=f"Unsupported vision engine: {vision_engine}"))
+                return self.RESP_WARN
+        except Exception as e:
+            logger.warning(f"Failed to start: {e}", exc_info=True)
+            redis_cli.rpush(reskey, dict(warn=f"Failed to start: {e}"))
+            return self.RESP_WARN

cmdbox/app/features/cli/cmdbox_web_apikey_add.py

@@ -33,7 +33,7 @@ class WebApikeyAdd(feature.UnsupportEdgeFeature):
             Dict[str, Any]: オプション
         """
         return dict(
-            use_redis=self.USE_REDIS_MEIGHT, nouse_webmode=False,
+            use_redis=self.USE_REDIS_MEIGHT, nouse_webmode=False, use_agent=False,
             description_ja="WebモードのユーザーのApiKeyを追加します。",
             description_en="Add an ApiKey for a user in Web mode.",
             choice=[

cmdbox/app/features/cli/cmdbox_web_apikey_del.py

@@ -33,7 +33,7 @@ class WebApikeyDel(feature.UnsupportEdgeFeature):
             Dict[str, Any]: オプション
         """
         return dict(
-            use_redis=self.USE_REDIS_MEIGHT, nouse_webmode=False,
+            use_redis=self.USE_REDIS_MEIGHT, nouse_webmode=False, use_agent=False,
             description_ja="WebモードのユーザーのApiKeyを削除します。",
             description_en="Del an ApiKey for a user in Web mode.",
             choice=[

cmdbox/app/features/cli/cmdbox_web_gencert.py

@@ -38,7 +38,7 @@ class WebGencert(feature.UnsupportEdgeFeature):
             Dict[str, Any]: オプション
         """
         return dict(
-            use_redis=self.USE_REDIS_FALSE, nouse_webmode=False,
+            use_redis=self.USE_REDIS_FALSE, nouse_webmode=False, use_agent=False,
             description_ja="webモードでSSLを簡易的に実装するために自己署名証明書を生成します。",
             description_en="Generate a self-signed certificate for simple implementation of SSL in web mode.",
             choice=[

cmdbox/app/features/cli/cmdbox_web_genpass.py

@@ -40,7 +40,7 @@ class WebGenpass(feature.OneshotResultEdgeFeature):
             Dict[str, Any]: オプション
         """
         return dict(
-            use_redis=self.USE_REDIS_FALSE, nouse_webmode=False,
+            use_redis=self.USE_REDIS_FALSE, nouse_webmode=False, use_agent=True,
             description_ja="webモードで使用できるパスワード文字列を生成します。",
             description_en="Generates a password string that can be used in web mode.",
             choice=[

cmdbox/app/features/cli/cmdbox_web_group_add.py

@@ -33,7 +33,7 @@ class WebGroupAdd(feature.UnsupportEdgeFeature):
             Dict[str, Any]: オプション
         """
         return dict(
-            use_redis=self.USE_REDIS_MEIGHT, nouse_webmode=False,
+            use_redis=self.USE_REDIS_MEIGHT, nouse_webmode=False, use_agent=False,
             description_ja="Webモードのグループを追加します。",
             description_en="Add a group in Web mode.",
             choice=[

cmdbox/app/features/cli/cmdbox_web_group_del.py

@@ -33,7 +33,7 @@ class WebGroupDel(feature.UnsupportEdgeFeature):
             Dict[str, Any]: オプション
         """
         return dict(
-            use_redis=self.USE_REDIS_MEIGHT, nouse_webmode=False,
+            use_redis=self.USE_REDIS_MEIGHT, nouse_webmode=False, use_agent=False,
             description_ja="Webモードのグループを削除します。",
             description_en="Del a group in Web mode.",
             choice=[

cmdbox/app/features/cli/cmdbox_web_group_edit.py

@@ -33,7 +33,7 @@ class WebGroupEdit(feature.UnsupportEdgeFeature):
             Dict[str, Any]: オプション
         """
         return dict(
-            use_redis=self.USE_REDIS_MEIGHT, nouse_webmode=False,
+            use_redis=self.USE_REDIS_MEIGHT, nouse_webmode=False, use_agent=False,
             description_ja="Webモードのグループを編集します。",
             description_en="Edit a group in Web mode.",
             choice=[

cmdbox/app/features/cli/cmdbox_web_group_list.py

@@ -6,7 +6,7 @@ import argparse
 import logging
 
 
-class WebGroupList(feature.OneshotResultEdgeFeature):
+class WebGroupList(feature.UnsupportEdgeFeature):
     def get_mode(self) -> Union[str, List[str]]:
         """
         この機能のモードを返します
@@ -33,7 +33,7 @@ class WebGroupList(feature.OneshotResultEdgeFeature):
             Dict[str, Any]: オプション
         """
         return dict(
-            use_redis=self.USE_REDIS_MEIGHT, nouse_webmode=False,
+            use_redis=self.USE_REDIS_MEIGHT, nouse_webmode=False, use_agent=False,
             description_ja="Webモードのグループー一覧を取得します。",
             description_en="Get a list of users in Web mode.",
             choice=[

cmdbox/app/features/cli/cmdbox_web_start.py

@@ -38,6 +38,7 @@ class WebStart(feature.UnsupportEdgeFeature, agent_base.AgentBase):
         opt = super().get_option()
         opt['use_redis'] = self.USE_REDIS_MEIGHT
         opt['nouse_webmode'] = True
+        opt['use_agent'] = False
         opt['description_ja'] = "Webモードを起動します。"
         opt['description_en'] = "Start Web mode."
         opt['choice'] += [

cmdbox/app/features/cli/cmdbox_web_stop.py

@@ -33,7 +33,7 @@ class WebStop(feature.UnsupportEdgeFeature):
             Dict[str, Any]: オプション
         """
         return dict(
-            use_redis=self.USE_REDIS_MEIGHT, nouse_webmode=True,
+            use_redis=self.USE_REDIS_MEIGHT, nouse_webmode=True, use_agent=False,
             description_ja="Webモードを停止します。",
             description_en="Stop Web mode.",
             choice=[

cmdbox/app/features/cli/cmdbox_web_user_add.py

@@ -33,7 +33,7 @@ class WebUserAdd(feature.UnsupportEdgeFeature):
             Dict[str, Any]: オプション
         """
         return dict(
-            use_redis=self.USE_REDIS_MEIGHT, nouse_webmode=False,
+            use_redis=self.USE_REDIS_MEIGHT, nouse_webmode=False, use_agent=False,
             description_ja="Webモードのユーザーを追加します。",
             description_en="Add a user in Web mode.",
             choice=[

cmdbox/app/features/cli/cmdbox_web_user_del.py

@@ -33,7 +33,7 @@ class WebUserDel(feature.UnsupportEdgeFeature):
             Dict[str, Any]: オプション
         """
         return dict(
-            use_redis=self.USE_REDIS_MEIGHT, nouse_webmode=False,
+            use_redis=self.USE_REDIS_MEIGHT, nouse_webmode=False, use_agent=False,
             description_ja="Webモードのユーザーを削除します。",
             description_en="Delete a user in Web mode.",
             choice=[

cmdbox/app/features/cli/cmdbox_web_user_edit.py

@@ -33,7 +33,7 @@ class WebUserEdit(feature.UnsupportEdgeFeature):
             Dict[str, Any]: オプション
         """
         return dict(
-            use_redis=self.USE_REDIS_MEIGHT, nouse_webmode=False,
+            use_redis=self.USE_REDIS_MEIGHT, nouse_webmode=False, use_agent=False,
             description_ja="Webモードのユーザーを編集します。",
             description_en="Edit users in Web mode.",
             choice=[

cmdbox/app/features/cli/cmdbox_web_user_list.py

@@ -6,7 +6,7 @@ import argparse
 import logging
 
 
-class WebUserList(feature.OneshotResultEdgeFeature):
+class WebUserList(feature.UnsupportEdgeFeature):
     def get_mode(self) -> Union[str, List[str]]:
         """
         この機能のモードを返します
@@ -33,7 +33,7 @@ class WebUserList(feature.OneshotResultEdgeFeature):
             Dict[str, Any]: オプション
         """
         return dict(
-            use_redis=self.USE_REDIS_MEIGHT, nouse_webmode=False,
+            use_redis=self.USE_REDIS_MEIGHT, nouse_webmode=False, use_agent=False,
             description_ja="Webモードのユーザー一覧を取得します。",
             description_en="Get a list of users in Web mode.",
             choice=[

cmdbox/app/features/web/cmdbox_web_agent.py

@@ -1,5 +1,6 @@
 from cmdbox.app import common, feature
 from cmdbox.app.auth import signin
+from cmdbox.app.features.cli import agent_base
 from cmdbox.app.web import Web
 from fastapi import FastAPI, Depends, HTTPException, Request, Response, WebSocket
 from fastapi.responses import HTMLResponse, StreamingResponse
@@ -38,6 +39,16 @@ class Agent(feature.WebFeature):
             web.options.audit_exec(req, res, web)
             return web.agent_html_data
 
+        @app.get('/agent/llmsetting')
+        async def agent(req:Request, res:Response):
+            signin = web.signin.check_signin(req, res)
+            if signin is not None:
+                return signin
+            res.headers['Access-Control-Allow-Origin'] = '*'
+            web.options.audit_exec(req, res, web)
+            cmd = agent_base.AgentBase(web.appcls, web.ver)
+            return cmd.get_option().get('choice')
+
         @app.post('/agent/session/list')
         async def agent_session_list(req:Request, res:Response):
             signin = web.signin.check_signin(req, res)
@@ -150,12 +161,9 @@ class Agent(feature.WebFeature):
             if 'signin' in session:
                 user_id = session['signin']['name']
                 groups = session['signin']['groups']
-            # 言語認識
-            language, _ = locale.getlocale()
-            is_japan = language.find('Japan') >= 0 or language.find('ja_JP') >= 0
             # セッションを作成する
             agent_session = await web.create_agent_session(web.agent_runner.session_service, user_id, session_id=session_id)
-            startmsg = "こんにちは！何かお手伝いできることはありますか？" if is_japan else "Hello! Is there anything I can help you with?"
+            startmsg = "こんにちは！何かお手伝いできることはありますか？" if common.is_japan() else "Hello! Is there anything I can help you with?"
             yield json.dumps(dict(message=startmsg), default=common.default_json_enc)
             def _replace_match(match_obj):
                 json_str = match_obj.group(0)
@@ -229,6 +237,10 @@ class Agent(feature.WebFeature):
                     break
                 except self.SSEDisconnect as e:
                     break
+                except NotImplementedError as e:
+                    web.logger.warning(f'The session table needs to be reloaded.{e}', exc_info=True)
+                    yield json.dumps(dict(message=f'The session table needs to be reloaded. Please reload your browser.'), default=common.default_json_enc)
+                    break
                 except Exception as e:
                     web.logger.warning(f'chat error.', exc_info=True)
                     yield json.dumps(dict(message=f'<pre>{traceback.format_exc()}</pre>'), default=common.default_json_enc)

cmdbox/app/features/web/cmdbox_web_get_cmd_choices.py

@@ -1,4 +1,4 @@
-from cmdbox.app import feature
+from cmdbox.app import common, feature
 from cmdbox.app.web import Web
 from fastapi import FastAPI, Request, Response, HTTPException
 
@@ -20,5 +20,13 @@ class GetCmdChoices(feature.WebFeature):
             form = await req.form()
             mode = form.get('mode')
             cmd = form.get('cmd')
-            ret = web.options.get_cmd_choices(mode, cmd, True)
+            ret = web.options.get_cmd_choices(mode, cmd, True).copy()
+            fobj = web.options.get_cmd_attr(mode, cmd, 'feature')
+            desc = web.options.get_cmd_attr(mode, cmd, 'description_en' if not common.is_japan() else 'description_ja')
+            desc_nouse_webmode = '\U00002B55 Web' if not web.options.get_cmd_attr(mode, cmd, 'nouse_webmode') else '\U0000274C Web'
+            desc_use_agent = '\U00002B55 Agent' if web.options.get_cmd_attr(mode, cmd, 'use_agent') else '\U0000274C Agent'
+            desc_edge = '\U00002B55 Edge' if not isinstance(fobj, feature.UnsupportEdgeFeature) else '\U0000274C Edge'
+            help = dict(opt="help", type=web.options.T_TEXT, default=f"\U00002B55 CLI, {desc_nouse_webmode}, {desc_use_agent}, {desc_edge}, {desc}",
+                        required=False, multi=False, hide=False, choice=None, description_ja="-", description_en="-")
+            ret.insert(0, help)
             return ret