Add application file

tmp/app.py

@@ -0,0 +1,80 @@
+import streamlit as st
+import pandas as pd
+import numpy as np
+from PIL import Image
+import pickle 
+# formata o valor na moeda brasileira
+import locale 
+
+# Esconder os menu padrao
+hide_streamlit_style = """
+            <style>
+            #MainMenu {visibility: hidden;}
+            footer {visibility: hidden;}
+            </style>
+            """
+st.markdown(hide_streamlit_style, unsafe_allow_html=True) 
+
+st.title('Sistema de Seguro de Saúde')
+st.sidebar.title("Informe os dados")
+
+# variaveis 
+atributos = ['idade', 'sexo', 'IMC', 'num_filhos', 'fumante', 'regiao']
+# prever o 'valor_seguro']
+
+dict_categorias = {'sexo': {'Feminino': 0, 'Masculino': 1},
+                   'fumante': {'nao': 0, 'sim': 1},
+                   'regiao': {'sul': 0, 'sudeste': 1, 'Norte': 2, 'Nordeste': 3}
+                   }
+
+with st.sidebar:
+
+    with st.form(key='my_form'):
+
+        # [ 'idade', 'sexo', 'IMC', 'num_filhos', 'fumante', 'regiao']
+
+        idade = st.number_input('idade', min_value=0, max_value=150, step=1, value=30)
+
+        cat_sexo  = st.selectbox('sexo', options=list(dict_categorias['sexo'].keys()))
+        sexo = dict_categorias['sexo'][cat_sexo]
+
+        IMC = st.number_input('IMC', min_value=10.0, max_value=350.0, value=25.0, step=1.0)
+
+        num_filhos = st.number_input('num_filhos', min_value=0, max_value=300, value=2)
+
+        cat_fumante = st.selectbox('fumante',options=list(dict_categorias['fumante'].keys()))
+        fumante = dict_categorias['fumante'][cat_fumante]
+ 
+        cat_regiao = st.selectbox('região',options=list(dict_categorias['regiao'].keys()))
+        regiao = dict_categorias['regiao'][cat_regiao]
+
+        predict_button = st.form_submit_button(label='Prever')
+
+
+# Pagina pricipal
+arquivo_modelo = 'modelo.pkl'
+with open(arquivo_modelo, 'rb') as f:
+    modelo = pickle.load(f)
+
+def previsao_seguro(modelo, idade, sexo, IMC, num_filhos, fumante, regiao):
+
+    new_X = np.array([idade, sexo, IMC, num_filhos, fumante, regiao])
+    valor_seguro = modelo.predict(new_X.reshape(1, -1) )[0]
+    return valor_seguro
+
+imagem = 'seguro.jpeg'
+
+image = Image.open(imagem)
+st.image(image, width=500)
+
+if predict_button:
+    valor_seguro  = previsao_seguro(modelo, idade, sexo, IMC, num_filhos, fumante, regiao)
+
+    locale.setlocale(locale.LC_ALL, 'pt_BR.UTF-8')
+    str_valor_seguro  = locale.currency(valor_seguro, grouping=True, symbol=None)
+    str_valor_seguro_mensal = locale.currency(valor_seguro / 12, grouping=True, symbol=None)
+    st.markdown('## Valor do Seguro (Anual): R$ ' +  str_valor_seguro )
+    st.markdown('## Valor Mensal: R$ ' +  str_valor_seguro_mensal )
+
+
+