Merging PR_218 openai_rev package with new streamlit chat app

venv/lib/python3.9/site-packages/pydeck/bindings/json_tools.py

@@ -0,0 +1,101 @@
+"""
+Support serializing objects into JSON
+"""
+import json
+
+from pydeck.types.base import PydeckType
+
+# Attributes to ignore during JSON serialization
+IGNORE_KEYS = [
+    "mapbox_key",
+    "google_maps_key",
+    "deck_widget",
+    "binary_data_sets",
+    "_binary_data",
+    "_tooltip",
+    "_kwargs",
+]
+
+
+def to_camel_case(snake_case):
+    """Makes a snake case string into a camel case one
+
+    Parameters
+    -----------
+    snake_case : str
+        Snake-cased string (e.g., "snake_cased") to be converted to camel-case (e.g., "camelCase")
+
+    Returns
+    -------
+    str
+        Camel-cased (e.g., "camelCased") version of input string
+    """
+    output_str = ""
+    should_upper_case = False
+    for i, c in enumerate(snake_case):
+        if c == "_" and i != 0:
+            should_upper_case = True
+            continue
+        output_str = output_str + c.upper() if should_upper_case else output_str + c
+        should_upper_case = False
+    return output_str
+
+
+def lower_first_letter(s):
+    return s[:1].lower() + s[1:] if s else ""
+
+
+def camel_and_lower(w):
+    return lower_first_letter(to_camel_case(w))
+
+
+def lower_camel_case_keys(attrs):
+    """Makes all the keys in a dictionary camel-cased and lower-case
+
+    Parameters
+    ----------
+    attrs : dict
+        Dictionary for which all the keys should be converted to camel-case
+    """
+    for snake_key in list(attrs.keys()):
+        if "_" not in snake_key:
+            continue
+        if snake_key == "_data":
+            camel_key = "data"
+        else:
+            camel_key = camel_and_lower(snake_key)
+        attrs[camel_key] = attrs.pop(snake_key)
+
+
+def default_serialize(o, remap_function=lower_camel_case_keys):
+    """Default method for rendering JSON from a dictionary"""
+    if issubclass(type(o), PydeckType):
+        return repr(o)
+    attrs = vars(o)
+    attrs = {k: v for k, v in attrs.items() if v is not None}
+    for ignore_attr in IGNORE_KEYS:
+        if attrs.get(ignore_attr):
+            del attrs[ignore_attr]
+    if remap_function:
+        remap_function(attrs)
+    return attrs
+
+
+def serialize(serializable):
+    """Takes a serializable object and JSONifies it"""
+    return json.dumps(serializable, sort_keys=True, default=default_serialize, indent=2)
+
+
+class JSONMixin(object):
+    def __repr__(self):
+        """
+        Override of string representation method to return a JSON-ified version of the
+        Deck object.
+        """
+        return serialize(self)
+
+    def to_json(self):
+        """
+        Return a JSON-ified version of the Deck object.
+        """
+        return serialize(self)