Initial commit

Dockerfile

@@ -0,0 +1,18 @@
+# read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
+# you will also find guides on how best to write your Dockerfile
+
+FROM python:3.12
+
+RUN useradd -m -u 1000 user
+
+WORKDIR /app
+
+COPY --chown=user ./requirements.txt requirements.txt
+
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
+
+COPY --chown=user . /app
+
+EXPOSE 7860
+
+CMD ["gunicorn", "-w", "4", "-b", "0.0.0.0:7860", "app:server"]

README.md

@@ -1,8 +1,8 @@
 ---
 title: Vizro Ai UI
-emoji: 📉
+emoji: 🌖
 colorFrom: blue
-colorTo: purple
+colorTo: pink
 sdk: docker
 pinned: false
 license: apache-2.0

_utils.py

@@ -0,0 +1,8 @@
+"""Utils file."""
+
+
+def check_file_extension(filename):
+    filename = filename.lower()
+
+    # Check if the filename ends with .csv or .xls
+    return filename.endswith(".csv") or filename.endswith(".xls") or filename.endswith(".xlsx")

actions.py

@@ -0,0 +1,129 @@
+"""Custom actions used within a dashboard."""
+
+import base64
+import io
+import logging
+
+import black
+import pandas as pd
+from _utils import check_file_extension
+from dash.exceptions import PreventUpdate
+from langchain_openai import ChatOpenAI
+from plotly import graph_objects as go
+from vizro.models.types import capture
+from vizro_ai import VizroAI
+
+logger = logging.getLogger(__name__)
+logger.setLevel(logging.INFO) #TODO: remove manual setting and make centrally controlled
+
+SUPPORTED_VENDORS = {"OpenAI": ChatOpenAI}
+
+
+def get_vizro_ai_plot(user_prompt, df, model, api_key, api_base, vendor_input):  # noqa: PLR0913
+    """VizroAi plot configuration."""
+    vendor = SUPPORTED_VENDORS[vendor_input]
+    llm = vendor(model_name=model, openai_api_key=api_key, openai_api_base=api_base)
+    vizro_ai = VizroAI(model=llm)
+    ai_outputs = vizro_ai.plot(df, user_prompt, explain=False, return_elements=True)
+
+    return ai_outputs
+
+
+@capture("action")
+def run_vizro_ai(user_prompt, n_clicks, data, model, api_key, api_base, vendor_input):  # noqa: PLR0913
+    """Gets the AI response and adds it to the text window."""
+
+    def create_response(ai_response, figure, user_prompt, filename):
+        plotly_fig = figure.to_json()
+        return (
+            ai_response,
+            figure,
+            {"ai_response": ai_response, "figure": plotly_fig, "prompt": user_prompt, "filename": filename},
+        )
+
+    if not n_clicks:
+        raise PreventUpdate
+
+    if not data:
+        ai_response = "Please upload data to proceed!"
+        figure = go.Figure()
+        return create_response(ai_response, figure, user_prompt, None)
+
+    if not api_key:
+        ai_response = "API key not found. Make sure you enter your API key!"
+        figure = go.Figure()
+        return create_response(ai_response, figure, user_prompt, data["filename"])
+
+    if api_key.startswith('"'):
+        ai_response = "Make sure you enter your API key without quotes!"
+        figure = go.Figure()
+        return create_response(ai_response, figure, user_prompt, data["filename"])
+
+    if api_base is not None and api_base.startswith('"'):
+        ai_response = "Make sure you enter your API base without quotes!"
+        figure = go.Figure()
+        return create_response(ai_response, figure, user_prompt, data["filename"])
+
+    try:
+        logger.info("Attempting chart code.")
+        df = pd.DataFrame(data["data"])
+        ai_outputs = get_vizro_ai_plot(
+            user_prompt=user_prompt,
+            df=df,
+            model=model,
+            api_key=api_key,
+            api_base=api_base,
+            vendor_input=vendor_input,
+        )
+        ai_code = ai_outputs.code
+        figure = ai_outputs.figure
+        formatted_code = black.format_str(ai_code, mode=black.Mode(line_length=100))
+
+        ai_response = "\n".join(["```python", formatted_code, "```"])
+        logger.info("Successful query produced.")
+        return create_response(ai_response, figure, user_prompt, data["filename"])
+
+    except Exception as exc:
+        logger.debug(exc)
+        logger.info("Chart creation failed.")
+        ai_response = f"Sorry, I can't do that. Following Error occurred: {exc}"
+        figure = go.Figure()
+        return create_response(ai_response, figure, user_prompt, data["filename"])
+
+
+@capture("action")
+def data_upload_action(contents, filename):
+    """Custom data upload action."""
+    if not contents:
+        raise PreventUpdate
+
+    if not check_file_extension(filename=filename):
+        return {"error_message": "Unsupported file extension.. Make sure to upload either csv or an excel file."}
+
+    content_type, content_string = contents.split(",")
+
+    try:
+        decoded = base64.b64decode(content_string)
+        if filename.endswith(".csv"):
+            # Handle CSV file
+            df = pd.read_csv(io.StringIO(decoded.decode("utf-8")))
+        else:
+            # Handle Excel file
+            df = pd.read_excel(io.BytesIO(decoded))
+
+        data = df.to_dict("records")
+        return {"data": data, "filename": filename}
+
+    except Exception as e:
+        logger.debug(e)
+        return {"error_message": "There was an error processing this file."}
+
+
+@capture("action")
+def display_filename(data):
+    """Custom action to display uploaded filename."""
+    if data is None:
+        raise PreventUpdate
+
+    display_message = data.get("filename") or data.get("error_message")
+    return f"Uploaded file name: '{display_message}'" if "filename" in data else display_message

