from tinydb import TinyDB, where
import streamlit as st

db = TinyDB("loja.json")
estoque = db.table("estoque")
vendas  = db.table("vendas")

st.set_page_config(page_title="Sistema Loja", page_icon="🏪")
st.title("🏪 Sistema de Loja")

aba_est, aba_cad, aba_ven, aba_hist = st.tabs(
    ["📦 Estoque", "➕ Cadastrar", "🛒 Vender", "📊 Vendas"]
)

# ════ ESTOQUE ════
with aba_est:
    st.header("Estoque atual")
    produtos = estoque.all()
    if len(produtos) == 0:
        st.info("Estoque vazio. Cadastre um produto na aba ao lado!")
    else:
        st.table(produtos)
        st.metric("Total de produtos", len(produtos))

# ════ CADASTRAR ════
with aba_cad:
    st.header("Cadastrar produto")
    nome  = st.text_input("Nome", key="cad_nome")
    preco = st.number_input("Preço (R$)", min_value=0.0, step=0.5, key="cad_preco")
    qtd   = st.number_input("Quantidade", min_value=0, step=1, key="cad_qtd")

    if st.button("Cadastrar", type="primary"):
        if nome.strip() == "":
            st.error("Digite o nome!")
        else:
            estoque.insert({
                "nome": nome, "preco": float(preco), "qtd": int(qtd)
            })
            st.success(f"✅ '{nome}' cadastrado!")
            st.rerun()

# ════ VENDER ════
with aba_ven:
    st.header("Registrar venda")
    produtos = estoque.all()
    if len(produtos) == 0:
        st.info("Sem produtos pra vender.")
    else:
        nomes = [p["nome"] for p in produtos if p["qtd"] > 0]
        if len(nomes) == 0:
            st.warning("Tudo zerado!")
        else:
            escolhido = st.selectbox("Produto", nomes, key="v_prod")
            qtd_v     = st.number_input("Qtd", min_value=1, step=1, key="v_qtd")
            cliente   = st.text_input("Cliente", key="v_cli")

            produto = estoque.get(where("nome") == escolhido)
            total   = produto["preco"] * qtd_v
            st.info(f"💰 Total: R$ {total:.2f}")

            if st.button("Confirmar venda", type="primary"):
                vendas.insert({
                    "produto": escolhido, "qtd": int(qtd_v),
                    "total": total, "cliente": cliente
                })
                estoque.update(
                    {"qtd": produto["qtd"] - int(qtd_v)},
                    where("nome") == escolhido
                )
                st.success("✅ Venda registrada!")
                st.balloons()
                st.rerun()

# ════ HISTÓRICO ════
with aba_hist:
    st.header("Histórico de vendas")
    lista = vendas.all()
    if len(lista) == 0:
        st.info("Sem vendas ainda.")
    else:
        st.table(lista)
        soma = sum(v["total"] for v in lista)
        col1, col2 = st.columns(2)
        col1.metric("Total de vendas", len(lista))
        col2.metric("Faturamento", f"R$ {soma:.2f}")