Update views/rci.py

views/rci.py

@@ -424,6 +424,32 @@ def create_word_doc(markdown_content):
     # Retornar o objeto BytesIO contendo o documento Word
     return docx_file
 
+dados_contagem = """
+                Aqui estão os dados demográficos e socioeconômicos relevantes da cidade
+                de Contagem, Minas Gerais, com base no Censo 2022 e outras fontes oficiais:
+                
+                - População total: 621.863 habitantes
+                - Densidade demográfica: 3.193,21 hab/km²
+                - Área territorial: 194,746 km²
+                - PIB per capita: R$ 54.136,41 (2021)
+                - 3º maior PIB de Minas Gerais
+                - Importante polo industrial da Região Metropolitana de Belo Horizonte
+                - Sede do CEASA-MG, o maior entreposto de hortifrutigranjeiros do Brasil
+                - 3ª cidade mais populosa de Minas Gerais
+                - 16ª cidade mais populosa da região Sudeste
+                - 33ª cidade mais populosa do Brasil
+                - Estimativa de população em idade escolar (4 a 17 anos): 22% do total
+                - Estimativa de idosos (60 anos ou mais): 14% do total
+                - Estimativa de distribuição por sexo: 52% mulheres, 48% homens
+                - Estimativa de distribuição racial: 50% pardos, 40% brancos, 9% pretos,
+                  1% outros
+                - Estimativa de pessoas com pelo menos uma deficiência: 8% da população
+                - Estimativa de prevalência de doenças crônicas: 25% da população adulta
+                - Número estimado de unidades de saúde: 80
+"""
+
+
+
 # Função para gerar a análise inicial
 def generate_initial_analysis(resumo_rci):
     """
@@ -458,42 +484,67 @@ def generate_initial_analysis(resumo_rci):
                 {
                     "role": "user",
                     "content": f"""
-                        Aqui estão os dados para análise:
-                        <resumo_rci>
-                        {resumo_rci}
-                        </resumo_rci>
-
-                        Analise cuidadosamente os dados fornecidos e crie um relatório
-                        seguindo o modelo apresentado. O relatório deve incluir as seguintes
-                        seções:
-                        1. Cabeçalho com informações gerais (data, cidadãos ativos, saída,
-                        mudança de território, óbito)
-                        2. Análise de Faixa Etária (Crianças de 0 a 2 anos, 0 a 4 anos, mulheres
-                        na faixa etária de preventivo 25 a 64 anos, mulheres na faixa etária de
-                        mamomagrafia 50a 69 anos, idosos acima de 60 anos total, masculino e feminino)
-                        3. Comparação de Sexo
-                        4. Análise de Raça / Cor
-                        5. Análise de Escolaridade
-                        6. Orientação Sexual
-                        7. Deficiências
-                        8. Situações de Saúde Gerais
-
-                        Para cada seção:
-                        - Calcule os totais e percentuais relevantes
-                        - Faça comparações quando apropriado (por exemplo, entre masculino
-                        e feminino)
-                        - Destaque, ao final as 4 informações mais significativas
-
-                        Formate seu relatório em markdown, usando negrito para títulos de seções e subtítulos.
-                        Use o título de Análise parcial, use listas com marcadores quando apropriado
-                        para melhorar a legibilidade.
-                        
-                    """,
+                                Aqui estão os dados para análise:
+                                    <resumo_rci>
+                                    {resumo_rci}
+                                    </resumo_rci>
+                                    
+                                Analise cuidadosamente os dados fornecidos e os dados de Contagem:
+                                    <resumo_contagem>
+                                    {dados_contagem}
+                                    </resumo_contagem>
+                                    
+                                Crie um relatório seguindo o modelo apresentado. O relatório deve
+                                incluir as seguintes seções:
+                                
+                                1. Informações gerais (data, cidadãos ativos, saída,
+                                mudança de território, óbito)
+                                
+                                2. Faixa Etária (Crianças de 0 a 2 anos, 0 a 4 anos,
+                                mulheres na faixa etária de preventivo 25 a 64 anos,
+                                mulheres na faixa etária de mamografia 50 a 69 anos,
+                                idosos acima de 60 anos total, masculino e feminino)
+                                
+                                3. Comparação de Sexo
+                                4. Análise de Raça / Cor
+                                5. Análise de Escolaridade
+                                6. Orientação Sexual
+                                7. Deficiências
+                                8. Situações de Saúde
+                                
+                                Para cada seção:
+                                - Calcule os totais e percentuais relevantes
+                                - Faça comparações quando apropriado (por exemplo,
+                                entre masculino e feminino)
+                                - Destaque, ao final, as 4 informações mais significativas
+                                - Formate seu relatório em markdown, usando negrito para títulos
+                                de seções e subtítulos. Use o título de Análise parcial, use listas
+                                com marcadores quando apropriado para melhorar a legibilidade.
+                                
+                                Após o relatório, forneça 5 insights relevantes, abordando:
+                                **Principais desafios de saúde pública para Contagem**
+                                **Necessidades específicas dos grupos etários (crianças, adultos e idosos)**
+                                **Implicações da distribuição racial e de gênero para políticas de saúde locais**
+                                **Estratégias de prevenção e promoção da saúde adequadas ao perfil da população**
+                                **Desafios educacionais e sua relação com a saúde da população de Contagem**
+                                
+                                Para cada insight:
+                                
+                                - Explique sua importância no contexto específico de Contagem
+                                - Sugira uma ação ou política que poderia ser implementada pela
+                                  administração municipal
+                                - Indique quais dados adicionais seriam úteis para uma análise
+                                  mais aprofundada da situação em Contagem
+                                - Considere as particularidades desta amostra da população de
+                                  Contagem em relação aos desafios típicos de cidades de porte
+                                  médio em Minas Gerais, quando possível.
+                                  """,
+
                 },
             ],
             model="llama3-70b-8192",
             temperature=0.2,
