Interactive cli for python converter

python/requirements.txt

@@ -4,3 +4,4 @@ numpy==1.16.4
 six==1.11.0
 tensorflow==1.14.0
 tensorflow-hub==0.5.0
+PyInquirer==1.0.3

python/setup.py

@@ -27,6 +27,7 @@ def _get_requirements(file):
 
 CONSOLE_SCRIPTS = [
     'tensorflowjs_converter = tensorflowjs.converters.converter:main',
+    'tensorflowjs_cli = tensorflowjs.cli:main',
 ]
 
 setuptools.setup(
@@ -54,6 +55,7 @@ def _get_requirements(file):
     ],
     py_modules=[
         'tensorflowjs',
+        'tensorflowjs.cli',
         'tensorflowjs.version',
         'tensorflowjs.quantization',
         'tensorflowjs.read_weights',

python/tensorflowjs/BUILD

@@ -99,6 +99,28 @@ py_test(
     ],
 )
 
+py_test(
+    name = "cli_test",
+    srcs = ["cli_test.py"],
+    srcs_version = "PY2AND3",
+    deps = [
+        ":expect_numpy_installed",
+        ":cli",
+    ],
+)
+
+py_binary(
+    name = "cli",
+    srcs = ["cli.py"],
+    srcs_version = "PY2AND3",
+    deps = [
+        ":converters/converter",
+        "//tensorflowjs:expect_h5py_installed",
+        "//tensorflowjs:expect_keras_installed",
+        "//tensorflowjs:expect_tensorflow_installed",
+    ],
+)
+
 # A filegroup BUILD target that includes all the op list json files in the
 # the op_list/ folder. The op_list folder itself is a symbolic link to the
 # actual op_list folder under src/.

python/tensorflowjs/__init__.py

@@ -21,5 +21,6 @@
 from tensorflowjs import converters
 from tensorflowjs import quantization
 from tensorflowjs import version
+from tensorflowjs import cli
 
 __version__ = version.version