import streamlit as st
import os

def main():
    # Layout inicial
    col1, col2, col3 = st.columns([4, 5, 1])
    if os.path.exists("images.jpg"):
        col1.image("images.jpg", width=250)
    else:
        col1.write("Imagem não encontrada.")
    col2.title("Energia Elétrica", anchor="right")
    st.write("Conheça esse calculador de tarifas de energia elétrica!")

    # Lista de eletrodomésticos
    if "eletrodomesticos" not in st.session_state:
        st.session_state.eletrodomesticos = []

    n = st.number_input("Quantos eletrodomésticos você deseja analisar?", min_value=1, step=1)

    for i in range(n):
        st.write(f"\nEletrodoméstico {i+1}:")
        nome = st.text_input(f"Digite o nome do eletrodoméstico {i+1}:", key=f"nome_{i}")
        potencia = st.number_input(f"Digite a potência (em watts) do eletrodoméstico {i+1}:", 
                                   min_value=0.0, key=f"potencia_{i}")
        tempo = st.number_input(f"Digite o tempo de uso diário (em horas) do eletrodoméstico {i+1}:", 
                                 min_value=0.0, key=f"tempo_{i}")
        dias = st.number_input(f"Digite o número de dias de uso por mês do eletrodoméstico {i+1}:", 
                                min_value=1, step=1, key=f"dias_{i}")
        
        if st.button(f"Adicionar Eletrodoméstico {i+1}", key=f"botao_{i}"):
            st.session_state.eletrodomesticos.append({
                "nome": nome,
                "potencia": potencia,
                "tempo": tempo,
                "dias": dias
            })

    # Calculando consumo
    consumos = []
    for e in st.session_state.eletrodomesticos:
        consumo_diario = (e["potencia"] * e["tempo"]) / 1000  # kWh por dia
        consumo_total = consumo_diario * e["dias"]           # kWh total
        consumos.append({
            "nome": e["nome"],
            "consumo_diario": consumo_diario,
            "consumo_total": consumo_total
        })

    # Exibindo resultados
    if consumos:
        st.write("\n=== Resultados ===")
        for c in consumos:
            st.write(f"- {c['nome']}: {c['consumo_total']:.2f} kWh")

        consumo_geral = sum(c["consumo_total"] for c in consumos)
        st.write(f"O consumo geral é de {consumo_geral:.2f} kWh")

        # Calculando custo
        calcular_custo = st.radio("\nDeseja calcular o custo de energia?", ('Sim', 'Não'))
        if calcular_custo == 'Sim':
            tarifa = st.number_input("Digite a tarifa de energia (R$ por kWh):", min_value=0.0)
            if tarifa:
                custo_total = consumo_geral * tarifa
                st.write(f"Custo total estimado: R$ {custo_total:.2f}")

if __name__ == "__main__":
    main()