cmdbox 0.6.4.1__tar.gz → 0.6.4.2__tar.gz

{cmdbox-0.6.4.1/cmdbox.egg-info → cmdbox-0.6.4.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cmdbox
-Version: 0.6.4.1
+Version: 0.6.4.2
 Summary: cmdbox: It is a command line application with a plugin mechanism.
 Home-page: https://github.com/hamacom2004jp/cmdbox
 Download-URL: https://github.com/hamacom2004jp/cmdbox

cmdbox-0.6.4.2/cmdbox/app/features/cli/cmdbox_vision_install.py

@@ -0,0 +1,213 @@
+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 cmdbox.app.features.cli import cmdbox_vision_start
+from pathlib import Path
+from typing import Dict, Any, Tuple, List, Union
+import argparse
+import glob
+import logging
+import pip
+import requests
+import shutil
+import subprocess
+import sys
+
+
+class VisionInstall(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 'install'
+    
+    def get_option(self):
+        """
+        この機能のオプションを返します
+
+        Returns:
+            Dict[str, Any]: オプション
+        """
+        return dict(
+            use_redis=self.USE_REDIS_MEIGHT, nouse_webmode=True, use_agent=False,
+            description_ja="画像/動画の推論エンジンをインストールします。",
+            description_en="Installs the image/video inference engine.",
+            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="300", required=False, multi=False, hide=True, choice=None,
+                     description_ja="サーバーの応答が返ってくるまでの最大待ち時間を指定。",
+                     description_en="Specify the maximum waiting time until the server responds."),
+                dict(opt="client_only", type=Options.T_BOOL, default=False, required=False, multi=False, hide=True, choice=[True, False],
+                     description_ja="サーバーへの接続を行わないようにします。",
+                     description_en="Do not make connections to the server."),
+                dict(opt="vision_engine", type=Options.T_STR, default="sam2", required=True, multi=False, hide=False,
+                     choice=["sam2"],
+                     description_ja="使用するVisionエンジンを指定します。",
+                     description_en="Specify the Vision engine to use."),
+            ]
+        )
+
+    def get_svcmd(self):
+        """
+        この機能のサーバー側のコマンドを返します
+
+        Returns:
+            str: サーバー側のコマンド
+        """
+        return 'vision_install'
+
+    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':
+            vision_engine_b64 = convert.str2b64str(args.vision_engine)
+            if args.client_only:
+                # クライアントのみの場合は、サーバーに接続せずに実行
+                ret = self.install(common.random_string(), args.vision_engine, logger)
+            else:
+                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],
+                                            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, None
+            return self.RESP_SUCCESS, ret, None
+
+        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])
+        ret = self.install(msg[1], vision_engine, logger)
+
+        if 'success' not in ret:
+            redis_cli.rpush(msg[1], ret)
+            return self.RESP_WARN
+        return self.RESP_SUCCESS
+
+    def install(self, reskey:str, vision_engine:str, logger:logging.Logger) -> Dict[str, Any]:
+        """
+        SAML2をインストールします
+
+        Args:
+            reskey (str): レスポンスキー
+            vision_engine (str): Visionエンジン
+            logger (logging.Logger): ロガー
+
+        Returns:
+            Dict[str, Any]: 結果
+        """
+        try:
+            if vision_engine == 'sam2':
+                #===============================================================
+                # SAM2のモデルファイルのダウンロード
+                sam2_dir = Path(version.__file__).parent / '.sam2' / 'model'
+                # すでにダウンローダーが存在する場合は削除
+                if sam2_dir.exists():
+                    shutil.rmtree(sam2_dir)
+                for mk, mv in self.VISION_MODEL.items():
+                    model_file = sam2_dir / mv['path']
+                    model_file.parent.mkdir(parents=True, exist_ok=True)
+                    # モデルファイルを保存
+                    logger.info(f"Downloading.. : {mv['url']}")
+                    responce = requests.get(f"{mv['url']}", allow_redirects=True)
+                    if responce.status_code != 200:
+                        _msg = f"Failed to download SAM2 model: {responce.status_code} {responce.reason}. {mv['url']}"
+                        logger.error(_msg, exc_info=True)
+                        return dict(warn=_msg)
+                    with open(model_file, mode='wb') as f:
+                        f.write(responce.content)
+                #===============================================================
+                # SAM2のpythonライブラリのインストール
+                whl_url = f'git+https://github.com/facebookresearch/sam2.git'
+                # whlファイルをpipでインストール
+                if logger.level == logging.DEBUG:
+                    logger.debug(f"pip install {whl_url}")
+                rescode = pip.main(['install', str(whl_url)])  # pipのインストール
+                logger.info(f"Install wheel: {whl_url}")
+                if rescode != 0:
+                    _msg = f"Failed to install SAM2 python library: Possible whl not in the environment. {whl_url}"
+                    logger.error(_msg, exc_info=True)
+                    return dict(warn=_msg)
+                #===============================================================
+                # 成功時の処理
+                rescode, _msg = (self.RESP_SUCCESS, dict(success=f'Success to install SAM2 python library. {whl_url}'))
+                return dict(success=_msg)
+            else:
+                _msg = f"Unsupported vision engine: {vision_engine}"
+                logger.error(_msg, exc_info=True)
+                return dict(warn=_msg)
+        except Exception as e:
+            _msg = f"Failed to install vision engine: {e}"
+            logger.warning(_msg, exc_info=True)
+            return dict(warn=_msg)

cmdbox-0.6.4.2/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-0.6.4.2/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-0.6.4.1 → cmdbox-0.6.4.2}/cmdbox/app/features/web/cmdbox_web_versions_used.py

@@ -25,7 +25,7 @@ class VersionsUsed(feature.WebFeature):
                 for i, line in enumerate(f.readlines()):
                     parts = line.strip().split('\t')
                     ret.append(parts)
-            with open(Path(cmdbox.__file__).parent / 'app' / 'lib_license_list.txt', 'r', encoding='utf-8') as f:
+            with open(Path(cmdbox.__file__).parent / 'web' / 'voicevox_license_list.txt', 'r', encoding='utf-8') as f:
                 for i, line in enumerate(f.readlines()):
                     parts = line.strip().split('\t')
                     ret.append(parts)