zaturn 0.1.3__py3-none-any.whl → 0.1.4__py3-none-any.whl

zaturn/visualizations.py

@@ -1,4 +1,5 @@
 from fastmcp import FastMCP, Image
+import math
 import matplotlib.pyplot as plt
 import os
 import seaborn as sns
@@ -7,6 +8,7 @@ from typing import Any, Union, Optional
 from zaturn import config, query_utils
 
 sns.set_theme()
+sns.set_style('ticks')
 
 mcp = FastMCP("Zaturn Visualizations")
 
@@ -14,7 +16,7 @@ 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)
+    figure.savefig(filepath, bbox_inches='tight')
     plt.clf()
     if config.RETURN_IMAGES:
         return Image(path=filepath)
@@ -22,6 +24,47 @@ def _plot_to_image(plot) -> Union[str, Image]:
         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()
@@ -42,6 +85,7 @@ def scatter_plot(
     """
     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)
 
 
@@ -62,6 +106,7 @@ def line_plot(
     """
     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)
 
 
@@ -108,6 +153,7 @@ def strip_plot(
     """
     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)
 
 
@@ -128,6 +174,7 @@ def box_plot(
     """
     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)
 
 
@@ -150,6 +197,7 @@ def bar_plot(
     """
     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)
 
 

{zaturn-0.1.3.dist-info → zaturn-0.1.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: zaturn
-Version: 0.1.3
+Version: 0.1.4
 Summary: AI Data Analysis MCP
 Author-email: Karthik Devan <krtdvn@gmail.com>
 Maintainer-email: Karthik Devan <krtdvn@gmail.com>
@@ -24,21 +24,31 @@ Requires-Dist: sqlalchemy>=2.0.40
 Requires-Dist: tabulate>=0.9.0
 Dynamic: license-file
 
-<p align="center">
-  <img src="https://github.com/kdqed/zaturn/raw/main/brand/logo.png" width="128" height="128">
-</p>
+<h1>
+  <img src="https://github.com/kdqed/zaturn/raw/main/brand/logo.png" width="24" height="24">
+  <span>Zaturn: Your Co-Pilot For Data Analytics & Business Insights</span>
+</h1>
+
+Zaturn let's you analyze your data using AI chat; without needing you to write SQL/Python code or fiddling with aesthetically pleasing (but overwhelming) dashboards. 
+
+You can add Zaturn MCP to Claude Desktop (or any MCP client), connect your data sources, ask questions in natural language, and get instant insights with visualizations. With Zaturn, your AI can automatically understand the kind of data you have, query it, and give you useful pointers with a coherent narrative. You can ask specific questions like "Who is our most valuable customer?", or let AI explore your data with a question like "Here's all the data we have, give us some ideas for the next quarter."
+
+[![PyPI Downloads](https://static.pepy.tech/badge/zaturn)](https://pepy.tech/projects/zaturn) 
+
+[Join The Discord](https://discord.gg/K8mECeVzpQ)
 
-# Zaturn: Your Co-Pilot For Data Analytics & BI
 
 <a href="https://glama.ai/mcp/servers/@kdqed/zaturn">
   <img width="380" height="200" src="https://glama.ai/mcp/servers/@kdqed/zaturn/badge" alt="Zaturn MCP server" />
 </a>
 
-https://github.com/user-attachments/assets/d42dc433-e5ec-4b3e-bef0-5cfc097396ab
+## But I can just upload my data to ChatGPT and ask it to analyze right?
 
-Zaturn is an open source, AI-powered data analysis/BI tool that can connect to your data sources, run SQL queries on it, and give you useful insights. Think of it like vibe data analysis, in the spirit of vibe coding. Currently Zaturn is available as an MCP (Model Context Protocol) Server that can be integrated into your favorite MCP Client (Claude, Cursor, etc.) A full fledged notebook interface is on the roadmap.
+Yes, but ChatGPT has an upload size limit of approximately 50MB for CSV files/spreadsheets, and uploading also takes time. Plus, it can't read data straight from your MySQL or PostgreSQL server. Zaturn can overcome all of these limitations, without moving your data anywhere. It simply equips your AI with SQL and visualization capabilities so AI can query your data directly, get the results, process them, and give you instant insights. With Zaturn, AI does not need to process your full dataset and keep it in its memory to answer your questions about the data.
 
-[Join The Discord](https://discord.gg/K8mECeVzpQ)
+## Zaturn in Action
+
+https://github.com/user-attachments/assets/d42dc433-e5ec-4b3e-bef0-5cfc097396ab
 
 ## Features:
 
@@ -64,12 +74,6 @@ In addition to providing tabular and textual summaries, Zaturn can also generate
 More visualization capabilities are being added.
 
 
-## How Does Zaturn Work?
-
-The naivest way to have an LLM analyze your data is to upload a dataset with a prompt. But that won't get you far, because AI has context window limitations, and it can only go through a few thousand rows at the best. Also, LLM's are not great at doing math.
-
-Using an MCP like Zaturn will keep your data where it is, and enable AI to draft and run SQL queries on the data. The LLM now processes only the queries and results instead of your entire dataset.
-
 ## Installation & Setup
 1. Install [uv](https://docs.astral.sh/uv/getting-started/installation/#installation-methods)
 
@@ -162,3 +166,7 @@ Analyst:
 ## 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.
+
+## Star History
+
+[![Star History Chart](https://api.star-history.com/svg?repos=kdqed/zaturn&type=Date)](https://www.star-history.com/#kdqed/zaturn&Date)

{zaturn-0.1.3.dist-info → zaturn-0.1.4.dist-info}/RECORD

@@ -2,11 +2,11 @@ zaturn/__init__.py,sha256=v4t5fkRuIJFE-SBxCa5pBjZv0EoC0eWK75nU9iaa7Rg,267
 zaturn/config.py,sha256=gF5M6Agmixw2A4vpWqIF3ICVnClPeASA51dhp1bkk04,3221
 zaturn/core.py,sha256=9zcKb0FbkgGkDtWfBA6_O5NQf6GVKG69HrqOr5nhsLU,4641
 zaturn/query_utils.py,sha256=zyQjcRnPKGHZdf0XHzQeMxHw9vieZIwXhBbVGP87ml4,2801
-zaturn/visualizations.py,sha256=Dj08msgZ_DIMEVvJIlybT3cIA3GNkcXVOa9TJsHK2yo,4551
+zaturn/visualizations.py,sha256=4RA4FytYNPiBdMyIqlw3dfjH7flT2Nz6ncVoVoIqPEA,5883
 zaturn/example_data/all_pokemon_data.csv,sha256=SUlGHHWbehuLg-ch1YUrQ6-xBtqHGw6rIkyn70fAgCk,130893
-zaturn-0.1.3.dist-info/licenses/LICENSE,sha256=mZSuFlbEBZGl0-8ULRMLdRDbhau5hrWRNQOjytYeaug,1070
-zaturn-0.1.3.dist-info/METADATA,sha256=BxES8Ifp0fusfDMeB2SNLpSUrZ2qsKWG8rgqhQnfD0Q,5961
-zaturn-0.1.3.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-zaturn-0.1.3.dist-info/entry_points.txt,sha256=N1UZC2zvod92_Brs4A2xZiAnt-iGLBNryglXfwhxfj4,43
-zaturn-0.1.3.dist-info/top_level.txt,sha256=KLUnwQwVZkfd5YCnnqR35MOOs8KLhanPGelvmRo2MVA,7
-zaturn-0.1.3.dist-info/RECORD,,
+zaturn-0.1.4.dist-info/licenses/LICENSE,sha256=mZSuFlbEBZGl0-8ULRMLdRDbhau5hrWRNQOjytYeaug,1070
+zaturn-0.1.4.dist-info/METADATA,sha256=VPtw2YcLYUz54oLKX72sCjzImF7xiqrM65e2u1j2-EE,6658
+zaturn-0.1.4.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
+zaturn-0.1.4.dist-info/entry_points.txt,sha256=N1UZC2zvod92_Brs4A2xZiAnt-iGLBNryglXfwhxfj4,43
+zaturn-0.1.4.dist-info/top_level.txt,sha256=KLUnwQwVZkfd5YCnnqR35MOOs8KLhanPGelvmRo2MVA,7
+zaturn-0.1.4.dist-info/RECORD,,

{zaturn-0.1.3.dist-info → zaturn-0.1.4.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (78.1.0)
+Generator: setuptools (79.0.0)
 Root-Is-Purelib: true
 Tag: py3-none-any