import streamlit as st
import pandas as pd
import matplotlib.pyplot as plt

# Function to visualize hydrogen station capacity per state
def plot_hydrogen_capacity(data):
    st.subheader("📊 Hydrogen Production Capacity by State")
    
    df = data.groupby("state")["capacity_kg_day"].sum().reset_index()

    plt.figure(figsize=(10, 5))
    plt.bar(df["state"], df["capacity_kg_day"], color="blue")
    plt.xlabel("State")
    plt.ylabel("Capacity (kg/day)")
    plt.title("Total Hydrogen Production Capacity per State")
    st.pyplot(plt)