Streamlit - rzut kostkami szerokościennymi

from random import randint
import streamlit as st


st.header("Gra w kostki szerokościenne")
wyniki = st.session_state.setdefault("wyniki", [])
option = st.selectbox(
    "Jakimi kostkami chcesz rzucać?",
    ("K4", "K6", "K8", "K10", "K12"),
    index=0,
    # placeholder="Wybierz rodzaj kostki...",
)

# scianki = {"K4":4, "K6":6, "K8":10, "K10":10, "K12":12}

# ponizej mapowanie z wybranej opcji np."K4" na "4" i zamiana na liczbę 4"

opt = int(option[1:])

st.write("Wybrana kostka:", opt)


ilosc_rzutow = st.slider("Iloma kostkami z przedziału od 1 do 20 chcesz rzucić?", min_value=1, max_value=20, value=20)
st.write(f"Wybrałeś, że chcesz rzucić {ilosc_rzutow} kostkami.")

rzuc_btn = st.button("Rzuć")

if rzuc_btn:
    st.session_state['wyniki'] = []
    for rzut in range(ilosc_rzutow):
        wynik_rzutu = randint(1, opt)
        st.session_state['wyniki'].append(wynik_rzutu)

st.write(f"Uzyskano: {st.session_state['wyniki']}")
st.write(f"Suma: {sum(st.session_state['wyniki'])}")