add altair

app.py

@@ -15,12 +15,13 @@ app = marimo.App(width="medium")
 
 @app.cell(hide_code=True)
 def __():
+    import os
     import textwrap
 
     import ell
-    import os
-    import openai
     import marimo as mo
+    import openai
+
     return ell, mo, openai, os, textwrap
 
 
@@ -69,9 +70,7 @@ def __():
             # so we first execute the previous statements
             if len(stmts) > 1:
                 exec(
-                    compile(
-                        ast.Module(body=stmts[:-1]), filename="<ast>", mode="exec"
-                    ),
+                    compile(ast.Module(body=stmts[:-1]), filename="<ast>", mode="exec"),
                     globals,
                     locals,
                 )
@@ -88,6 +87,7 @@ def __():
         else:
             # otherwise we just execute the entire code
             return exec(script, globals, locals)
+
     return (exec_with_result,)
 
 
@@ -96,7 +96,6 @@ def __(ell, exec_with_result, mo):
     def code_fence(code):
         return f"```python\n\n{code}\n\n```"
 
-
     @ell.tool()
     def execute_code(code: str):
         """
@@ -113,6 +112,7 @@ def __(ell, exec_with_result, mo):
                 code_fence(result if result is not None else output),
             ]
             return mo.md("\n\n".join(results))
+
     return code_fence, execute_code
 
 
@@ -128,12 +128,12 @@ def __(client, ell, execute_code, mo, model):
             for message in messages
         ]
 
-
     def my_model(messages, config):
         response = custom_chatbot(messages, config)
         if response.tool_calls:
             return response.tool_calls[0]()
         return mo.md(response.text)
+
     return custom_chatbot, my_model