-            max_tokens=1500,
+            max_tokens=4000,
         )
         return initial_analysis.choices[0].message.content
     except Exception as e:
@@ -528,77 +579,31 @@ col1, col2 = st.columns(2)
 
 with col1:
     with st.expander("Resumo Inicial", expanded=True):
-        if "initial_analysis" in st.session_state:
+        if "initial_analysis" in st.session_state and st.session_state.initial_analysis:
             st.write("**Panorama Geral dos Dados:**")
             markdown_content = st.session_state.initial_analysis
         
-            html_content = f'''
-            <div style="height:600px; overflow-y: auto; padding: 10px; border: 1px solid #CCC;">
-                {markdown_content}
-            </div>
-            '''
+            if markdown_content and markdown_content.strip():
+                html_content = f'''
+                <div style="height:300px; overflow-y: auto; padding: 10px; border: 1px solid #CCC;">
+                    {markdown_content}
+                </div>
+                '''
         
-            st.markdown(html_content, unsafe_allow_html=True)
+                st.markdown(html_content, unsafe_allow_html=True)
         
-            # Botão para baixar o documento em formato Word
-            docx_file = create_word_doc(markdown_content)
-            st.download_button(
-                label="Baixar em formato Word",
-                data=docx_file.getvalue(),
-                file_name="analise_inicial.docx",
-                mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
-            )
+                # Botão para baixar o documento em formato Word
+                docx_file = create_word_doc(markdown_content)
+                st.download_button(
+                    label="Baixar em formato Word",
+                    data=docx_file.getvalue(),
+                    file_name="analise_inicial.docx",
+                    mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
+                )
+            else:
+                st.warning("Não há conteúdo disponível para gerar o documento.")
         else:
             st.write("**Clique em 'Gerar Resumo' para analisar os dados.**")
-    
-    # with st.expander("Resumo Inicial", expanded=True):
-    #     if "initial_analysis" in st.session_state:
-    #         st.write("**Panorama Geral dos Dados:**")
-    #         markdown_content = st.session_state.initial_analysis
-        
-    #         html_content = f'''
-    #         <div style="height:600px; overflow-y: auto; padding: 10px; border: 1px solid #CCC;">
-    #             {markdown_content}
-    #         </div>
-    #         '''
-        
-    #         st.markdown(html_content, unsafe_allow_html=True)
-        
-    #         # Botão para baixar o documento em formato Word
-    #         docx_file = create_word_doc(markdown_content)
-    #         st.download_button(
-    #             label="Baixar em formato Word",
-    #             data=docx_file,
-    #             file_name="analise_inicial.docx",
-    #             mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
-    #         )
-    #     else:
-    #         st.write("**Clique em 'Gerar Resumo' para analisar os dados.**")
-    
-        # if "initial_analysis" in st.session_state:
-        #     st.write("**Panorama Geral dos Dados:**")
-        #     markdown_content = st.session_state.initial_analysis
-
-        #     html_content = f'''
-        #     <div style="height:500px; overflow-y: auto; padding: 10px; border: 1px solid #CCC;">
-        #         {markdown_content}
-        #     </div>
-        #     '''
-
-        #     st.markdown(html_content, unsafe_allow_html=True)
-        # else:
-        #     st.write("**Clique em 'Gerar Resumo' para analisar os dados.**")
-            
-        # if "initial_analysis" in st.session_state:
-        #     st.write("**Panorama Geral dos Dados:**")
-        #     st.text_area(
-        #         "Análise Inicial",
-        #         value=st.session_state.initial_analysis,
-        #         height=300,
-        #         disabled=False,
-        #     )
-        # else:
-        #     st.write("**Clique em 'Gerar Resumo' para analisar os dados.**")
 
 with col2:
     with st.expander("Converse com o 🤖 Zé Flamengo", expanded=True):
@@ -704,3 +709,37 @@ with col2:
             st.session_state.groq_chat_history = []
             st.rerun()
 
+
+
+# f"""
+#     Aqui estão os dados para análise:
+#     <resumo_rci>
+#     {resumo_rci}
+#     </resumo_rci>
+
+#     Analise cuidadosamente os dados fornecidos e crie um relatório
+#     seguindo o modelo apresentado. O relatório deve incluir as seguintes
+#     seções:
+#     1. Informações gerais (data, cidadãos ativos, saída,
+#     mudança de território, óbito)
+#     2. Faixa Etária (Crianças de 0 a 2 anos, 0 a 4 anos, mulheres
+#     na faixa etária de preventivo 25 a 64 anos, mulheres na faixa etária de
+#     mamomagrafia 50a 69 anos, idosos acima de 60 anos total, masculino e feminino)
+#     3. Comparação de Sexo
+#     4. Análise de Raça / Cor
+#     5. Análise de Escolaridade
+#     6. Orientação Sexual
+#     7. Deficiências
+#     8. Situações de Saúde Gerais
+
+#     Para cada seção:
+#     - Calcule os totais e percentuais relevantes
+#     - Faça comparações quando apropriado (por exemplo, entre masculino
+#     e feminino)
+#     - Destaque, ao final as 4 informações mais significativas
+
+#     Formate seu relatório em markdown, usando negrito para títulos de seções e subtítulos.
+#     Use o título de Análise parcial, use listas com marcadores quando apropriado
+#     para melhorar a legibilidade.
+    
+# """,