Code Editor for app.py

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

@st.cache_data
def load_data(url):
    return pd.read_csv(url)

st.title('Penguins Data Visualization')

url = st.text_input('Enter CSV URL:', 'https://raw.githubusercontent.com/mwaskom/seaborn-data/master/penguins.csv')

# Load data
try:
    df = load_data(url)
    st.write('Data successfully loaded!')
except Exception as e:
    st.error(f'Error loading data: {e}')
    df = pd.DataFrame()

if not df.empty:
    
    # Control number of rows to display
    n_rows = st.slider('Number of rows to display:', min_value=1, max_value=len(df), value=10)
    st.write(df.head(n_rows))

    # Checkbox to shuffle data
    if st.checkbox('Shuffle data'):
        df = df.sample(frac=1).reset_index(drop=True)
        st.write('Data has been shuffled.')
        st.write(df.head(n_rows))

    # Select variables for visualization
    variables = df.select_dtypes(include=[np.number]).columns.tolist()
    selected_vars = st.multiselect('Select variables for visualization:', variables)    

    if len(selected_vars) == 2:
        chart = alt.Chart(df).mark_circle().encode(
            x=selected_vars[0],
            y=selected_vars[1],
            tooltip=selected_vars
        ).interactive()
        st.altair_chart(chart, use_container_width=True)
    elif len(selected_vars) == 1:
        chart = alt.Chart(df).mark_bar().encode(
            x=selected_vars[0],
            y='count()'
        ).interactive()
        st.altair_chart(chart, use_container_width=True)
    elif len(selected_vars) == 0:
        st.write('Please select at least one variable to visualize.')