Update app.py

app.py

@@ -23,18 +23,28 @@ modelo = ModeloRiscoCol(df['N'], df['PN'], df['X'], df['PX'])
 
 option = st.selectbox(
      'O que você gostaria de calcular?',
-     ('Prêmio puro', 'Prêmio comercial', 'Prêmio puro pela aproximação normal', 'Probabiliade de insolvência'))
+     ('Prêmio puro', 'Prêmio comercial', 'Prêmio pela aproximação normal', 'Carregamento de segurança'))
 
 st.write('Você selecionou:', option)
 
 if option == 'Prêmio puro':
-    number = st.number_input('Qual o valor do carregamento de segurança:', min_value=0.0, value=0.1)
-    st.write('carregamento de segurança de ', number)
-    premio = modelo.premio_puro(number)
+    carreg_seg = st.number_input('Qual o valor do carregamento de segurança:', min_value=0.0, value=0.1)
+    st.write('carregamento de segurança de ', carreg_seg)
+    premio = modelo.premio_puro(carreg_seg)
     st.write('Prêmio puro =', premio) 
 elif option == 'Prêmio comercial':
-    number = st.number_input('Qual o valor do carregamento de segurança:', min_value=0.0, value=0.1)
-    number2 = st.number_input('Qual o valor do carregamento administrativo:', min_value=0.0, value=0.1)                 
-    premio = modelo.premio_comercial(number, number2)
+    carreg_seg = st.number_input('Qual o valor do carregamento de segurança:', min_value=0.0, value=0.1)
+    carreg_adm = st.number_input('Qual o valor do carregamento administrativo:', min_value=0.0, value=0.1)                 
+    premio = modelo.premio_comercial(carreg_seg, carreg_adm)
     st.write('Prêmio comercial =', premio)               
-
+elif option == 'Prêmio pela aproximação normal':
+    alfa = st.number_input('Qual a probabilidade de o sinistro agregado superar o prêmio puro total:', min_value=0.0,     
+                             max_value=1.0,value=0.05)
+    premio = modelo.premio_aprox_normal(alfa) 
+    st.write('Prêmio =', premio) 
+else:
+    alfa = st.number_input('Qual a probabilidade de o sinistro agregado superar o prêmio puro total:', min_value=0.0,     
+                             max_value=1.0,value=0.05)
+    carreg_seg = modelo.carreg_seg(alfa)
+    st.write('Carregamento de segurança =', carreg_seg)
+