import streamlit as st
import plotly.express as px
import pandas as pd
import numpy as np
import base64

def generate_values(n):
    return {"HealthPoints": np.random.randint(50, 100, size=n), 
            "Coins": np.random.randint(10, 50, size=n)}

def app():
    st.title("Game Mechanics Treemap Chart")
    st.write("This app displays a Treemap chart of game mechanics.")

    n = len(game_mechanics := ["Action Queue ⏱️", "Action Retrieval 🔍", 
                              "Campaign / Battle Card Driven 🗃️", "Card Play Conflict Resolution 💳🤝", 
                              "Communication Limits 🙊", "Cooperative Game 🤝👥", "Critical Hits and Failures 💥💔", 
                              "Deck Construction 🎴🛠️", "Grid Movement 🗺️", "Hand Management 🖐️📊", 
                              "Hexagon Grid 🔳", "Legacy Game 🎓🎮", "Line of Sight 👀", "Modular Board 🧩", 
                              "Once-Per-Game Abilities 🌟", "Role Playing 🎭", "Scenario / Mission / Campaign Game 🎯", 
                              "Simultaneous Action Selection 🤜🤛", "Solo / Solitaire Game 🕺", "Storytelling 📖", 
                              "Variable Player Powers 🦸‍♂️🦹‍♀️"])
    
    values = generate_values(n)
    data = pd.DataFrame({"category": ["Category 1"] * n, "mechanic": game_mechanics, 
                         "value": list(values.values()), "HealthPoints": values["HealthPoints"], 
                         "Coins": values["Coins"], "description": ["Description"] * n})
    data["value"] = data["value"].apply(lambda x: sum(x))

    fig = px.treemap(data, path=["category", "mechanic"], values="value", 
                     color="HealthPoints", hover_data=["Coins"])
    st.plotly_chart(fig)

    csv = data.to_csv(index=False)
    b64 = base64.b64encode(csv.encode()).decode()
    href = f'<a href="data:file/csv;base64,{b64}" download="game_mechanics.csv">Download CSV</a>'
    st.markdown(href, unsafe_allow_html=True)

if __name__ == '__main__':
    app()