tjl8's picture
Update app.py
d3bf697 verified
import streamlit as st
import pandas as pd
import plotly.graph_objects as go
import matplotlib.pyplot as plt
import seaborn as sns
#trial
# Set Streamlit page config
st.set_page_config(page_title="Legislative Visualizations", layout="wide")
st.title("Legislative Bill Analysis Dashboard")
# Upload dataset
uploaded_file = st.file_uploader("Illinois_Entire_Data_Insights_Final_v2.csv", type=["csv", "xlsx"])
if uploaded_file:
# Load dataset
if uploaded_file.name.endswith('.csv'):
df = pd.read_csv(uploaded_file)
else:
df = pd.read_excel(uploaded_file)
st.success("File uploaded and read successfully!")
# Sankey Diagram
st.header("πŸ”— Sankey Diagram: Intent β†’ Stance β†’ Beneficiaries")
sankey_df = df[['intent_standardized', 'stance_standardized', 'intended_beneficiaries_standardized']].dropna()
if not sankey_df.empty:
labels = list(pd.unique(sankey_df['intent_standardized'].tolist() +
sankey_df['stance_standardized'].tolist() +
sankey_df['intended_beneficiaries_standardized'].tolist()))
label_map = {label: i for i, label in enumerate(labels)}
intent_stance = sankey_df.groupby(['intent_standardized', 'stance_standardized']).size().reset_index(name='count')
stance_beneficiary = sankey_df.groupby(['stance_standardized', 'intended_beneficiaries_standardized']).size().reset_index(name='count')
source = intent_stance['intent_standardized'].map(label_map).tolist() + stance_beneficiary['stance_standardized'].map(label_map).tolist()
target = intent_stance['stance_standardized'].map(label_map).tolist() + stance_beneficiary['intended_beneficiaries_standardized'].map(label_map).tolist()
value = intent_stance['count'].tolist() + stance_beneficiary['count'].tolist()
fig_sankey = go.Figure(data=[go.Sankey(
node=dict(pad=15, thickness=20, line=dict(color="black", width=0.5), label=labels),
link=dict(source=source, target=target, value=value)
)])
fig_sankey.update_layout(title_text="Sankey: Intent β†’ Stance β†’ Beneficiary", font_size=12)
st.plotly_chart(fig_sankey, use_container_width=True)
else:
st.warning("Sankey input columns contain only null values or are missing.")
# Heatmap
st.header("🧯 Heatmap: Category vs Policy Impact Area")
heat_df = df[['category_&_subcategory_standardized', 'policy_impact_areas_standardized']].dropna()
if not heat_df.empty:
heat = heat_df.pivot_table(index='category_&_subcategory_standardized',
columns='policy_impact_areas_standardized',
aggfunc=len,
fill_value=0)
plt.figure(figsize=(14, 8))
sns.heatmap(heat, cmap='coolwarm', annot=False)
plt.title("Heatmap: Category vs Policy Impact Area")
plt.xlabel("Policy Impact Area")
plt.ylabel("Category")
plt.tight_layout()
st.pyplot(plt)
else:
st.warning("Heatmap input columns contain only null values or are missing.")
else:
st.info("Please upload a dataset file to view the visualizations.")