Code Editor for app.py

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

# Function to load a CSV file from a URL
@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")

try:
    df = load_data(url)
    st.write("### DataFrame", df)
    
    # Control for number of rows displayed
    num_rows = st.slider("Select number of rows to display:", 1, df.shape[0], 10)
    st.write(df.head(num_rows))
    
    # Checkbox for shuffling DataFrame
    if st.checkbox("Shuffle the DataFrame"):
        df = df.sample(frac=1).reset_index(drop=True)
        st.write("### Shuffled DataFrame", df.head(num_rows))
    
    # Selection of variables for plotting
    variables = df.select_dtypes(include=['float64', 'int64']).columns.tolist()
    x_var = st.selectbox("Select X-axis variable:", options=variables)
    y_var = st.selectbox("Select Y-axis variable:", options=variables) if st.checkbox("Select Y-axis variable") else None
    
    # Plotting
    if y_var:
        chart = alt.Chart(df).mark_circle(size=60).encode(
            x=x_var,
            y=y_var,
            tooltip=[x_var, y_var]
        ).interactive()
        st.altair_chart(chart, use_container_width=True)
    else:
        chart = alt.Chart(df).mark_bar().encode(
            x=x_var,
            y='count()'
        ).interactive()
        st.altair_chart(chart, use_container_width=True)

except Exception as e:
    st.error(f"Error loading data: {e}")