Pandas

This example and walkthrough has the following features:

• querychat interaction with a pandas dataframe
• Reads in a data description file
• Reads in a greeting file

Data

This examples uses the seaborn library to load the titanic dataset.

Greeting file

Save this file as greeting.md:

Hello! Welcome to your Titanic data dashboard. I'm here to help you filter, sort, and analyze the data. Here are a few ideas to get you started:

* Explore the data
  * <span class="suggestion">Show me all passengers who survived</span>
  * <span class="suggestion">Show only first class passengers</span>
* Analyze statistics
  * <span class="suggestion">What is the average age of passengers?</span>
  * <span class="suggestion">How many children were on board?</span>
* Compare and dig deeper
  * <span class="suggestion">Which class had the highest survival rate?</span>
  * <span class="suggestion">Show the fare distribution by embarkation town</span>

Let me know what you'd like to explore!

Data description file

Save this file as data_description.md:

# Data Dictionary

- **survival**: Survival status  
  - 0 = No  
  - 1 = Yes  

- **pclass**: Ticket class  
  - 1 = 1st class  
  - 2 = 2nd class  
  - 3 = 3rd class  

- **sex**: Sex of the passenger  

- **age**: Age in years  

- **sibsp**: Number of siblings or spouses aboard the Titanic  

- **parch**: Number of parents or children aboard the Titanic  

- **ticket**: Ticket number  

- **fare**: Passenger fare  

- **cabin**: Cabin number  

- **embarked**: Port of embarkation  
  - C = Cherbourg  
  - Q = Queenstown  
  - S = Southampton  

## Variable Notes

- **pclass** is a proxy for socio-economic status (SES):  
  - 1st = Upper class  
  - 2nd = Middle class  
  - 3rd = Lower class  

- **age**:  
  - If less than 1 year old, age is fractional.  
  - Estimated ages are represented as `xx.5`.  

- **sibsp**: Family relations are defined as:  
  - Sibling = brother, sister, stepbrother, stepsister  
  - Spouse = husband, wife (mistresses and fiancés were ignored)  

- **parch**: Family relations are defined as:  
  - Parent = mother, father  
  - Child = daughter, son, stepdaughter, stepson  
  - Some children traveled only with a nanny, so `parch = 0` for them.  

The application

Our application will read the the greeting.md and data_description.md files and pass them along to the querychat.init() function.

Here is our pandas example app, save the contents to app.py.

GitHub Models and GitHub Personal Access Tokens

This example does not use the default OpenAI model directly from OpenAI, which would require you to create an OpenAI API key and save it as an environment variable named OPENAI_API_KEY. Instead we are using GitHub Models as a free way to access the latest LLMs, with a rate-limit. You can follow the instructions on the GitHub Docs or Axure AI Demo on creating a PAT.

We suggest you save your PAT into 2 environment variables: GITHUB_TOKEN, and GITHUB_PAT.

from pathlib import Path

import chatlas
import querychat
from seaborn import load_dataset
from shiny import App, render, ui

titanic = load_dataset("titanic")

greeting = Path(__file__).parent / "greeting.md"
data_desc = Path(__file__).parent / "data_description.md"

# 1. Configure querychat

def use_github_models(system_prompt: str) -> chatlas.Chat:
    # GitHub models give us free rate-limited access to the latest LLMs
    # you will need to have GITHUB_PAT defined in your environment
    return chatlas.ChatGithub(
        model="gpt-4.1",
        system_prompt=system_prompt,
    )

qc_config = querychat.init(
    titanic,
    "titanic",
    greeting=greeting,
    data_description=data_desc,
    client=use_github_models,
)

# Create UI
app_ui = ui.page_sidebar(
    # 2. Place the chat component in the sidebar
    querychat.sidebar("chat"),
    # Main panel with data viewer
    ui.card(
        ui.output_data_frame("data_table"),
        fill=True,
    ),
    title="querychat with Python",
    fillable=True,
    class_="bslib-page-dashboard",
)


# Define server logic
def server(input, output, session):
    # 3. Initialize querychat server with the config from step 1
    qc = querychat.server("chat", qc_config)

    # 4. Display the filtered dataframe
    @render.data_frame
    def data_table():
        # Access filtered data via chat.df() reactive
        return qc.df()


# Create Shiny app
app = App(app_ui, server)