app.py

@@ -0,0 +1,199 @@
+"""VizroAI UI dashboard configuration."""
+
+import json
+
+import dash_bootstrap_components as dbc
+import pandas as pd
+import vizro.models as vm
+import vizro.plotly.express as px
+from actions import data_upload_action, display_filename, run_vizro_ai
+from components import (
+    CodeClipboard,
+    CustomDashboard,
+    Icon,
+    MyDropdown,
+    MyPage,
+    OffCanvas,
+    UserPromptTextArea,
+    UserUpload,
+    Modal
+)
+from dash import Input, Output, State, callback, get_asset_url, html
+from dash.exceptions import PreventUpdate
+from vizro import Vizro
+
+vm.Container.add_type("components", UserUpload)
+vm.Container.add_type("components", MyDropdown)
+vm.Container.add_type("components", OffCanvas)
+vm.Container.add_type("components", CodeClipboard)
+vm.Container.add_type("components", Icon)
+vm.Container.add_type("components", Modal)
+
+MyPage.add_type("components", UserPromptTextArea)
+MyPage.add_type("components", UserUpload)
+MyPage.add_type("components", MyDropdown)
+MyPage.add_type("components", OffCanvas)
+MyPage.add_type("components", CodeClipboard)
+MyPage.add_type("components", Icon)
+MyPage.add_type("components", Modal)
+
+
+SUPPORTED_MODELS = [
+    "gpt-4o-mini",
+    "gpt-4",
+    "gpt-4-turbo",
+    "gpt-3.5-turbo",
+    "gpt-4o",
+]
+
+
+plot_page = MyPage(
+    id="vizro_ai_plot_page",
+    title="Vizro-AI - effortlessly create interactive charts with Plotly",
+    layout=vm.Layout(
+        grid=[
+            [3, 3, -1, 5],
+            [1, 1, 2, 2],
+            [4, 4, 2, 2],
+            *[[0, 0, 2, 2]] * 6,
+        ]
+    ),
+    components=[
+        vm.Container(title="", components=[CodeClipboard(id="plot")]),
+        UserPromptTextArea(
+            id="text-area-id",
+        ),
+        vm.Graph(id="graph-id", figure=px.scatter(pd.DataFrame())),
+        vm.Container(
+            title="",
+            layout=vm.Layout(grid=[[1], [0]], row_gap="0px"),
+            components=[
+                UserUpload(
+                    id="data-upload-id",
+                    actions=[
+                        vm.Action(
+                            function=data_upload_action(),
+                            inputs=["data-upload-id.contents", "data-upload-id.filename"],
+                            outputs=["data-store-id.data"],
+                        ),
+                        vm.Action(
+                            function=display_filename(),
+                            inputs=["data-store-id.data"],
+                            outputs=["upload-message-id.children"],
+                        ),
+                    ],
+                ),
+                vm.Card(id="upload-message-id", text="Upload your data file (csv or excel)"),
+            ],
+        ),
+        vm.Container(
+            title="",
+            layout=vm.Layout(grid=[[2, 3, -1, -1, -1, 1, 1, 0, 0]], row_gap="0px", col_gap="4px"),
+            components=[
+                vm.Button(
+                    id="trigger-button-id",
+                    text="Run VizroAI",
+                    actions=[
+                        vm.Action(
+                            function=run_vizro_ai(),
+                            inputs=[
+                                "text-area-id.value",
+                                "trigger-button-id.n_clicks",
+                                "data-store-id.data",
+                                "model-dropdown-id.value",
+                                "settings-api-key.value",
+                                "settings-api-base.value",
+                                "settings-dropdown.value",
+                            ],
+                            outputs=["plot-code-markdown.children", "graph-id.figure", "outputs-store-id.data"],
+                        ),
+                    ],
+                ),
+                MyDropdown(options=SUPPORTED_MODELS, value="gpt-4o-mini", multi=False, id="model-dropdown-id"),
+                OffCanvas(id="settings", options=["OpenAI"], value="OpenAI"),
+                Modal(id="modal"),
+            ],
+        ),
+        Icon(id="open-settings-id"),
+    ],
+)
+
+
+dashboard = CustomDashboard(pages=[plot_page])
+
+
+# pure dash callbacks
+@callback(
+    [
+        Output("plot-code-markdown", "children", allow_duplicate=True),
+        Output("graph-id", "figure", allow_duplicate=True),
+        Output("text-area-id", "value"),
+        Output("upload-message-id", "children"),
+    ],
+    [Input("on_page_load_action_trigger_vizro_ai_plot_page", "data")],
+    [State("outputs-store-id", "data")],
+    prevent_initial_call="initial_duplicate",
+)
+def update_data(page_data, outputs_data):
+    """Callback for retrieving latest vizro-ai output from dcc store."""
+    if not outputs_data:
+        raise PreventUpdate
+
+    ai_response = outputs_data["ai_response"]
+    fig = json.loads(outputs_data["figure"])
+    filename = f"File uploaded: '{outputs_data['filename']}'"
+    prompt = outputs_data["prompt"]
+
+    return ai_response, fig, prompt, filename
+
+
+@callback(
+    Output("settings", "is_open"),
+    Input("open-settings-id", "n_clicks"),
+    [State("settings", "is_open")],
+)
+def open_settings(n_clicks, is_open):
+    """Callback for opening and closing offcanvas settings component."""
+    return not is_open if n_clicks else is_open
+
+
+@callback(
+    Output("settings-api-key", "type"),
+    Input("settings-api-key-toggle", "value"),
+)
+def show_api_key(value):
+    """Callback to show api key."""
+    return "text" if value else "password"
+
+
+@callback(
+    Output("settings-api-base", "type"),
+    Input("settings-api-base-toggle", "value"),
+)
+def show_api_base(value):
+    """Callback to show api base."""
+    return "text" if value else "password"
+
+
+app = Vizro().build(dashboard)
+app.dash.layout.children.append(
+    html.Div(
+        [
+            dbc.NavLink("Contact Vizro", href="https://github.com/mckinsey/vizro/issues"),
+            dbc.NavLink("GitHub", href="https://github.com/mckinsey/vizro"),
+            dbc.NavLink("Docs", href="https://vizro.readthedocs.io/projects/vizro-ai/"),
+            html.Div(
+                [
+                    "Made using ",
+                    html.Img(src=get_asset_url("logo.svg"), id="banner", alt="Vizro logo"),
+                    "vizro",
+                ],
+            ),
+        ],
+        className="anchor-container",
+    )
+)
+
+
+if __name__ == "__main__":
+    app.run()

