Update views/pdf_chat.py

views/pdf_chat.py

@@ -11,17 +11,15 @@ st.title("Utilidades")
 nested_menu = option_menu(
     menu_title=None,
     options=[
-        "Incidências radiológicas", "RAG", "Geral"
+        "Incidências radiológicas", "Calculadora TFG", "Escala de Risco de Coelho e Savassi", "Escore de Risco Global (ERG)"
     ],
-    icons=['lungs', 'gear', 'envelope'],
+    icons=['lungs', 'gear', 'envelope', 'heart'],
     menu_icon="cast",
     default_index=0,
     orientation="horizontal",
 )
 
 
-st.markdown('Calculadoras')
-
 
 # Conteúdo das subpáginas
 if nested_menu == "Incidências radiológicas":
@@ -137,14 +135,14 @@ if nested_menu == "Incidências radiológicas":
     Feito com ❤️ e ☕ por **[Dr. Guilherme]**
     """)
 
-elif nested_menu == "RAG":    
+elif nested_menu == "Calculadora TFG":    
     import pathlib
 
     import streamlit as st
     import streamlit.components.v1 as components
     from streamlit_extras.add_vertical_space import add_vertical_space
 
-    st.header("RAG")
+    st.header("Calculadora TFG")
     
     static_dir = pathlib.Path('static').absolute()
     html_with_js= ''
@@ -631,12 +629,11 @@ elif nested_menu == "RAG":
         with st.expander(label='Recomendações', expanded=False, icon="🔥"):
             st.markdown(recomendacoes, unsafe_allow_html=True)
 
-elif nested_menu == "Geral":
+elif nested_menu == "Escala de Risco de Coelho e Savassi":
     import pandas as pd
     import streamlit as st
     
-    st.header("Geral")
-    
+    st.header("Escala de Risco de Coelho e Savassi")
     
     STYLE = "static/style.css"
     with open(STYLE, "r", encoding="utf-8") as f:
@@ -686,7 +683,7 @@ elif nested_menu == "Geral":
     st.title("📊 Calculadora da Escala de Risco de Coelho e Savassi")
     
     # Criando colunas para organizar o layout
-    col1, col2 = st.columns(2)
+    col1, col2, col3 = st.columns(3)
     
     # Fatores de risco
     fatores = [{
@@ -753,72 +750,306 @@ elif nested_menu == "Geral":
                                         min_value=1,
                                         value=1)
     
+        # Botão para calcular o risco
+        if st.button("Calcular Risco", use_container_width=True, icon="🧮️" ):
+            pontuacao_total = sum(fator["peso"] for fator in fatores
+                                  if checkboxes[fator["id"]])
+        
+            # Cálculo do escore de cômodos por pessoa
+            escore_comodo_pessoa = num_comodos / num_moradores
+            if escore_comodo_pessoa < 1:
+                pontuacao_total += 3
+        
+            # Determinação da classificação de risco
+            if pontuacao_total <= 4:
+                classificacao = "Risco Menor"
+                classe_css = "risk-menor"
+                icone = "✅"
+            elif pontuacao_total <= 6:
+                classificacao = "Risco Médio"
+                classe_css = "risk-medio"
+                icone = "⚠️"
+            elif pontuacao_total <= 8:
+                classificacao = "Risco Maior"
+                classe_css = "risk-maior"
+                icone = "🚨"
+            else:
+                classificacao = "Risco Máximo"
+                classe_css = "risk-maximo"
+                icone = "☠️"
+        
+            # Exibição do resultado
+            st.markdown(f"""
+            <div class='risk-result {classe_css}'>
+                {icone} Classificação: {classificacao}<br>
+                📊 Pontuação: {pontuacao_total}<br>
+                🏠 Cômodos por pessoa: {escore_comodo_pessoa:.2f}
+            </div>
+            """,
+                        unsafe_allow_html=True)
+    
+            st.subheader("Detalhes da Pontuação")
+            detalhes = []
+            for fator in fatores:
+                if checkboxes[fator["id"]]:
+                    detalhes.append({"Fator": fator["label"], "Peso": fator["peso"]})
+            if escore_comodo_pessoa < 1:
+                detalhes.append({"Fator": "🏠 Menos de 1 cômodo por pessoa", "Peso": 3})
+        
+            df_detalhes = pd.DataFrame(detalhes)
+            st.table(df_detalhes)
+    
+    
+    with col3, st.expander("Sobre a Escala de Risco de Coelho e Savassi", icon='❤️', expanded=False):
+        st.markdown("""
+        ---
+        ### Sobre a Escala de Risco de Coelho e Savassi
+        
+        A Escala de Risco de Coelho e Savassi é uma ferramenta utilizada na Atenção Primária à Saúde no Brasil para avaliar o risco familiar em territórios de saúde. 
+        Ela foi desenvolvida pelos médicos brasileiros Marcos Alencar Abaide Balbinotti Coelho e Maria Inês Ribeiro de Miranda Savassi.
+        
+        #### Como interpretar os resultados:
+        - **Risco Menor (≤ 4 pontos)**: Acompanhamento de rotina
+        - **Risco Médio (5-6 pontos)**: Acompanhamento programado
+        - **Risco Maior (7-8 pontos)**: Acompanhamento prioritário
+        - **Risco Máximo (≥ 9 pontos)**: Acompanhamento intensivo
+        
+        Esta calculadora é uma ferramenta de apoio e não substitui a avaliação profissional.
+        """)
+
+elif nested_menu == "Escore de Risco Global (ERG)":
+    import streamlit as st
+    import pandas as pd
+    import numpy as np
+    
+    # CSS personalizado
+    st.markdown("""
+    <style>
+        .stRadio > label {
+            font-weight: bold;
+            color: #4A4A4A;
+        }
+        .stButton > button {
+            width: 100%;
+        }
+        .risk-result {
+            font-size: 24px;
+            font-weight: bold;
+            padding: 10px;
+            border-radius: 5px;
+            margin-top: 20px;
+            text-align: center;
+        }
+        .risk-baixo {
+            background-color: #2ecc71;
+            color: white;
+        }
+        .risk-intermediario {
+            background-color: #f39c12;
+            color: white;
+        }
+        .risk-alto {
+            background-color: #e74c3c;
+            color: white;
+        }
+        .risk-muito-alto {
+            background-color: #c0392b;
+            color: white;
+        }
+    </style>
+    """,
+                unsafe_allow_html=True)
+    
+    # Título
+    st.header("Escore de Risco Global (ERG)")
+    st.markdown("Recomendado pela Sociedade Brasileira de Cardiologia")
+    
+    # Criando colunas para organizar o layout
+    col1, col2 = st.columns(2)
+    
+    # Inputs
+    with col1:
+        st.subheader("Dados Pessoais")
+        sexo = st.radio("Sexo", ["Masculino", "Feminino"])
+        idade = st.number_input("Idade", min_value=30, max_value=74, value=45)
+        tabagismo = st.checkbox("Tabagismo")
+        diabetes = st.checkbox("Diabetes")
+    
+    with col2:
+        st.subheader("Dados Clínicos")
+        colesterol_total = st.number_input("Colesterol Total (mg/dL)",
+                                           min_value=0,
+                                           value=200)
+        hdl = st.number_input("HDL Colesterol (mg/dL)", min_value=0, value=50)
+        pas = st.number_input("Pressão Arterial Sistólica (mmHg)",
+                              min_value=0,
+                              value=120)
+    
+    
+    # Função para calcular o ERG
+    def calcular_erg(sexo, idade, tabagismo, diabetes, colesterol_total, hdl, pas):
+        pontos = 0
+    
+        # Idade
+        if sexo == "Masculino":
+            if idade < 35: pontos -= 1
+            elif 35 <= idade <= 39: pontos += 0
+            elif 40 <= idade <= 44: pontos += 1
+            elif 45 <= idade <= 49: pontos += 2
+            elif 50 <= idade <= 54: pontos += 3
+            elif 55 <= idade <= 59: pontos += 4
+            elif 60 <= idade <= 64: pontos += 5
+            elif 65 <= idade <= 69: pontos += 6
+            else: pontos += 7
+        else:  # Feminino
+            if idade < 35: pontos -= 9
+            elif 35 <= idade <= 39: pontos -= 4
+            elif 40 <= idade <= 44: pontos += 0
+            elif 45 <= idade <= 49: pontos += 3
+            elif 50 <= idade <= 54: pontos += 6
+            elif 55 <= idade <= 59: pontos += 7
+            elif 60 <= idade <= 64: pontos += 8
+            elif 65 <= idade <= 69: pontos += 8
+            else: pontos += 8
+    
+        # Colesterol Total
+        if sexo == "Masculino":
+            if colesterol_total < 160: pontos -= 3
+            elif 160 <= colesterol_total <= 199: pontos += 0
+            elif 200 <= colesterol_total <= 239: pontos += 1
+            elif 240 <= colesterol_total <= 279: pontos += 2
+            else: pontos += 3
+        else:  # Feminino
+            if colesterol_total < 160: pontos -= 2
+            elif 160 <= colesterol_total <= 199: pontos += 0
+            elif 200 <= colesterol_total <= 239: pontos += 1
+            elif 240 <= colesterol_total <= 279: pontos += 1
+            else: pontos += 3
+    
+        # HDL
+        if hdl < 40: pontos += 2
+        elif 40 <= hdl <= 49: pontos += 1
+        elif 50 <= hdl <= 59: pontos += 0
+        else: pontos -= 2
+    
+        # Pressão Arterial Sistólica
+        if pas < 120: pontos += 0
+        elif 120 <= pas <= 129: pontos += 0
+        elif 130 <= pas <= 139: pontos += 1
+        elif 140 <= pas <= 159: pontos += 2
+        else: pontos += 3
+    
+        # Diabetes
+        if diabetes:
+            if sexo == "Masculino": pontos += 2
+            else: pontos += 4
+    
+        # Tabagismo
+        if tabagismo:
+            if sexo == "Masculino": pontos += 2
+            else: pontos += 2
+    
+        return pontos
+    
+    
     # Botão para calcular o risco
     if st.button("Calcular Risco"):
-        pontuacao_total = sum(fator["peso"] for fator in fatores
-                              if checkboxes[fator["id"]])
-    
-        # Cálculo do escore de cômodos por pessoa
-        escore_comodo_pessoa = num_comodos / num_moradores
-        if escore_comodo_pessoa < 1:
-            pontuacao_total += 3
-    
-        # Determinação da classificação de risco
-        if pontuacao_total <= 4:
-            classificacao = "Risco Menor"
-            classe_css = "risk-menor"
-            icone = "✅"
-        elif pontuacao_total <= 6:
-            classificacao = "Risco Médio"
-            classe_css = "risk-medio"
-            icone = "⚠️"
-        elif pontuacao_total <= 8:
-            classificacao = "Risco Maior"
-            classe_css = "risk-maior"
-            icone = "🚨"
-        else:
-            classificacao = "Risco Máximo"
-            classe_css = "risk-maximo"
-            icone = "☠️"
+        pontos = calcular_erg(sexo, idade, tabagismo, diabetes, colesterol_total,
+                              hdl, pas)
+    
+        # Interpretação do risco
+        if sexo == "Masculino":
+            if pontos <= 5:
+                risco = "Baixo"
+                classe_css = "risk-baixo"
+                probabilidade = "< 5%"
+            elif 6 <= pontos <= 10:
+                risco = "Intermediário"
+                classe_css = "risk-intermediario"
+                probabilidade = "5 a 10%"
+            elif 11 <= pontos <= 14:
+                risco = "Alto"
+                classe_css = "risk-alto"
+                probabilidade = "10 a 20%"
+            else:
+                risco = "Muito Alto"
+                classe_css = "risk-muito-alto"
+                probabilidade = "> 20%"
+        else:  # Feminino
+            if pontos <= 7:
+                risco = "Baixo"
+                classe_css = "risk-baixo"
+                probabilidade = "< 5%"
+            elif 8 <= pontos <= 12:
+                risco = "Intermediário"
+                classe_css = "risk-intermediario"
+                probabilidade = "5 a 10%"
+            elif 13 <= pontos <= 17:
+                risco = "Alto"
+                classe_css = "risk-alto"
+                probabilidade = "10 a 20%"
+            else:
+                risco = "Muito Alto"
+                classe_css = "risk-muito-alto"
+                probabilidade = "> 20%"
     
         # Exibição do resultado
         st.markdown(f"""
         <div class='risk-result {classe_css}'>
