Update app.py

app.py

@@ -1,5 +1,6 @@
-"""_summary_
+"""
 LIMPEZA E VISUALIZAÇÃO DE CSV
+
 Returns:
     DATAFRAMES
 """
@@ -15,7 +16,7 @@ from streamlit_extras.stylable_container import stylable_container
 from streamlit_folium import folium_static
 from streamlit_option_menu import option_menu
 
-from data_cleaning import criar_dataframe, iniciar, limpa_rci1, limpa_rci2, separa_grupos, renomear_escola, limpar_dfs, processar_arquivo
+from data_cleaning import criar_dataframe, iniciar, limpa_rci1, limpa_rci2, separa_grupos, renomear_escola, limpar_dfs
 
 st.set_page_config(
     page_title="Dashboard UBS Flamengo",
@@ -24,7 +25,6 @@ st.set_page_config(
     initial_sidebar_state="expanded",
 )
 
-
 def processar_arquivo(file):
     """
     A function that processes the uploaded file by writing it to a temporary CSV file,
@@ -32,15 +32,20 @@ def processar_arquivo(file):
     Returns:
     - dataframes: the processed dataframes generated from the uploaded file
     """
-    with open("temp.csv", "wb") as f:
-        f.write(file.getbuffer())
-    csv_st = iniciar("temp.csv")
-    clean_txt = limpa_rci(csv_st)
+    csv_st = iniciar(file)
+    clean_txt = limpa_rci1(csv_st)
     grupos_encontrados = separa_grupos(clean_txt)
-    dfs = criar_dataframe(grupos_encontrados)
+    dataframes = criar_dataframe(grupos_encontrados)
+    
+    # Renomear descrições do DataFrame de escolaridade
+    dataframes["Escola"] = renomear_escola(dataframes["Escola"])
 
-    return dfs
+    # Aplicar limpeza final nos DataFrames
+    clean_final = r"(\sativ\w+)|(\sde\scid.+o)|(iciê.+ia)|((?<=rans)gên.+o|omem|ulher|ário|Homoss.+\(|\))|(\d{10}\s-\s)"
+    substituicao = ""
+    dataframes = limpar_dfs(dataframes, clean_final, substituicao)
 
+    return dataframes
 
 # Inicializar variáveis dos DataFrames como None
 DF_DATA = None
@@ -56,11 +61,6 @@ DF_TRANSGEN = None
 # Upload de arquivo CSV na barra lateral
 uploaded_file = st.sidebar.file_uploader("Escolha um arquivo CSV", type="csv")
 
-def processar_arquivo(uploaded_file):
-    # Função que processa o arquivo CSV e retorna os DataFrames
-    # (Esta função deve ser importada ou definida aqui)
-    pass
-
 if uploaded_file is not None:
     dataframes = processar_arquivo(uploaded_file)
 
@@ -130,7 +130,6 @@ def gerar_resumo_df():
 # Mostrar o resumo no Streamlit
 st.text(gerar_resumo_df())
 
-
 #######################################
 #######################################
 ############ BANNER  ##################
@@ -154,7 +153,6 @@ with stylable_container(
 ):
     st.image("./banner.png")
 
-
 #######################################
 ############ BANNER  ##################
 #######################################
@@ -166,7 +164,6 @@ with c1:
     st.markdown(
         """
                 ##  Dashboard UBS Flamengo
-
                 ###### :brain: _by Dr. Guilherme Apolinário_
                 
                 #### Bem-vindo! :wave:
@@ -191,7 +188,6 @@ with c2:
         st.image("./graph1.png")
 st.divider()
 
-
 ###########################################
 ###########################################
 ###########################################
@@ -206,7 +202,6 @@ st.sidebar.markdown(
     - Análise de dados do relatório de cadastro individual. 
     - Iniciativa - Ubs Flamengo
     - Acesso livre
-
     ### Links:
     ##
     ##### - [PEC SUS](https://sape.pecsus.com.br/) 📝
@@ -232,7 +227,6 @@ st.markdown(
 )
 st.divider()
 
-
 st.markdown(
     """
     ### :world_map: **Conhecendo a área de abrangencia daUBS Flamengo: (IBGE 2022)**
@@ -303,14 +297,6 @@ st.markdown(
             ## Dados de saúde Relatório de cadastro individual.
             """
 )
-#
-#
-#
-#
-#
-#
-#
-
 
 if uploaded_file is not None:
     dataframes = processar_arquivo(uploaded_file)
@@ -332,10 +318,6 @@ if uploaded_file is not None:
             df["Descrição"] = df["Descrição"].str.capitalize()
 else:
     st.sidebar.info("Adicione um arquivo .csv.")
-#
-#
-#
-#
 
 def criar_grafico_personalizado(df, x_col, y_col, titulo):
     """
@@ -515,7 +497,6 @@ st.info(
 
 add_vertical_space(1)
 
-
 with stylable_container(
     key="brain",
     css_styles="""
@@ -530,6 +511,7 @@ with stylable_container(
     """,
 ):
     st.image("./brain.png")
+
 # Widget expander para interação da IA
 with st.expander(" Converse com o 🤖 Zé Flamengo", expanded=True):
     client = Groq(
@@ -560,9 +542,7 @@ with st.expander(" Converse com o 🤖 Zé Flamengo", expanded=True):
                                 evite conversar sobre outros temas, sempre retornando ao tema da
                                 conversa.
                                 Aqui estão os resumos dos dataframes disponíveis:
-
                                 {RESUMO_DF}
-
                                 Use essas informações para responder às perguntas do usuário.
                                 Regras:
                                 1- Seja sempre cortês.
@@ -592,7 +572,6 @@ with st.expander(" Converse com o 🤖 Zé Flamengo", expanded=True):
         else:
             st.warning("Por favor, digite uma pergunta antes de enviar.")
 
-
 # Adicione este código fora do expander para evitar a reexecução do script ao pressionar Enter
 if "USER_CHAT_input" in st.session_state and st.session_state.USER_CHAT_input:
     st.session_state.USER_CHAT_input = ""