assets/custom_css.css

@@ -0,0 +1,159 @@
+#page-header {
+  display: none;
+}
+
+.card-body {
+  color: var(--text-light-mode-secondary);
+}
+
+.textbox {
+  border-radius: 24px;
+  font-size: var(--text-size-02);
+  margin-bottom: 20px;
+  max-width: 60%;
+  padding: 4px 12px;
+  width: max-content;
+}
+
+.user_input:focus {
+  background: var(--field-enabled);
+  box-shadow: 0 0 0 2px var(--focus-focus) inset;
+  color: var(--text-primary);
+  outline-width: 0;
+}
+
+#text-area-id {
+  background-color: inherit;
+  border: 1px solid var(--border-subtle-alpha-01);
+  color: var(--text-primary);
+  min-height: 90px;
+  padding: 8px;
+  width: 100%;
+}
+
+#code-clipboard {
+  padding: 8px;
+}
+
+.code-clipboard {
+  font-size: 20px;
+  position: absolute;
+  right: 14px;
+  top: 12px;
+}
+
+.code-clipboard-container {
+  background: var(--surfaces-bg-card);
+  font-family: monospace;
+  height: 500px;
+  max-height: 500px;
+  overflow: auto;
+  padding: 1rem;
+  position: relative;
+}
+
+.code-clipboard-container::-webkit-scrollbar-thumb {
+  border-color: var(--surfaces-bg-card);
+}
+
+#model-dropdown .Select--single .Select-value {
+  background-color: inherit;
+  font-size: 12px;
+}
+
+#model-dropdown .Select-control {
+  background-color: inherit;
+  font-size: 12px;
+}
+
+#model-dropdown-id .Select-menu-outer {
+  font-size: 12px;
+
+  /* top: 0; */
+
+  /* transform: translateY(3px) translateY(-100%); */
+}
+
+#model-dropdow-idn .dash-dropdown {
+  background-color: inherit;
+  font-size: 12px;
+}
+
+#trigger-button-id {
+  width: 100%;
+}
+
+#dashboard-container .dash-dropdown {
+  background-color: inherit;
+}
+
+#model-dropdown-id .Select-clear {
+  display: none;
+}
+
+#save-button-id {
+  width: 50%;
+}
+
+.card:has(#upload-message-id) {
+  background-color: inherit;
+  box-shadow: none;
+  font-size: 12px;
+  overflow: hidden;
+  padding-bottom: 0;
+  padding-left: 0;
+  padding-top: 8px;
+}
+
+.card:has(#settings-card-id) {
+  background-color: inherit;
+  box-shadow: none;
+  font-size: 12px;
+  overflow: hidden;
+  padding-bottom: 0;
+  padding-left: 0;
+}
+
+.settings-div {
+  display: flex;
+  justify-content: end;
+  padding-right: 2px;
+  width: 100%;
+}
+
+#data-upload-id {
+  border: 1px dashed var(--border-subtle-alpha-01);
+  border-radius: 5px;
+  color: var(--text-primary);
+  height: 46px;
+  line-height: 46px;
+  text-align: center;
+}
+
+#settings-api-key-toggle .form-check-input {
+  border-radius: 8px;
+}
+
+#settings-api-base-toggle .form-check-input {
+  border-radius: 8px;
+}
+
+#toggle-div-api-base,
+#toggle-div-api-key {
+  align-items: center;
+  display: flex;
+  gap: 4px;
+  justify-content: center;
+}
+
+.anchor-container {
+  background: #060a17;
+  bottom: 0;
+  display: flex;
+  font-weight: 600;
+  gap: 2rem;
+  padding: 4px;
+  place-content: baseline center;
+  position: fixed;
+  width: 100%;
+}

