add altair

app.py

@@ -5,6 +5,7 @@
 #     "marimo",
 #     "openai==1.53.0",
 #     "polars==1.12.0",
+#     "altair==5.4.1",
 # ]
 # ///
 
@@ -41,9 +42,7 @@ def __(dataset_input, mo, pl):
     try:
         df = pl.read_csv(dataset_input.value)
         mo.output.replace(
-            mo.md(
-                f"Loaded dataset with {len(df)} rows and {len(df.columns)} columns."
-            )
+            mo.md(f"Loaded dataset with {len(df)} rows and {len(df.columns)} columns.")
         )
     except Exception as e:
         df = pl.DataFrame()
@@ -56,8 +55,10 @@ def __(dataset_input, mo, pl):
 @app.cell
 def __():
     import os
+
     import marimo as mo
     import polars as pl
+
     return mo, os, pl
 
 
@@ -91,7 +92,6 @@ def __(api_key_input, mo):
 def __(df, mo):
     import ell
 
-
     @ell.tool()
     def chart_data(x_encoding: str, y_encoding: str, color: str):
         """Generate an altair chart"""
@@ -104,7 +104,6 @@ def __(df, mo):
             .properties(width=500)
         )
 
-
     @ell.tool()
     def filter_dataset(sql_query: str):
         """
@@ -118,6 +117,7 @@ def __(df, mo):
             selection=None,
             show_column_summaries=False,
         )
+
     return chart_data, ell, filter_dataset
 
 
@@ -128,14 +128,12 @@ def __(chart_data, client, df, ell, filter_dataset, mo):
         """You are a data scientist that can analyze a dataset"""
         return f"I have a dataset with schema: {df.schema}. \n{prompt}"
 
-
     def my_model(messages):
         response = analyze_dataset(messages)
         if response.tool_calls:
             return response.tool_calls[0]()
         return response.text
 
-
     mo.ui.chat(
         my_model,
         prompts=[

requirements.txt

@@ -2,6 +2,7 @@ marimo
 ell-ai==0.0.14
 openai==1.53.0
 polars==1.12.0
+altair==5.4.1
 # Or a specific version
 # marimo>=0.9.0