zaturn 0.1.4__tar.gz → 0.1.5__tar.gz

{zaturn-0.1.4/zaturn.egg-info → zaturn-0.1.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: zaturn
-Version: 0.1.4
+Version: 0.1.5
 Summary: AI Data Analysis MCP
 Author-email: Karthik Devan <krtdvn@gmail.com>
 Maintainer-email: Karthik Devan <krtdvn@gmail.com>
@@ -12,8 +12,10 @@ License-File: LICENSE
 Requires-Dist: cryptography>=44.0.2
 Requires-Dist: duckdb>=1.2.1
 Requires-Dist: fastmcp>=0.4.1
+Requires-Dist: kaleido==0.2.1
 Requires-Dist: pandas>=2.2.3
 Requires-Dist: platformdirs>=4.3.7
+Requires-Dist: plotly[express]>=6.0.1
 Requires-Dist: psycopg2-binary>=2.9.10
 Requires-Dist: pyarrow>=19.0.1
 Requires-Dist: pymysql>=1.1.1
@@ -158,11 +160,20 @@ Analyst:
 ```
 - A native notebook interface 
 
-## Support And Feedback
+## Help And Feedback
 
 [Raise an issue](https://github.com/kdqed/zaturn/issues) or [join the Discord](https://discord.gg/K8mECeVzpQ).
 
 
+## Support The Project
+
+If you find Zaturn useful, please support this project by:
+- Starring the Project
+- Spreading the word
+- [Pledging $9/month on Patreon](https://www.patreon.com/kdqed?utm_medium=github&utm_source=join_link&utm_campaign=creatorshare_creator&utm_content=copyLink)
+
+Your support will enable me to dedicate more of my time to Zaturn.
+
 ## Example Dataset Credits
 
 The [pokemon dataset compiled by Sarah Taha and PokéAPI](https://www.kaggle.com/datasets/sarahtaha/1025-pokemon) has been included under the [CC BY-NC-SA 4.0](https://creativecommons.org/licenses/by-nc-sa/4.0/) license for demonstration purposes.

{zaturn-0.1.4 → zaturn-0.1.5}/README.md

@@ -132,11 +132,20 @@ Analyst:
 ```
 - A native notebook interface 
 
-## Support And Feedback
+## Help And Feedback
 
 [Raise an issue](https://github.com/kdqed/zaturn/issues) or [join the Discord](https://discord.gg/K8mECeVzpQ).
 
 
+## Support The Project
+
+If you find Zaturn useful, please support this project by:
+- Starring the Project
+- Spreading the word
+- [Pledging $9/month on Patreon](https://www.patreon.com/kdqed?utm_medium=github&utm_source=join_link&utm_campaign=creatorshare_creator&utm_content=copyLink)
+
+Your support will enable me to dedicate more of my time to Zaturn.
+
 ## Example Dataset Credits
 
 The [pokemon dataset compiled by Sarah Taha and PokéAPI](https://www.kaggle.com/datasets/sarahtaha/1025-pokemon) has been included under the [CC BY-NC-SA 4.0](https://creativecommons.org/licenses/by-nc-sa/4.0/) license for demonstration purposes.

{zaturn-0.1.4 → zaturn-0.1.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "zaturn"
-version = "0.1.4"
+version = "0.1.5"
 description = "AI Data Analysis MCP"
 authors = [
   {name = "Karthik Devan", email = "krtdvn@gmail.com"},
@@ -14,8 +14,10 @@ dependencies = [
     "cryptography>=44.0.2",
     "duckdb>=1.2.1",
     "fastmcp>=0.4.1",
+    "kaleido==0.2.1",
     "pandas>=2.2.3",
     "platformdirs>=4.3.7",
+    "plotly[express]>=6.0.1",
     "psycopg2-binary>=2.9.10",
     "pyarrow>=19.0.1",
     "pymysql>=1.1.1",

{zaturn-0.1.4 → zaturn-0.1.5}/uv.lock

@@ -445,6 +445,19 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/c0/5a/9cac0c82afec3d09ccd97c8b6502d48f165f9124db81b4bcb90b4af974ee/jedi-0.19.2-py2.py3-none-any.whl", hash = "sha256:a8ef22bde8490f57fe5c7681a3c83cb58874daf72b4784de3cce5b6ef6edb5b9", size = 1572278 },
 ]
 
+[[package]]
+name = "kaleido"
+version = "0.2.1"
+source = { registry = "https://pypi.org/simple" }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/e0/f7/0ccaa596ec341963adbb4f839774c36d5659e75a0812d946732b927d480e/kaleido-0.2.1-py2.py3-none-macosx_10_11_x86_64.whl", hash = "sha256:ca6f73e7ff00aaebf2843f73f1d3bacde1930ef5041093fe76b83a15785049a7", size = 85153681 },
+    { url = "https://files.pythonhosted.org/packages/45/8e/4297556be5a07b713bb42dde0f748354de9a6918dee251c0e6bdcda341e7/kaleido-0.2.1-py2.py3-none-macosx_11_0_arm64.whl", hash = "sha256:bb9a5d1f710357d5d432ee240ef6658a6d124c3e610935817b4b42da9c787c05", size = 85808197 },
+    { url = "https://files.pythonhosted.org/packages/ae/b3/a0f0f4faac229b0011d8c4a7ee6da7c2dca0b6fd08039c95920846f23ca4/kaleido-0.2.1-py2.py3-none-manylinux1_x86_64.whl", hash = "sha256:aa21cf1bf1c78f8fa50a9f7d45e1003c387bd3d6fe0a767cfbbf344b95bdc3a8", size = 79902476 },
+    { url = "https://files.pythonhosted.org/packages/a1/2b/680662678a57afab1685f0c431c2aba7783ce4344f06ec162074d485d469/kaleido-0.2.1-py2.py3-none-manylinux2014_aarch64.whl", hash = "sha256:845819844c8082c9469d9c17e42621fbf85c2b237ef8a86ec8a8527f98b6512a", size = 83711746 },
+    { url = "https://files.pythonhosted.org/packages/88/89/4b6f8bb3f9ab036fd4ad1cb2d628ab5c81db32ac9aa0641d7b180073ba43/kaleido-0.2.1-py2.py3-none-win32.whl", hash = "sha256:ecc72635860be616c6b7161807a65c0dbd9b90c6437ac96965831e2e24066552", size = 62312480 },
+    { url = "https://files.pythonhosted.org/packages/f7/9a/0408b02a4bcb3cf8b338a2b074ac7d1b2099e2b092b42473def22f7b625f/kaleido-0.2.1-py2.py3-none-win_amd64.whl", hash = "sha256:4670985f28913c2d063c5734d125ecc28e40810141bdb0a46f15b76c1d45f23c", size = 65945521 },
+]
+
 [[package]]
 name = "kiwisolver"
 version = "1.4.8"
@@ -594,6 +607,15 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/b3/38/89ba8ad64ae25be8de66a6d463314cf1eb366222074cfda9ee839c56a4b4/mdurl-0.1.2-py3-none-any.whl", hash = "sha256:84008a41e51615a49fc9966191ff91509e3c40b939176e643fd50a5c2196b8f8", size = 9979 },
 ]
 
+[[package]]
+name = "narwhals"
+version = "1.35.0"
+source = { registry = "https://pypi.org/simple" }
+sdist = { url = "https://files.pythonhosted.org/packages/ee/6a/a98fa5e9d530a428a0cd79d27f059ed65efd3a07aad61a8c93e323c9c20b/narwhals-1.35.0.tar.gz", hash = "sha256:07477d18487fbc940243b69818a177ed7119b737910a8a254fb67688b48a7c96", size = 265784 }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/80/b3/5781eb874f04cb1e882a7d93cf30abcb00362a3205c5f3708a7434a1a2ac/narwhals-1.35.0-py3-none-any.whl", hash = "sha256:7562af132fa3f8aaaf34dc96d7ec95bdca29d1c795e8fcf14e01edf1d32122bc", size = 325708 },
+]
+
 [[package]]
 name = "numpy"
 version = "2.2.4"
@@ -781,6 +803,24 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/6d/45/59578566b3275b8fd9157885918fcd0c4d74162928a5310926887b856a51/platformdirs-4.3.7-py3-none-any.whl", hash = "sha256:a03875334331946f13c549dbd8f4bac7a13a50a895a0eb1e8c6a8ace80d40a94", size = 18499 },
 ]
 
+[[package]]
+name = "plotly"
+version = "6.0.1"
+source = { registry = "https://pypi.org/simple" }
+dependencies = [
+    { name = "narwhals" },
+    { name = "packaging" },
+]
+sdist = { url = "https://files.pythonhosted.org/packages/c7/cc/e41b5f697ae403f0b50e47b7af2e36642a193085f553bf7cc1169362873a/plotly-6.0.1.tar.gz", hash = "sha256:dd8400229872b6e3c964b099be699f8d00c489a974f2cfccfad5e8240873366b", size = 8094643 }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/02/65/ad2bc85f7377f5cfba5d4466d5474423a3fb7f6a97fd807c06f92dd3e721/plotly-6.0.1-py3-none-any.whl", hash = "sha256:4714db20fea57a435692c548a4eb4fae454f7daddf15f8d8ba7e1045681d7768", size = 14805757 },
+]
+
+[package.optional-dependencies]
+express = [
+    { name = "numpy" },
+]
+
 [[package]]
 name = "pluggy"
 version = "1.5.0"
@@ -1329,14 +1369,16 @@ wheels = [
 
 [[package]]
 name = "zaturn"
-version = "0.1.3"
+version = "0.1.5"
 source = { editable = "." }
 dependencies = [
     { name = "cryptography" },
     { name = "duckdb" },
     { name = "fastmcp" },
+    { name = "kaleido" },
     { name = "pandas" },
     { name = "platformdirs" },
+    { name = "plotly", extra = ["express"] },
     { name = "psycopg2-binary" },
     { name = "pyarrow" },
     { name = "pymysql" },
@@ -1352,8 +1394,10 @@ requires-dist = [
     { name = "cryptography", specifier = ">=44.0.2" },
     { name = "duckdb", specifier = ">=1.2.1" },
     { name = "fastmcp", specifier = ">=0.4.1" },
+    { name = "kaleido", specifier = "==0.2.1" },
     { name = "pandas", specifier = ">=2.2.3" },
     { name = "platformdirs", specifier = ">=4.3.7" },
+    { name = "plotly", extras = ["express"], specifier = ">=6.0.1" },
     { name = "psycopg2-binary", specifier = ">=2.9.10" },
     { name = "pyarrow", specifier = ">=19.0.1" },
     { name = "pymysql", specifier = ">=1.1.1" },

zaturn-0.1.5/zaturn/visualizations.py

@@ -0,0 +1,155 @@
+from fastmcp import FastMCP, Image
+import math
+import os
+import plotly.express as px
+import time
+from typing import Any, Union, Optional
+from zaturn import config, query_utils
+
+
+mcp = FastMCP("Zaturn Visualizations")
+
+
+def _fig_to_image(fig) -> Union[str, Image]:
+    filepath = os.path.join(config.VISUALS_DIR, str(int(time.time())) + '.png')
+    fig.write_image(filepath)
+    if config.RETURN_IMAGES:
+        return Image(path=filepath)
+    else:
+        return filepath
+        
+
+# Relationships
+
+@mcp.tool()
+def scatter_plot(
+    query_id: str,
+    x: str,
+    y: str,
+    color: str = None
+    ):
+    """
+    Make a scatter plot with the dataframe obtained from running SQL Query against source
+    If this returns an image, display it. If it returns a file path, mention it.
+    Args:
+        query_id: Previously run query to use for plotting
+        x: Column name from SQL result to use for x-axis
+        y: Column name from SQL result to use for y-axis
+        color: Optional; column name from SQL result to use for coloring the points, with color representing another dimension
+    """
+    df = query_utils.load_query(query_id)
+    fig = px.scatter(df, x=x, y=y, color=color)
+    fig.update_xaxes(autotickangles=[0, 45, 60, 90])
+    return _fig_to_image(fig)
+
+
+@mcp.tool()
+def line_plot(
+    query_id: str,
+    x: str,
+    y: str,
+    color: str = None
+    ):
+    """
+    Make a line plot with the dataframe obtained from running SQL Query against source
+    Args:
+        query_id: Previously run query to use for plotting
+        x: Column name from SQL result to use for x-axis
+        y: Column name from SQL result to use for y-axis
+        color: Optional; column name from SQL result to use for drawing multiple colored lines representing another dimension
+    """
+    df = query_utils.load_query(query_id)
+    fig = px.line(df, x=x, y=y, color=color)
+    fig.update_xaxes(autotickangles=[0, 45, 60, 90])
+    return _fig_to_image(fig)
+
+
+# Distributions
+
+@mcp.tool()
+def histogram(
+    query_id: str,
+    column: str,
+    color: str = None,
+    nbins: int = None
+    ):
+    """
+    Make a histogram with a column of the dataframe obtained from running SQL Query against source
+    Args:
+        query_id: Previously run query to use for plotting
+        column: Column name from SQL result to use for the histogram
+        color: Optional; column name from SQL result to use for drawing multiple colored histograms representing another dimension
+        nbins: Optional; number of bins
+    """
+    df = query_utils.load_query(query_id)
+    fig = px.histogram(df, x=column, color=color, nbins=nbins)
+    fig.update_xaxes(autotickangles=[0, 45, 60, 90])
+    return _fig_to_image(fig)
+
+# Categorical
+
+@mcp.tool()
+def strip_plot(
+    query_id: str,
+    x: str,
+    y: str = None,
+    color: str = None
+    ):
+    """
+    Make a strip plot with the dataframe obtained from running SQL Query against source
+    Args:
+        query_id: Previously run query to use for plotting
+        x: Column name from SQL result to use for x axis
+        y: Optional; column name from SQL result to use for y axis
+        color: Optional column name from SQL result to show multiple colored strips representing another dimension
+    """
+    df = query_utils.load_query(query_id)
+    fig = px.strip(df, x=x, y=y, color=color)
+    fig.update_xaxes(autotickangles=[0, 45, 60, 90])
+    return _fig_to_image(fig)
+
+
+@mcp.tool()
+def box_plot(
+    query_id: str,
+    y: str,
+    x: str = None,
+    color: str = None
+    ):
+    """
+    Make a box plot with the dataframe obtained from running SQL Query against source
+    Args:
+        query_id: Previously run query to use for plotting
+        y: Column name from SQL result to use for y axis
+        x: Optional; Column name from SQL result to use for x axis
+        color: Optional column name from SQL result to show multiple colored bars representing another dimension
+    """
+    df = query_utils.load_query(query_id)
+    fig = px.box(df, x=x, y=y, color=color)
+    fig.update_xaxes(autotickangles=[0, 45, 60, 90])
+    return _fig_to_image(fig)
+
+
+@mcp.tool()
+def bar_plot(
+    query_id: str,
+    x: str,
+    y: str = None,
+    color: str = None,
+    orientation: str = 'v'
+    ):
+    """
+    Make a bar plot with the dataframe obtained from running SQL Query against source
+    Args:
+        query_id: Previously run query to use for plotting
+        x: Column name from SQL result to use for x axis
+        y: Optional; column name from SQL result to use for y axis
+        color: Optional column name from SQL result to use as a 3rd dimension by splitting each bar into colored sections
+        orientation: Orientation of the box plot, use 'v' for vertical (default) and 'h' for horizontal. Be mindful of choosing the correct X and Y columns as per orientation
+    """
+    df = query_utils.load_query(query_id)
+    fig = px.bar(df, x=x, y=y, color=color, orientation=orientation)
+    fig.update_xaxes(autotickangles=[0, 45, 60, 90])
+    return _fig_to_image(fig)
+
+

{zaturn-0.1.4 → zaturn-0.1.5/zaturn.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: zaturn
-Version: 0.1.4
+Version: 0.1.5
 Summary: AI Data Analysis MCP
 Author-email: Karthik Devan <krtdvn@gmail.com>
 Maintainer-email: Karthik Devan <krtdvn@gmail.com>
@@ -12,8 +12,10 @@ License-File: LICENSE
 Requires-Dist: cryptography>=44.0.2
 Requires-Dist: duckdb>=1.2.1
 Requires-Dist: fastmcp>=0.4.1
+Requires-Dist: kaleido==0.2.1
 Requires-Dist: pandas>=2.2.3
 Requires-Dist: platformdirs>=4.3.7
+Requires-Dist: plotly[express]>=6.0.1
 Requires-Dist: psycopg2-binary>=2.9.10
 Requires-Dist: pyarrow>=19.0.1
 Requires-Dist: pymysql>=1.1.1
@@ -158,11 +160,20 @@ Analyst:
 ```
 - A native notebook interface 
 
-## Support And Feedback
+## Help And Feedback
 
 [Raise an issue](https://github.com/kdqed/zaturn/issues) or [join the Discord](https://discord.gg/K8mECeVzpQ).
 
 
+## Support The Project
+
+If you find Zaturn useful, please support this project by:
+- Starring the Project
+- Spreading the word
+- [Pledging $9/month on Patreon](https://www.patreon.com/kdqed?utm_medium=github&utm_source=join_link&utm_campaign=creatorshare_creator&utm_content=copyLink)
+
+Your support will enable me to dedicate more of my time to Zaturn.
+
 ## Example Dataset Credits
 
 The [pokemon dataset compiled by Sarah Taha and PokéAPI](https://www.kaggle.com/datasets/sarahtaha/1025-pokemon) has been included under the [CC BY-NC-SA 4.0](https://creativecommons.org/licenses/by-nc-sa/4.0/) license for demonstration purposes.

{zaturn-0.1.4 → zaturn-0.1.5}/zaturn.egg-info/requires.txt

@@ -1,8 +1,10 @@
 cryptography>=44.0.2
 duckdb>=1.2.1
 fastmcp>=0.4.1
+kaleido==0.2.1
 pandas>=2.2.3
 platformdirs>=4.3.7
+plotly[express]>=6.0.1
 psycopg2-binary>=2.9.10
 pyarrow>=19.0.1
 pymysql>=1.1.1

zaturn-0.1.4/zaturn/visualizations.py

@@ -1,203 +0,0 @@
-from fastmcp import FastMCP, Image
-import math
-import matplotlib.pyplot as plt
-import os
-import seaborn as sns
-import time
-from typing import Any, Union, Optional
-from zaturn import config, query_utils
-
-sns.set_theme()
-sns.set_style('ticks')
-
-mcp = FastMCP("Zaturn Visualizations")
-
-
-def _plot_to_image(plot) -> Union[str, Image]:
-    figure = plot.get_figure()
-    filepath = os.path.join(config.VISUALS_DIR, str(int(time.time())) + '.png')
-    figure.savefig(filepath, bbox_inches='tight')
-    plt.clf()
-    if config.RETURN_IMAGES:
-        return Image(path=filepath)
-    else:
-        return filepath
-
-
-def _fix_x_labels(plot, labels):
-    max_label_length = max(list(labels.map(lambda x: len(str(x)))))
-
-    LABEL_HIDE_FACTOR = 1
-    if len(labels) > 20:
-        LABEL_HIDE_FACTOR = math.ceil(len(labels)/20)
-
-    labels_to_show = list(labels)
-    ticks = list(plot.get_xticks()) 
-    if LABEL_HIDE_FACTOR > 1:
-        ticks = ticks[::LABEL_HIDE_FACTOR]
-        labels_to_show = labels_to_show[::LABEL_HIDE_FACTOR]
-            
-    plot.set_xticks(ticks, labels_to_show)
-    cutoff = 2 # for rotation
-    
-    if max_label_length >= 12:
-        cutoff = 3
-    elif max_label_length >= 10:
-        cutoff = 4
-    elif max_label_length >= 8:
-        cutoff = 5
-    elif max_label_length >= 7:
-        cutoff = 5
-    elif max_label_length >= 6:
-        cutoff = 6
-    elif max_label_length >= 5:
-        cutoff = 7
-    elif max_label_length >= 4:
-        cutoff = 9
-    elif max_label_length >= 3:
-        cutoff = 13
-    else:
-        cutoff = 15
-    
-    if len(labels)>cutoff:
-        plot.set_xticklabels(plot.get_xticklabels(), rotation=-45, ha='left', va='top')
-    
-    return plot
-
-
-# Relationships
-
-@mcp.tool()
-def scatter_plot(
-    query_id: str,
-    x: str,
-    y: str,
-    hue: str = None
-    ):
-    """
-    Make a scatter plot with the dataframe obtained from running SQL Query against source
-    If this returns an image, display it. If it returns a file path, mention it.
-    Args:
-        query_id: Previously run query to use for plotting
-        x: Column name from SQL result to use for x-axis
-        y: Column name from SQL result to use for y-axis
-        hue: Optional String; Column name from SQL result to use for coloring the points  
-    """
-    df = query_utils.load_query(query_id)
-    plot = sns.scatterplot(df, x=x, y=y, hue=hue)
-    plot = _fix_x_labels(plot, df[x])
-    return _plot_to_image(plot)
-
-
-@mcp.tool()
-def line_plot(
-    query_id: str,
-    x: str,
-    y: str,
-    hue: str = None
-    ):
-    """
-    Make a line plot with the dataframe obtained from running SQL Query against source
-    Args:
-        query_id: Previously run query to use for plotting
-        x: Column name from SQL result to use for x-axis
-        y: Column name from SQL result to use for y-axis
-        hue: Optional; column name from SQL result to use for drawing multiple colored lines
-    """
-    df = query_utils.load_query(query_id)
-    plot = sns.lineplot(df, x=x, y=y, hue=hue)
-    plot = _fix_x_labels(plot, df[x])
-    return _plot_to_image(plot)
-
-
-# Distributions
-
-@mcp.tool()
-def histogram(
-    query_id: str,
-    column: str,
-    hue: str = None,
-    bins: int = None
-    ):
-    """
-    Make a histogram with a column of the dataframe obtained from running SQL Query against source
-    Args:
-        query_id: Previously run query to use for plotting
-        column: Column name from SQL result to use for the histogram
-        hue: Optional; column name from SQL result to use for drawing multiple colored histograms
-        bins: Optional; number of bins
-    """
-    df = query_utils.load_query(query_id)
-    plot = sns.histplot(df, x=column, hue=hue, bins=bins)
-    return _plot_to_image(plot)
-
-
-# Categorical
-
-@mcp.tool()
-def strip_plot(
-    query_id: str,
-    x: str,
-    y: str = None,
-    hue: str = None,
-    legend: bool = False
-    ):
-    """
-    Make a strip plot with the dataframe obtained from running SQL Query against source
-    Args:
-        query_id: Previously run query to use for plotting
-        x: Column name from SQL result to use for x axis
-        y: Optional; column name from SQL result to use for y axis
-        hue: Optional; column name from SQL result to use for coloring the points
-        legend: Whether to draw a legend for the hue
-    """
-    df = query_utils.load_query(query_id)
-    plot = sns.stripplot(df, x=x, y=y, hue=hue, legend=legend)
-    plot = _fix_x_labels(plot, df[x])
-    return _plot_to_image(plot)
-
-
-@mcp.tool()
-def box_plot(
-    query_id: str,
-    x: str,
-    y: str = None,
-    hue: str = None
-    ):
-    """
-    Make a box plot with the dataframe obtained from running SQL Query against source
-    Args:
-        query_id: Previously run query to use for plotting
-        x: Column name from SQL result to use for x axis
-        y: Optional; column name from SQL result to use for y axis
-        hue: Optional column name from SQL result to use for coloring the points
-    """
-    df = query_utils.load_query(query_id)
-    plot = sns.boxplot(df, x=x, y=y, hue=hue)
-    plot = _fix_x_labels(plot, df[x])
-    return _plot_to_image(plot)
-
-
-@mcp.tool()
-def bar_plot(
-    query_id: str,
-    x: str,
-    y: str = None,
-    hue: str = None,
-    orient: str = 'v'
-    ):
-    """
-    Make a bar plot with the dataframe obtained from running SQL Query against source
-    Args:
-        query_id: Previously run query to use for plotting
-        x: Column name from SQL result to use for x axis
-        y: Optional; column name from SQL result to use for y axis
-        hue: Optional column name from SQL result to use for coloring the bars
-        orient: Orientation of the box plot, use 'v' for vertical and 'h' for horizontal
-    """
-    df = query_utils.load_query(query_id)
-    plot = sns.barplot(df, x=x, y=y, hue=hue, orient=orient)
-    plot = _fix_x_labels(plot, df[x])
-    return _plot_to_image(plot)
-
-