components.py

@@ -0,0 +1,244 @@
+"""Contains custom components used within a dashboard."""
+
+from typing import List, Literal
+
+import black
+import dash_bootstrap_components as dbc
+import vizro.models as vm
+from dash import dcc, html
+from pydantic import PrivateAttr
+from vizro.models import Action
+from vizro.models._action._actions_chain import _action_validator_factory
+from vizro.models._models_utils import _log_call
+
+
+class UserPromptTextArea(vm.VizroBaseModel):
+    """Input component `UserPromptTextArea`.
+
+    Based on the underlying [`dcc.Input`](https://dash.plotly.com/dash-core-components/input).
+
+    Args:
+        type (Literal["user_input"]): Defaults to `"user_text_area"`.
+        title (str): Title to be displayed. Defaults to `""`.
+        placeholder (str): Default text to display in input field. Defaults to `""`.
+        actions (Optional[List[Action]]): Defaults to `[]`.
+
+    """
+
+    type: Literal["user_text_area"] = "user_text_area"
+    actions: List[Action] = []  # noqa: RUF012
+
+    _set_actions = _action_validator_factory("value")
+
+    @_log_call
+    def build(self):
+        """Returns the text area component to display vizro-ai code output."""
+        return html.Div(
+            children=[
+                dcc.Textarea(
+                    id=self.id,
+                    placeholder="Describe the chart you want to create, e.g. "
+                    "'Visualize the life expectancy per continent.'",
+                )
+            ]
+        )
+
+
+class UserUpload(vm.VizroBaseModel):
+    """Component enabling data upload.
+
+    Args:
+        type (Literal["upload"]): Defaults to `"upload"`.
+        title (str): Title to be displayed.
+        actions (List[Action]): See [`Action`][vizro.models.Action]. Defaults to `[]`.
+
+    """
+
+    type: Literal["upload"] = "upload"
+    actions: List[Action] = []  # noqa: RUF012
+
+    # 'contents' property is input to custom action callback
+    _input_property: str = PrivateAttr("contents")
+    # change in 'contents' property of Upload component triggers the actions
+    _set_actions = _action_validator_factory("contents")
+
+    def build(self):
+        """Returns the upload component for data upload."""
+        return html.Div(
+            [
+                dcc.Upload(
+                    id=self.id,
+                    children=html.Div(
+                        ["Drag and Drop or ", html.A("Select Files")], style={"fontColor": "rgba(255, 255, 255, 0.6)"}
+                    ),
+                ),
+            ]
+        )
+
+
+class CodeClipboard(vm.VizroBaseModel):
+    """Code snippet with a copy to clipboard button."""
+
+    type: Literal["code_clipboard"] = "code_clipboard"
+    code: str = ""
+    language: str = "python"
+
+    def build(self):
+        """Returns the code clipboard component inside a output text area."""
+        code = black.format_str(self.code, mode=black.Mode(line_length=120))
+        code = code.strip("'\"")
+
+        markdown_code = "\n".join(["```python", code, "```"])
+
+        return html.Div(
+            [
+                dcc.Clipboard(target_id=f"{self.id}-code-markdown", className="code-clipboard"),
+                dcc.Markdown(markdown_code, id=f"{self.id}-code-markdown"),
+            ],
+            className="code-clipboard-container",
+        )
+
+
+class MyDropdown(vm.Dropdown):
+    """Custom dropdown component."""
+
+    type: Literal["my_dropdown"] = "my_dropdown"
+
+    def build(self):
+        """Returns custom dropdown component that cannot be cleared."""
+        dropdown_build_obj = super().build()
+        dropdown_build_obj.id = f"{self.id}_outer_div"
+        dropdown_build_obj.children[1].clearable = False
+
+        return dropdown_build_obj
+
+
+class Modal(vm.VizroBaseModel):
+    """Modal to convey warning message"""
+    
+    type: Literal["modal"] = "modal"
+    
+    def build(self):
+        """Returns the modal component."""
+        return dbc.Modal(
+            # id=self.id,
+            children=[
+                dbc.ModalHeader(children=dcc.Markdown("""# Warning""")),
+                dbc.ModalBody(children=dcc.Markdown("""### Do NOT upload any sensitive or personally identifying data.
+
+#### Reasoning:
+This space is hosted publicly running one server in a single container. Further this UI executes dynamically created code on the server. It thus
+cannot guarantee the security of your data. In addition it sends the user query and the data to the chosen LLM vendor API. This is not an exhaustive list.
+
+#### Alternatives:
+If sending your query and data to a LLM is acceptable, you can pull and run this image locally. This will avoid sharing
+an instance with others. You can do so by clicking the three dots in the top right of the HuggingFace banner und click `Run with Docker`.
+
+In any case, please remain cautious and understand the responsibilities of sharing data.
+""")),
+            ],
+            size="l",
+            is_open=True,
+        )
+
+class OffCanvas(vm.VizroBaseModel):
+    """OffCanvas component for settings."""
+
+    type: Literal["offcanvas"] = "offcanvas"
+    options: List[str]
+    value: str
+
+    def build(self):
+        """Returns the off canvas component for settings."""
+        input_groups = html.Div(
+            [
+                dbc.InputGroup(
+                    [
+                        dbc.InputGroupText("API Key"),
+                        dbc.Input(placeholder="API key", type="password", id=f"{self.id}-api-key"),
+                        html.Div(
+                            dbc.Checklist(
+                                id=f"{self.id}-api-key-toggle",
+                                options=[{"label": "", "value": False}],
+                                switch=True,
+                                inline=True,
+                            ),
+                            id="toggle-div-api-key",
+                        ),
+                    ],
+                    className="mb-3",
+                ),
+                dbc.InputGroup(
+                    [
+                        dbc.InputGroupText("API base"),
+                        dbc.Input(placeholder="(optional) API base", type="password", id=f"{self.id}-api-base"),
+                        html.Div(
+                            dbc.Checklist(
+                                id=f"{self.id}-api-base-toggle",
+                                options=[{"label": "", "value": False}],
+                                switch=True,
+                                inline=True,
+                            ),
+                            id="toggle-div-api-base",
+                        ),
+                    ],
+                    className="mb-3",
+                ),
+                dbc.InputGroup(
+                    [
+                        dbc.InputGroupText("Choose your vendor"),
+                        dbc.Select(options=self.options, value=self.value, id=f"{self.id}-dropdown"),
+                    ],
+                    className="mb-3",
+                ),
+            ],
+            className="mb-3",
+        )
+
+        offcanvas = dbc.Offcanvas(
+            id=self.id,
+            children=[
+                html.Div(
+                    children=[
+                        input_groups,
+                    ]
+                )
+            ],
+            title="Settings",
+            is_open=True,
+        )
+        return offcanvas
+
+
+class MyPage(vm.Page):
+    """Custom page."""
+
+    type: Literal["my_page"] = "my_page"
+
+    def pre_build(self):
+        """Overwriting pre_build."""
+        pass
+
+
+class Icon(vm.VizroBaseModel):
+    """Icon component for settings."""
+
+    type: Literal["icon"] = "icon"
+
+    def build(self):
+        """Returns the icon for api settings."""
+        return html.Div(
+            children=[html.Span("settings", className="material-symbols-outlined", id=self.id)],
+            className="settings-div",
+        )
+
+
+class CustomDashboard(vm.Dashboard):
+    """Custom Dashboard model."""
+
+    def build(self):
+        """Returns custom dashboard."""
+        dashboard_build_obj = super().build()
+        dashboard_build_obj.children.append(dcc.Store(id="data-store-id", storage_type="session"))
+        dashboard_build_obj.children.append(dcc.Store(id="outputs-store-id", storage_type="session"))
+        return dashboard_build_obj

