first commit to public bananaml

Author: erik-dunteman

.gitignore

@@ -1,3 +1,6 @@
+# Tests
+test*
+
 # Byte-compiled / optimized / DLL files
 __pycache__/
 *.py[cod]

README.md

@@ -1 +1,48 @@
-# banana-python-sdk
+# Banana Python SDK
+
+### Getting Started
+
+Install via pip
+`pip3 install banana-dev`
+
+Get your API Key
+- [Sign in / log in here](https://app.banana.dev)
+
+Run:
+```python
+import banana_dev as banana
+
+api_key = "YOUR_API_KEY"
+model_key = "YOUR_MODEL_KEY"
+model_inputs = {
+    # a json specific to your model. For example:
+    "a": 1,
+    "b": 2
+}
+
+out = banana.run(api_key, model_key, model_inputs)
+print(out)
+```
+
+Return type:
+```python
+{
+    {
+        "id": "12345678-1234-1234-1234-123456789012", 
+        "message": "success", 
+        "created": 1649712752, 
+        "apiVersion": "26 Nov 2021", 
+        "modelOutputs": [
+            {
+                # a json specific to your model. In this example, the sum of "a" and "b" from the above model_parameters
+                "sum": 3, 
+            }
+        ]
+    }
+}
+```
+
+Parse the server output:
+```python
+model_out = out["modelOutputs"][0]
+```

banana_dev/__init__.py

@@ -0,0 +1 @@
+from .package import run, start, check

banana_dev/config.py

@@ -0,0 +1,22 @@
+import os
+import json
+from uuid import uuid4
+
+def load_config():
+    user_path = os.path.expanduser("~")
+    cache_path = os.path.join(user_path, ".banana")
+    config_path = os.path.join(cache_path, "config.json")
+    if not os.path.exists(config_path):
+        os.makedirs(cache_path, exist_ok=True)
+        default_config = {
+            "machineID": str(uuid4())
+        }
+        with open(config_path, "w+") as f:
+            f.write(json.dumps(default_config))
+        return default_config
+    else:
+        with open(config_path, "r") as f:
+            return json.loads(f.read())
+
+if __name__ == "__main__":
+    print(load_config())

banana_dev/generics.py

@@ -0,0 +1,107 @@
+import requests
+import time
+import os
+import json
+from uuid import uuid4
+
+from .config import load_config
+
+endpoint = 'https://api.banana.dev/'
+# Endpoint override for development
+if 'BANANA_URL' in os.environ:
+    print("Dev Mode")
+    if os.environ['BANANA_URL'] == 'local':
+        endpoint = 'http://localhost/'
+    else:
+        endpoint = os.environ['BANANA_URL']
+    print("Hitting endpoint:", endpoint)
+
+config = load_config()
+
+# THE MAIN FUNCTIONS
+# ___________________________________
+
+
+def run_main(api_key, model_key, model_inputs):
+    call_id = start_api(api_key, model_key, model_inputs)
+    while True:
+        dict_out = check_api(api_key, call_id)
+        if dict_out['message'].lower() == "success":
+            return dict_out
+
+def start_main(api_key, model_key, model_inputs):
+    call_id = start_api(api_key, model_key, model_inputs)
+    return call_id
+
+def check_main(api_key, call_id):
+    dict_out = check_api(api_key, call_id)
+    return dict_out
+
+
+# THE API CALLING FUNCTIONS
+# ________________________
+
+# Takes in start params, returns call ID
+def start_api(api_key, model_key, model_inputs):
+    global endpoint
+    global config
+    route_start = "start/v2/"
+    url_start = endpoint + route_start
+
+    payload = {
+        "id": str(uuid4()),
+        "created": time.time(),
+        "apiKey" : api_key,
+        "modelKey" : model_key,
+        "modelInputs" : model_inputs,
+        "config": config
+    }
+
+    response = requests.post(url_start, json=payload)
+
+    if response.status_code != 200:
+        raise Exception("server error: status code {}".format(response.status_code))
+
+    try:
+        out = response.json()
+    except:
+        raise Exception("server error: returned invalid json")
+
+    try:
+        if "error" in out['message'].lower():
+            raise Exception(out['message'])
+        call_id = out['callID']
+        return call_id
+    except:
+        raise Exception("server error: Failed to return call_id")
+
+# The bare async checker.
+def check_api(api_key, call_id):
+    global endpoint
+    route_check = "check/v2/"
+    url_check = endpoint + route_check
+    # Poll server for completed task
+
+    payload = {
+        "id": str(uuid4()),
+        "created": int(time.time()),
+        "longPoll": True,
+        "callID": call_id, 
+        "apiKey": api_key
+    }
+    response = requests.post(url_check, json=payload)
+
+    if response.status_code != 200:
+        raise Exception("server error: status code {}".format(response.status_code))
+
+    try:
+        out = response.json()
+    except:
+        raise Exception("server error: returned invalid json")
+
+    try:
+        if "error" in out['message'].lower():
+            raise Exception(out['message'])
+        return out
+    except Exception as e:
+        raise e

banana_dev/package.py

@@ -0,0 +1,25 @@
+from .generics import run_main, start_main, check_main
+
+# Generics
+def run(api_key, model_key, model_inputs):
+    out = run_main(
+        api_key = api_key, 
+        model_key = model_key, 
+        model_inputs = model_inputs
+    )
+    return out
+
+def start(api_key, model_key, model_inputs):
+    out = start_main(
+        api_key = api_key, 
+        model_key = model_key, 
+        model_inputs = model_inputs
+    )
+    return out
+    
+def check(api_key, call_id):
+    out_dict = check_main(
+        api_key = api_key,
+        call_id = call_id
+    )
+    return out_dict

deploy.sh

@@ -0,0 +1,3 @@
+!#/bin/bash
+python3 setup.py sdist bdist_wheel
+python3 -m twine upload dist/* --skip-existing

setup.cfg

@@ -0,0 +1,4 @@
+
+# Inside of setup.cfg
+[metadata]
+description-file = README.md

setup.py

@@ -0,0 +1,36 @@
+from distutils.core import setup
+import setuptools
+from pathlib import Path
+
+this_directory = Path(__file__).parent
+long_description = (this_directory / "README.md").read_text()
+
+setup(
+  name = 'banana_dev',
+  packages = ['banana_dev'],
+  version = '3.0.0',
+  license='MIT',
+  description = 'The banana package is a python client to interact with your machine learning models hosted on Banana',   # Give a short description about your library
+  long_description=long_description,
+  long_description_content_type='text/markdown',
+  author = 'Erik Dunteman',
+  author_email = '[email protected]',
+  url = 'https://www.banana.dev',
+  keywords = ['Banana client', 'API wrapper', 'Banana', 'SDK'],
+  setup_requires = ['wheel'],
+  install_requires=[
+    "certifi==2021.10.8",
+    "charset-normalizer==2.0.7",
+    "idna==3.3",
+    "requests==2.26.0",
+    "urllib3==1.26.7",
+  ],
+  classifiers=[
+    'Development Status :: 5 - Production/Stable',      # Chose either "3 - Alpha", "4 - Beta" or "5 - Production/Stable" as the current state of your package
+    'Intended Audience :: Developers',
+    'Topic :: Software Development :: Build Tools',
+    'License :: OSI Approved :: MIT License',
+    'Programming Language :: Python :: 3.7',
+    'Programming Language :: Python :: 3.8',
+  ],
+)