Code Editor for app.py

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

# Function to load data
@st.cache_data
def load_data(url):
    return pd.read_csv(url)

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

# Load the DataFrame
try:
    df = load_data(url)
    st.write('Data loaded successfully!')
except Exception as e:
    st.write('Error loading data:', e)

# Controls for display
num_rows = st.slider('Select number of rows to display:', 1, len(df), 10)
shuffle = st.checkbox('Shuffle DataFrame')

if shuffle:
    df = df.sample(frac=1).reset_index(drop=True)

# Display DataFrame
st.write(df.head(num_rows))

# Plotting controls
st.subheader('Plotting Variables')
variables = df.columns.tolist()
selected_vars = st.multiselect('Select variables to plot:', variables, default=[variables[0]])

if len(selected_vars) > 0:
    if len(selected_vars) == 1:
        # Create a histogram
        chart = alt.Chart(df).mark_bar().encode(
            alt.X(selected_vars[0], bin=True),
            y='count()'
        ).properties(title=f'Histogram of {selected_vars[0]}')
    elif len(selected_vars) == 2:
        # Create a scatter plot
        chart = alt.Chart(df).mark_point().encode(
            x=selected_vars[0],
            y=selected_vars[1],
            tooltip=selected_vars
        ).properties(title=f'Scatter Plot of {selected_vars[0]} vs {selected_vars[1]}')
    else:
        st.write('Please select one or two variables for plotting.')
    st.altair_chart(chart, use_container_width=True)