requirements.in

@@ -0,0 +1,4 @@
+gunicorn
+vizro-ai>=0.2.1
+black
+jupyter

requirements.txt

@@ -0,0 +1,507 @@
+# This file was autogenerated by uv via the following command:
+#    uv pip compile requirements.in -o requirements.txt
+aiohappyeyeballs==2.4.0
+    # via aiohttp
+aiohttp==3.10.5
+    # via langchain
+aiosignal==1.3.1
+    # via aiohttp
+annotated-types==0.7.0
+    # via pydantic
+anyio==4.4.0
+    # via
+    #   httpx
+    #   jupyter-server
+    #   openai
+appnope==0.1.4
+    # via ipykernel
+argon2-cffi==23.1.0
+    # via jupyter-server
+argon2-cffi-bindings==21.2.0
+    # via argon2-cffi
+arrow==1.3.0
+    # via isoduration
+asttokens==2.4.1
+    # via stack-data
+async-lru==2.0.4
+    # via jupyterlab
+attrs==24.2.0
+    # via
+    #   aiohttp
+    #   jsonschema
+    #   referencing
+babel==2.16.0
+    # via jupyterlab-server
+beautifulsoup4==4.12.3
+    # via nbconvert
+black==24.8.0
+    # via -r requirements.in
+bleach==6.1.0
+    # via nbconvert
+blinker==1.8.2
+    # via flask
+cachelib==0.9.0
+    # via flask-caching
+certifi==2024.7.4
+    # via
+    #   httpcore
+    #   httpx
+    #   requests
+cffi==1.17.0
+    # via argon2-cffi-bindings
+charset-normalizer==3.3.2
+    # via requests
+click==8.1.7
+    # via
+    #   black
+    #   flask
+comm==0.2.2
+    # via
+    #   ipykernel
+    #   ipywidgets
+dash==2.17.1
+    # via
+    #   dash-ag-grid
+    #   dash-bootstrap-components
+    #   vizro
+dash-ag-grid==31.2.0
+    # via vizro
+dash-bootstrap-components==1.6.0
+    # via vizro
+dash-core-components==2.0.0
+    # via dash
+dash-html-components==2.0.0
+    # via dash
+dash-mantine-components==0.12.1
+    # via vizro
+dash-table==5.0.0
+    # via dash
+debugpy==1.8.5
+    # via ipykernel
+decorator==5.1.1
+    # via ipython
+defusedxml==0.7.1
+    # via nbconvert
+distro==1.9.0
+    # via openai
+executing==2.0.1
+    # via stack-data
+fastjsonschema==2.20.0
+    # via nbformat
+flask==3.0.3
+    # via
+    #   dash
+    #   flask-caching
+flask-caching==2.3.0
+    # via vizro
+fqdn==1.5.1
+    # via jsonschema
+frozenlist==1.4.1
+    # via
+    #   aiohttp
+    #   aiosignal
+gunicorn==23.0.0
+    # via -r requirements.in
+h11==0.14.0
+    # via httpcore
+httpcore==1.0.5
+    # via httpx
+httpx==0.27.2
+    # via
+    #   jupyterlab
+    #   langsmith
+    #   openai
+idna==3.8
+    # via
+    #   anyio
+    #   httpx
+    #   jsonschema
+    #   requests
+    #   yarl
+importlib-metadata==8.4.0
+    # via dash
+ipykernel==6.29.5
+    # via
+    #   jupyter
+    #   jupyter-console
+    #   jupyterlab
+    #   qtconsole
+ipython==8.26.0
+    # via
+    #   ipykernel
+    #   ipywidgets
+    #   jupyter-console
+ipywidgets==8.1.5
+    # via jupyter
+isoduration==20.11.0
+    # via jsonschema
+itsdangerous==2.2.0
+    # via flask
+jedi==0.19.1
+    # via ipython
+jinja2==3.1.4
+    # via
+    #   flask
+    #   jupyter-server
+    #   jupyterlab
+    #   jupyterlab-server
+    #   nbconvert
+jiter==0.5.0
+    # via openai
+json5==0.9.25
+    # via jupyterlab-server
+jsonpatch==1.33
+    # via langchain-core
+jsonpointer==3.0.0
+    # via
+    #   jsonpatch
+    #   jsonschema
+jsonschema==4.23.0
+    # via
+    #   jupyter-events
+    #   jupyterlab-server
+    #   nbformat
+jsonschema-specifications==2023.12.1
+    # via jsonschema
+jupyter==1.0.0
+    # via -r requirements.in
+jupyter-client==8.6.2
+    # via
+    #   ipykernel
+    #   jupyter-console
+    #   jupyter-server
+    #   nbclient
+    #   qtconsole
+jupyter-console==6.6.3
+    # via jupyter
+jupyter-core==5.7.2
+    # via
+    #   ipykernel
+    #   jupyter-client
+    #   jupyter-console
+    #   jupyter-server
+    #   jupyterlab
+    #   nbclient
+    #   nbconvert
+    #   nbformat
+    #   qtconsole
+jupyter-events==0.10.0
+    # via jupyter-server
+jupyter-lsp==2.2.5
+    # via jupyterlab
+jupyter-server==2.14.2
+    # via
+    #   jupyter-lsp
+    #   jupyterlab
+    #   jupyterlab-server
+    #   notebook
+    #   notebook-shim
+jupyter-server-terminals==0.5.3
+    # via jupyter-server
+jupyterlab==4.2.5
+    # via notebook
+jupyterlab-pygments==0.3.0
+    # via nbconvert
+jupyterlab-server==2.27.3
+    # via
+    #   jupyterlab
+    #   notebook
+jupyterlab-widgets==3.0.13
+    # via ipywidgets
+langchain==0.2.15
+    # via vizro-ai
+langchain-core==0.2.36
+    # via
+    #   langchain
+    #   langchain-openai
+    #   langchain-text-splitters
+    #   langgraph
+    #   langgraph-checkpoint
+langchain-openai==0.1.23
+    # via vizro-ai
+langchain-text-splitters==0.2.2
+    # via langchain
+langgraph==0.2.14
+    # via vizro-ai
+langgraph-checkpoint==1.0.6
+    # via langgraph
+langsmith==0.1.106
+    # via
+    #   langchain
+    #   langchain-core
+markupsafe==2.1.5
+    # via
+    #   jinja2
+    #   nbconvert
+    #   werkzeug
+matplotlib-inline==0.1.7
+    # via
+    #   ipykernel
+    #   ipython
+mistune==3.0.2
+    # via nbconvert
+multidict==6.0.5
+    # via
+    #   aiohttp
+    #   yarl
+mypy-extensions==1.0.0
+    # via black
+nbclient==0.10.0
+    # via nbconvert
+nbconvert==7.16.4
+    # via
+    #   jupyter
+    #   jupyter-server
+nbformat==5.10.4
+    # via
+    #   jupyter-server
+    #   nbclient
+    #   nbconvert
+nest-asyncio==1.6.0
+    # via
+    #   dash
+    #   ipykernel
+notebook==7.2.2
+    # via jupyter
+notebook-shim==0.2.4
+    # via
+    #   jupyterlab
+    #   notebook
+numpy==1.26.4
+    # via
+    #   langchain
+    #   pandas
+openai==1.42.0
+    # via
+    #   langchain-openai
+    #   vizro-ai
+orjson==3.10.7
+    # via langsmith
+overrides==7.7.0
+    # via jupyter-server
+packaging==24.1
+    # via
+    #   black
+    #   gunicorn
+    #   ipykernel
+    #   jupyter-server
+    #   jupyterlab
+    #   jupyterlab-server
+    #   langchain-core
+    #   nbconvert
+    #   plotly
+    #   qtconsole
+    #   qtpy
+pandas==2.2.2
+    # via
+    #   vizro
+    #   vizro-ai
+pandocfilters==1.5.1
+    # via nbconvert
+parso==0.8.4
+    # via jedi
+pathspec==0.12.1
+    # via black
+pexpect==4.9.0
+    # via ipython
+platformdirs==4.2.2
+    # via
+    #   black
+    #   jupyter-core
+plotly==5.23.0
+    # via dash
+prometheus-client==0.20.0
+    # via jupyter-server
+prompt-toolkit==3.0.47
+    # via
+    #   ipython
+    #   jupyter-console
+psutil==6.0.0
+    # via ipykernel
+ptyprocess==0.7.0
+    # via
+    #   pexpect
+    #   terminado
+pure-eval==0.2.3
+    # via stack-data
+pycparser==2.22
+    # via cffi
+pydantic==2.8.2
+    # via
+    #   langchain
+    #   langchain-core
+    #   langsmith
+    #   openai
+    #   vizro
+pydantic-core==2.20.1
+    # via pydantic
+pygments==2.18.0
+    # via
+    #   ipython
+    #   jupyter-console
+    #   nbconvert
+    #   qtconsole
+python-dateutil==2.9.0.post0
+    # via
+    #   arrow
+    #   jupyter-client
+    #   pandas
+python-dotenv==1.0.1
+    # via vizro-ai
+python-json-logger==2.0.7
+    # via jupyter-events
+pytz==2024.1
+    # via pandas
+pyyaml==6.0.2
+    # via
+    #   jupyter-events
+    #   langchain
+    #   langchain-core
+pyzmq==26.2.0
+    # via
+    #   ipykernel
+    #   jupyter-client
+    #   jupyter-console
+    #   jupyter-server
+qtconsole==5.6.0
+    # via jupyter
+qtpy==2.4.1
+    # via qtconsole
+referencing==0.35.1
+    # via
+    #   jsonschema
+    #   jsonschema-specifications
+    #   jupyter-events
+regex==2024.7.24
+    # via tiktoken
+requests==2.32.3
+    # via
+    #   dash
+    #   jupyterlab-server
+    #   langchain
+    #   langsmith
+    #   tiktoken
+retrying==1.3.4
+    # via dash
+rfc3339-validator==0.1.4
+    # via
+    #   jsonschema
+    #   jupyter-events
+rfc3986-validator==0.1.1
+    # via
+    #   jsonschema
+    #   jupyter-events
+rpds-py==0.20.0
+    # via
+    #   jsonschema
+    #   referencing
+ruff==0.6.2
+    # via vizro
+send2trash==1.8.3
+    # via jupyter-server
+setuptools==74.0.0
+    # via
+    #   dash
+    #   jupyterlab
+six==1.16.0
+    # via
+    #   asttokens
+    #   bleach
+    #   python-dateutil
+    #   retrying
+    #   rfc3339-validator
+sniffio==1.3.1
+    # via
+    #   anyio
+    #   httpx
+    #   openai
+soupsieve==2.6
+    # via beautifulsoup4
+sqlalchemy==2.0.32
+    # via langchain
+stack-data==0.6.3
+    # via ipython
+tabulate==0.9.0
+    # via vizro-ai
+tenacity==8.5.0
+    # via
+    #   langchain
+    #   langchain-core
+    #   plotly
+terminado==0.18.1
+    # via
+    #   jupyter-server
+    #   jupyter-server-terminals
+tiktoken==0.7.0
+    # via langchain-openai
+tinycss2==1.3.0
+    # via nbconvert
+tornado==6.4.1
+    # via
+    #   ipykernel
+    #   jupyter-client
+    #   jupyter-server
+    #   jupyterlab
+    #   notebook
+    #   terminado
+tqdm==4.66.5
+    # via openai
+traitlets==5.14.3
+    # via
+    #   comm
+    #   ipykernel
+    #   ipython
+    #   ipywidgets
+    #   jupyter-client
+    #   jupyter-console
+    #   jupyter-core
+    #   jupyter-events
+    #   jupyter-server
+    #   jupyterlab
+    #   matplotlib-inline
+    #   nbclient
+    #   nbconvert
+    #   nbformat
+    #   qtconsole
+types-python-dateutil==2.9.0.20240821
+    # via arrow
+typing-extensions==4.12.2
+    # via
+    #   dash
+    #   langchain-core
+    #   openai
+    #   pydantic
+    #   pydantic-core
+    #   sqlalchemy
+tzdata==2024.1
+    # via pandas
+uri-template==1.3.0
+    # via jsonschema
+urllib3==2.2.2
+    # via requests
+vizro==0.1.21
+    # via vizro-ai
+vizro-ai==0.2.1
+    # via -r requirements.in
+wcwidth==0.2.13
+    # via prompt-toolkit
+webcolors==24.8.0
+    # via jsonschema
+webencodings==0.5.1
+    # via
+    #   bleach
+    #   tinycss2
+websocket-client==1.8.0
+    # via jupyter-server
+werkzeug==3.0.4
+    # via
+    #   dash
+    #   flask
+widgetsnbextension==4.0.13
+    # via ipywidgets
+wrapt==1.16.0
+    # via vizro
+yarl==1.9.4
+    # via aiohttp
+zipp==3.20.1
+    # via importlib-metadata