plugin: link/image: commit py node

1 files changed, 42 insertions, 0 deletions

diff --git a/priv/python/pyerlai/node.py b/priv/python/pyerlai/node.py
new file mode 100644
index 0000000..b79b526
--- /dev/null
+++ b/priv/python/pyerlai/node.py
@@ -0,0 +1,42 @@
+import logging
+import argparse
+import os
+import sys
+from pyrlang import Node
+
+from genservers.image_to_text import ImageToTextViTGPT2
+from genservers.clip_ask import ClipAsk
+from genservers.pil import Pil
+
+def main():
+    parser = argparse.ArgumentParser()
+
+    parser.add_argument("host_node", help="host node")
+    parser.add_argument("node", help="python node name")
+
+    parser.add_argument("cookie", help="node cookie. prefer use NODE_COOKIE environment variable.")
+    parser.add_argument("process", help="process name (pyerlai)", default="pyerlai")
+
+    args = parser.parse_args()
+
+    cookie = os.environ.get('NODE_COOKIE')
+    if cookie is None:
+        if args.cookie is None:
+            sys.exit("pyerlai-python: missing NODE_COOKIE or --cookie")
+        else:
+            cookie = args.cookie
+
+    logger = logging.getLogger("pyerlai-python:" + args.node)
+    logging.getLogger("").setLevel(logging.DEBUG)
+
+    logger.info("starting node")
+    node = Node(node_name=args.node, cookie=cookie, hidden=True)
+
+    ImageToTextViTGPT2(node, logger)
+    ClipAsk(node, logger)
+    Pil(node, logger)
+
+    node.run()
+
+if __name__ == "__main__":
+    main()