-            {icone} Classificação: {classificacao}<br>
-            📊 Pontuação: {pontuacao_total}<br>
-            🏠 Cômodos por pessoa: {escore_comodo_pessoa:.2f}
+            Risco Cardiovascular: {risco}<br>
+            Pontuação: {pontos}<br>
+            Probabilidade de evento cardiovascular em 10 anos: {probabilidade}
         </div>
         """,
                     unsafe_allow_html=True)
     
         # Tabela de detalhes
         st.subheader("Detalhes da Pontuação")
-        detalhes = []
-        for fator in fatores:
-            if checkboxes[fator["id"]]:
-                detalhes.append({"Fator": fator["label"], "Peso": fator["peso"]})
-        if escore_comodo_pessoa < 1:
-            detalhes.append({"Fator": "🏠 Menos de 1 cômodo por pessoa", "Peso": 3})
-    
+        detalhes = [{
+            "Fator": "Idade",
+            "Valor": f"{idade} anos"
+        }, {
+            "Fator": "Sexo",
+            "Valor": sexo
+        }, {
+            "Fator": "Colesterol Total",
+            "Valor": f"{colesterol_total} mg/dL"
+        }, {
+            "Fator": "HDL Colesterol",
+            "Valor": f"{hdl} mg/dL"
+        }, {
+            "Fator": "Pressão Arterial Sistólica",
+            "Valor": f"{pas} mmHg"
+        }, {
+            "Fator": "Diabetes",
+            "Valor": "Sim" if diabetes else "Não"
+        }, {
+            "Fator": "Tabagismo",
+            "Valor": "Sim" if tabagismo else "Não"
+        }]
         df_detalhes = pd.DataFrame(detalhes)
         st.table(df_detalhes)
     
-    # Informações adicionais
-    st.markdown("""
-    ---
-    ### Sobre a Escala de Risco de Coelho e Savassi
-    
-    A Escala de Risco de Coelho e Savassi é uma ferramenta utilizada na Atenção Primária à Saúde no Brasil para avaliar o risco familiar em territórios de saúde. 
-    Ela foi desenvolvida pelos médicos brasileiros Marcos Alencar Abaide Balbinotti Coelho e Maria Inês Ribeiro de Miranda Savassi.
-    
-    #### Como interpretar os resultados:
-    - **Risco Menor (≤ 4 pontos)**: Acompanhamento de rotina
-    - **Risco Médio (5-6 pontos)**: Acompanhamento programado
-    - **Risco Maior (7-8 pontos)**: Acompanhamento prioritário
-    - **Risco Máximo (≥ 9 pontos)**: Acompanhamento intensivo
-    
-    Esta calculadora é uma ferramenta de apoio e não substitui a avaliação profissional.
-    """)
-    
-    
-    
+    with st.expander("Sobre o Escore de Risco Global (ERG)", icon='❤️', expanded=False):
+        st.markdown("""
+        ---
+        ### Sobre o Escore de Risco Global (ERG)
+        
+        O Escore de Risco Global (ERG) é a ferramenta principal recomendada pela Sociedade Brasileira de Cardiologia para avaliação do risco cardiovascular. Foi desenvolvido com base em dados da população brasileira e considera os seguintes fatores:
+        
+        - Idade
+        - Sexo
+        - Tabagismo
+        - Diabetes
+        - Níveis de colesterol total e HDL
+        - Pressão arterial sistólica
+        
+        #### Como interpretar os resultados:
+        - **Baixo Risco**: Probabilidade < 5% de eventos cardiovasculares em 10 anos
+        - **Risco Intermediário**: Probabilidade de 5 a 10% de eventos cardiovasculares em 10 anos
+        - **Alto Risco**: Probabilidade de 10 a 20% de eventos cardiovasculares em 10 anos
+        - **Muito Alto Risco**: Probabilidade > 20% de eventos cardiovasculares em 10 anos
+        
+        Esta calculadora é uma ferramenta de apoio e não substitui a avaliação profissional. Consulte sempre um médico para interpretação adequada dos resultados e recomendações personalizadas.
+        """)
+