import streamlit as st
import pandas as pd
import altair as alt

# 加载数据集
@st.cache
def load_data():
    url = "https://raw.githubusercontent.com/mwaskom/seaborn-data/master/diamonds.csv"
    data = pd.read_csv(url)
    return data

# 创建可视化图表
def create_chart(data):
    chart = alt.Chart(data).mark_point().encode(
        x='carat',
        y='price',
        color='cut'
    ).interactive()
    return chart

# 主函数
def main():
    st.title('Diamonds Dataset')
    st.write("This is a demo of Streamlit app")

    # 加载数据
    data = load_data()

    # 显示数据集
    if st.checkbox('Show raw data'):
        st.write(data.head())

    # 显示数据集描述信息
    st.write("Dataset Description:")
    st.write(data.describe())

    # 可视化图表
    st.write("Chart:")
    chart = create_chart(data)
    st.write(chart)

if __name__ == '__main__':
    main()