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 Dataset Viewer')

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

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

# Display DataFrame
if not df.empty:
    rows_to_display = st.number_input('Select number of rows to display:', min_value=1, max_value=len(df), value=5)
    shuffle_data = st.checkbox('Shuffle DataFrame', value=False)

    if shuffle_data:
        df = df.sample(frac=1).reset_index(drop=True)
    st.write(df.head(rows_to_display))

    # Variable selection
    variable1 = st.selectbox('Select first variable:', df.columns)
    variable2 = st.selectbox('Select second variable (optional):', ["None"] + list(df.columns))

    # Visualization
    if variable2 == "None":
        chart = alt.Chart(df).mark_bar().encode(
            x=variable1,
            y='count()'
        )
    else:
        chart = alt.Chart(df).mark_circle().encode(
            x=variable1,
            y=variable2,
            tooltip=list(df.columns)
        )
    st.altair_chart(chart)