# -*- coding: utf-8 -*-
"""
Created on Fri Jun 16 07:44:31 2023

@author: mritchey
"""
# Query Pandas App
from yolopandas import pd
# import pandas as pd
import streamlit as st

product_df = pd.DataFrame(
    [
        {"name": "The Da Vinci Code", "type": "book", "price": 15, "quantity": 300, "rating": 4},
        {"name": "Jurassic Park", "type": "book", "price": 12, "quantity": 400, "rating": 4.5},
        {"name": "Jurassic Park", "type": "film", "price": 8, "quantity": 6, "rating": 5},
        {"name": "Matilda", "type": "book", "price": 5, "quantity": 80, "rating": 4},
        {"name": "Clockwork Orange", "type": None, "price": None, "quantity": 20, "rating": 4},
        {"name": "Walden", "type": None, "price": None, "quantity": 100, "rating": 4.5},
    ],
)

product_df
product_df.llm.query("Impute the type column with random fruits.", yolo=True)