Update app.py

app.py

@@ -1,28 +1,43 @@
 import streamlit as st
+import os
 
-eletrodomesticos = []
+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 = []
 
-# LISTA DOS ELETRODOMESTICOS
-try:
     n = st.number_input("Quantos eletrodomésticos você deseja analisar?", min_value=1, step=1)
-    for i in range(n):
 
+    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}:")
-        potencia = st.number_input(f"Digite a potência (em watts) do eletrodoméstico {i+1}:", min_value=0.0)
-        tempo = st.number_input(f"Digite o tempo de uso diário (em horas) do eletrodoméstico {i+1}:", min_value=0.0)
-        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)
+        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}")
         
-        eletrodomesticos.append({
-            "nome": nome,
-            "potencia": potencia,
-            "tempo": tempo,
-            "dias": dias
-        })
+        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
+    # Calculando consumo
     consumos = []
-    for e in eletrodomesticos:
+    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({
@@ -31,43 +46,22 @@ try:
             "consumo_total": consumo_total
         })
 
-    # RESULTADO DO CONSUMO
-    st.write("\n=== Resultados ===")
-    st.write("Consumo por eletrodoméstico:")
-
-    for c in consumos:
-        st.write(f"- {c['nome']}: {c['consumo_total']:.2f} kWh")
+    # 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 A TARIFA
-    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)
-    else:
-        tarifa = None
-
-    # RESULTADO DA TARIFA
-    if tarifa:
-        st.write("\n=== VALOR APROXIMADO ===")
-
-        custo_total = consumo_geral * tarifa
-
-        st.write(f"Custo total estimado: R$ {custo_total:.2f}")
-
-except ValueError:
-    st.write("Por favor, verifique se as informações estão corretas.")
-def main():      
-        
-    col1, col2, col3 = st.columns([4, 5, 1])
-    col1.image("images.jpg", width=250)
-    col2.title("energia elétrica",  anchor="right")    
-        
-    st.write("Conheça esse calculador de tarifs de energia elétrica!.")    
+        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()
-
-