Update views/pdf_chat.py

views/pdf_chat.py

@@ -8,312 +8,294 @@ from streamlit_option_menu import option_menu
 
 # Função principal
 def main():
-    st.set_page_config(page_title="Utilidades", layout="wide")
+    st.title("Calculadoras")
+
+    # Menu aninhado usando streamlit-option-menu na horizontal
+    nested_menu = option_menu(
+        menu_title=None,
+        options=[
+            "Incidências radiológicas", "RAG", "Geral"
+        ],
+        icons=['lungs', 'gear', 'envelope'],
+        menu_icon="cast",
+        default_index=0,
+        orientation="horizontal",
+    )
+
+    # Conteúdo das subpáginas
+    if nested_menu == "Incidências radiológicas":
+        st.header("Apoio clínico na solicitação de radiografias!")
+
+        c1, c2, c3, c4 = st.columns([1, 3, 3, 1])
+
+        with c1:
+            st.markdown("")
+
+        with c2:
+            st.header(":blue[Incidências Radiográficas] ☢️")
+
+        with c3:
+            st.image("icons/rx2.svg", width=400)
+        with c4:
+            st.markdown("")
+
+        add_vertical_space()
+        # # Descrição com estilo e emojis
+        st.markdown("""
+        ##### ✅ **Vamos melhorar a visualização das incidências médicas!**
+        """)
+
+        # Função para exibir regiões e incidências
+        def exibir_incidencias(regiao, incidencias):
+            st.markdown(f"## {regiao}")
+            for incidencia in incidencias:
+                st.markdown(f"- {incidencia}")
+
+        add_vertical_space(3)
+
+        st.markdown("### 🧍‍♂️ Selecione a Região do Corpo:")
+
+        incidencias_radiograficas = {
+            "👨‍⚕️ Tórax e Vias Aéreas": [
+                "👨‍⚕️ Tórax - PA e Lateral",
+                "🦴 Esterno – OAD e Lateral",
+                "🦴 Art. Esterno-Clavicular – PA, OAD e OAE",
+                "🦴 Costelas Posteriores – AP, OPD e OPE",
+                "🦴 Costelas Anteriores – PA, OAD e OPE",
+                "👃 Vias aéreas superiores - AP e Lateral",
+            ],
+            "💢 Abdome e Pelve": [
+                "💢 Abdome - AP em Decúbito Dorsal",
+                "💢 RAA - Tórax PA, Abdome em Decúbito Dorsal e Abdome em AP Ortostrica",
+                "🦵 Pelves (Bacia) - AP e AP perna de Rã, para traumatismo apenas AP",
+                "🦵 Quadril - AP e Axial Lateral",
+                "🦴 Art. Sacro-Ilíaca – Axial AP e Oblíquas",
+            ],
+            "✋ Membros Superiores": [
+                "✋ Dedos da mão do 2º ao 5º - PA, Lateral e Oblíqua",
+                "👍 Polegar - AP, Lateral e Oblíqua",
+                "🖐️ Mão - PA, Lateral e Oblíqua",
+                "✊ Mão e Punho para Idade Óssea - PA Comparativa",
+                "✊ Punho - PA, Lateral e Oblíqua",
+                "🦴 Antebraço - AP e Lateral",
+                "🦴 Cotovelo - AP e Lateral",
+                "🦴 Úmero - AP e Lateral",
+                "💪 Ombro - AP com rotação externa e interna",
+                "💪 Ombro em traumatismo - AP com rotação neutra, Lateral escapular ou Transtorácica",
+                "🦴 Clavícula - AP e AP Axial",
+                "🦴 Art. Acromioclavicular - AP Bilateral com e sem estresse",
+                "🦴 Escápula - AP e Lateral",
+            ],
+            "🦵 Membros Inferiores": [
+                "🦶 Dedos dos pés - AP, Lateral e Oblíqua",
+                "🦶 Pé - AP, Lateral e Oblíqua",
+                "🦶 Calcâneo - Axial e Lateral",
+                "🦵 Tornozelo - AP, AP do encaixe, Lateral e Oblíqua",
+                "🦿 Joelho - AP, Lateral e Oblíqua",
+                "🦿 Patela - PA, Lateral e Tangencial",
+                "🦿 Perna - AP e Lateral",
+                "🦿 Fêmur – AP e Lateral",
+            ],
+            "🦴 Coluna Vertebral": [
+                "🦴 Coluna Cervical - Axial AP, Transoral, Lateral e Oblíqua",
+                "🦴 Coluna Dorsal ou Torácica - AP e Lateral",
+                "🦴 Coluna Lombar - AP, Lateral e Oblíqua",
+                "🦴 Sacro e Cóccix - Axial AP e Lateral",
+                "🦴 Rotina para Escoliose – AP e Lateral",
+                "🦴 Rotina para Fusão Vertebral – AP com inclinação Esquerda e Direita Lateral com Hiperextensão e Hiperflexão",
+            ],
+            "🦷 Cabeça e Face": [
+                "👤 Crânio - AP Axial, PA e Lateral",
+                "👃 Ossos da Face - Caldwell, Waters e Lateral",
+                "👂 Ossos Nasais - Waters e Lateral",
+                "👄 Mandíbula - Axial Lateral, PA e Axial AP",
+                "🧠 Seios da Face - Caldwell, Waters e Lateral",
+                "👂 Arcos Zigomáticos – Axial AP Submentovertice e Tangencial",
+                "👂 Pirâmide – Rhêse, Caldwell e Waters",
+                "🦴 ATM's – Axial Lateral e Oblíqua Axial Lateral",
+                "🦴 Mastoides – Oblíqua Axial Lateral, Oblíqua Axial Anterior e Axial AP",
+                "🦴 Sela Turcica – Axial AP e Lateral",
+            ]
+        }
+
+        cx1, cx2 = st.columns([1, 1])
+
+        with cx1:
+            st.markdown("")
+            regioes = list(incidencias_radiograficas.keys())
+            regiao_escolhida = st.radio("", regioes)
+
+        with cx2:
+            if regiao_escolhida:
+                exibir_incidencias(
+                    regiao_escolhida,
+                    incidencias_radiograficas[regiao_escolhida])
+
+        st.markdown("""
+        ---
+        Feito com ❤️ e ☕ por **[Dr. Guilherme]**
+        """)
+
+    elif nested_menu == "RAG":
+        st.header("RAG")
+        my_secret = os.environ['GROQ_API_KEY']
+        client = Groq(api_key=my_secret)
     
-    st.title("Utilidades")
+        if "messages" not in st.session_state:
+            st.session_state.messages = []
     
-    main_menu = ["Auxílio clínico", "Calculadoras", "Sobre"]
-    main_choice = st.sidebar.selectbox("Menu Principal", main_menu)
-
-    if main_choice == "Auxílio clínico":
-        st.title("Auxílio clínico")
-        st.write("Bem-vindo à página inicial!")
-
-    elif main_choice == "Calculadoras":
-        st.title("Calculadoras")
-
-        # Menu aninhado usando streamlit-option-menu na horizontal
-        nested_menu = option_menu(
-            menu_title=None,
-            options=[
-                "Incidências radiológicas", "RAG", "Geral"
-            ],
-            icons=['lungs', 'gear', 'envelope'],
-            menu_icon="cast",
-            default_index=0,
-            orientation="horizontal",
-        )
-
-        # Conteúdo das subpáginas
-        if nested_menu == "Incidências radiológicas":
-            st.header("Apoio clínico na solicitação de radiografias!")
-
-            c1, c2, c3, c4 = st.columns([1, 3, 3, 1])
-
-            with c1:
-                st.markdown("")
-
-            with c2:
-                st.header(":blue[Incidências Radiográficas] ☢️")
-
-            with c3:
-                st.image("icons/rx2.svg", width=400)
-            with c4:
-                st.markdown("")
-
-            add_vertical_space()
-            # # Descrição com estilo e emojis
-            st.markdown("""
-            ##### ✅ **Vamos melhorar a visualização das incidências médicas!**
-            """)
-
-            # Função para exibir regiões e incidências
-            def exibir_incidencias(regiao, incidencias):
-                st.markdown(f"## {regiao}")
-                for incidencia in incidencias:
-                    st.markdown(f"- {incidencia}")
-
-            add_vertical_space(3)
-
-            st.markdown("### 🧍‍♂️ Selecione a Região do Corpo:")
-
-            incidencias_radiograficas = {
-                "👨‍⚕️ Tórax e Vias Aéreas": [
-                    "👨‍⚕️ Tórax - PA e Lateral",
-                    "🦴 Esterno – OAD e Lateral",
-                    "🦴 Art. Esterno-Clavicular – PA, OAD e OAE",
-                    "🦴 Costelas Posteriores – AP, OPD e OPE",
-                    "🦴 Costelas Anteriores – PA, OAD e OPE",
-                    "👃 Vias aéreas superiores - AP e Lateral",
-                ],
-                "💢 Abdome e Pelve": [
-                    "💢 Abdome - AP em Decúbito Dorsal",
-                    "💢 RAA - Tórax PA, Abdome em Decúbito Dorsal e Abdome em AP Ortostrica",
-                    "🦵 Pelves (Bacia) - AP e AP perna de Rã, para traumatismo apenas AP",
-                    "🦵 Quadril - AP e Axial Lateral",
-                    "🦴 Art. Sacro-Ilíaca – Axial AP e Oblíquas",
-                ],
-                "✋ Membros Superiores": [
-                    "✋ Dedos da mão do 2º ao 5º - PA, Lateral e Oblíqua",
-                    "👍 Polegar - AP, Lateral e Oblíqua",
-                    "🖐️ Mão - PA, Lateral e Oblíqua",
-                    "✊ Mão e Punho para Idade Óssea - PA Comparativa",
-                    "✊ Punho - PA, Lateral e Oblíqua",
-                    "🦴 Antebraço - AP e Lateral",
-                    "🦴 Cotovelo - AP e Lateral",
-                    "🦴 Úmero - AP e Lateral",
-                    "💪 Ombro - AP com rotação externa e interna",
-                    "💪 Ombro em traumatismo - AP com rotação neutra, Lateral escapular ou Transtorácica",
-                    "🦴 Clavícula - AP e AP Axial",
-                    "🦴 Art. Acromioclavicular - AP Bilateral com e sem estresse",
-                    "🦴 Escápula - AP e Lateral",
-                ],
-                "🦵 Membros Inferiores": [
-                    "🦶 Dedos dos pés - AP, Lateral e Oblíqua",
-                    "🦶 Pé - AP, Lateral e Oblíqua",
-                    "🦶 Calcâneo - Axial e Lateral",
-                    "🦵 Tornozelo - AP, AP do encaixe, Lateral e Oblíqua",
-                    "🦿 Joelho - AP, Lateral e Oblíqua",
-                    "🦿 Patela - PA, Lateral e Tangencial",
-                    "🦿 Perna - AP e Lateral",
-                    "🦿 Fêmur ��� AP e Lateral",
-                ],
-                "🦴 Coluna Vertebral": [
-                    "🦴 Coluna Cervical - Axial AP, Transoral, Lateral e Oblíqua",
-                    "🦴 Coluna Dorsal ou Torácica - AP e Lateral",
-                    "🦴 Coluna Lombar - AP, Lateral e Oblíqua",
-                    "🦴 Sacro e Cóccix - Axial AP e Lateral",
-                    "🦴 Rotina para Escoliose – AP e Lateral",
-                    "🦴 Rotina para Fusão Vertebral – AP com inclinação Esquerda e Direita Lateral com Hiperextensão e Hiperflexão",
-                ],
-                "🦷 Cabeça e Face": [
-                    "👤 Crânio - AP Axial, PA e Lateral",
-                    "👃 Ossos da Face - Caldwell, Waters e Lateral",
-                    "👂 Ossos Nasais - Waters e Lateral",
-                    "👄 Mandíbula - Axial Lateral, PA e Axial AP",
-                    "🧠 Seios da Face - Caldwell, Waters e Lateral",
-                    "👂 Arcos Zigomáticos – Axial AP Submentovertice e Tangencial",
-                    "👂 Pirâmide – Rhêse, Caldwell e Waters",
-                    "🦴 ATM's – Axial Lateral e Oblíqua Axial Lateral",
-                    "🦴 Mastoides – Oblíqua Axial Lateral, Oblíqua Axial Anterior e Axial AP",
-                    "🦴 Sela Turcica – Axial AP e Lateral",
-                ]
-            }
-
-            cx1, cx2 = st.columns([1, 1])
-
-            with cx1:
-                st.markdown("")
-                regioes = list(incidencias_radiograficas.keys())
-                regiao_escolhida = st.radio("", regioes)
-
-            with cx2:
-                if regiao_escolhida:
-                    exibir_incidencias(
-                        regiao_escolhida,
-                        incidencias_radiograficas[regiao_escolhida])
-
-            st.markdown("""
-            ---
-            Feito com ❤️ e ☕ por **[Dr. Guilherme]**
-            """)
-
-        elif nested_menu == "RAG":
-            st.header("RAG")
-            my_secret = os.environ['GROQ_API_KEY']
-            client = Groq(api_key=my_secret)
+        col1, col2 = st.columns(2)
+        with col1:
+            st.write('')
+        with col2:
+            st.markdown("### 📚  - Documentos, pesquisas e temas gerais!")
+    
+        prompt_model = """
+        Você é um assistente virtual especializado em trabalhar na Unidade Básica
+        de Saúde (UBS) chamada UBS Flamengo, em Contagem, Minas Gerais, Brasil. Sua
+        principal função é auxiliar profissionais de saúde e funcionários
+        administrativos com várias tarefas, sempre se comunicando em português 
+        formal do Brasil.
         
-            if "messages" not in st.session_state:
-                st.session_state.messages = []
+        Inicie cada interação perguntando o nome do usuário. Use o seguinte formato:
+        Olá! Bem-vindo(a) à UBS Flamengo. Meu nome é Zé Flamengo. 
+        Como posso te chamar?
         
-            col1, col2 = st.columns(2)
-            with col1:
-                st.write('')
-            with col2:
-                st.markdown("### 📚  - Documentos, pesquisas e temas gerais!")
+        Dirija-se ao usuário respeitosamente, usando o nome dele durante toda a 
+        conversa. Mantenha um tom profissional, cortês e empático em todas as 
+        interações. Adapte o nível de linguagem ao do usuário, sempre mantendo-se
+        claro e compreensível. Evite usar termos técnicos excessivos, a menos que
+        solicitados ou apropriados ao contexto.
+
+        Gere uma resposta com base nas seguintes diretrizes:
+        1. Se a consulta for sobre a preparação de relatórios, ofereça ajuda
+        para estruturar o relatório, sugerindo pontos-chave a serem incluídos
+        e fornecendo dicas para uma redação clara e concisa.
+        2. Para perguntas sobre diretrizes de saúde ou notas técnicas, explique
+        o conteúdo de forma simples, dividindo informações complexas em partes
+        facilmente digestíveis.
+        3. Ao resumir textos relacionados à saúde, forneça uma visão geral concisa
+        seguida de pontos-chave, usando marcadores para maior clareza.
+        4. Se for solicitado um insight ou análise, baseie sua resposta em
+        informações atualizadas e confiáveis, deixando claro que as informações
+        devem ser verificadas com fontes oficiais.
+        5. Para perguntas sobre procedimentos padrão da UBS, forneça explicações 
+        passo a passo, usando listas numeradas quando apropriado.
+        6. Se a consulta estiver além de seu conhecimento ou exigir experiência
+        médica, explique educadamente suas limitações e sugira que o usuário 
+        consulte um profissional de saúde qualificado.
+        7. Em caso de perguntas relacionadas a emergências, enfatize a importância
+        de procurar ajuda médica imediata e forneça informações básicas sobre como
+        entrar em contato com os serviços de emergência.
         
-            prompt_model = """
-            Você é um assistente virtual especializado em trabalhar na Unidade Básica
-            de Saúde (UBS) chamada UBS Flamengo, em Contagem, Minas Gerais, Brasil. Sua
-            principal função é auxiliar profissionais de saúde e funcionários
-            administrativos com várias tarefas, sempre se comunicando em português 
-            formal do Brasil.
-            
-            Inicie cada interação perguntando o nome do usuário. Use o seguinte formato:
-            Olá! Bem-vindo(a) à UBS Flamengo. Meu nome é Zé Flamengo. 
-            Como posso te chamar?
-            
-            Dirija-se ao usuário respeitosamente, usando o nome dele durante toda a 
-            conversa. Mantenha um tom profissional, cortês e empático em todas as 
-            interações. Adapte o nível de linguagem ao do usuário, sempre mantendo-se
-            claro e compreensível. Evite usar termos técnicos excessivos, a menos que
-            solicitados ou apropriados ao contexto.
-
-            Gere uma resposta com base nas seguintes diretrizes:
-            1. Se a consulta for sobre a preparação de relatórios, ofereça ajuda
-            para estruturar o relatório, sugerindo pontos-chave a serem incluídos
-            e fornecendo dicas para uma redação clara e concisa.
-            2. Para perguntas sobre diretrizes de saúde ou notas técnicas, explique
-            o conteúdo de forma simples, dividindo informações complexas em partes
-            facilmente digestíveis.
-            3. Ao resumir textos relacionados à saúde, forneça uma visão geral concisa
-            seguida de pontos-chave, usando marcadores para maior clareza.
-            4. Se for solicitado um insight ou análise, baseie sua resposta em
-            informações atualizadas e confiáveis, deixando claro que as informações
-            devem ser verificadas com fontes oficiais.
-            5. Para perguntas sobre procedimentos padrão da UBS, forneça explicações 
-            passo a passo, usando listas numeradas quando apropriado.
-            6. Se a consulta estiver além de seu conhecimento ou exigir experiência
-            médica, explique educadamente suas limitações e sugira que o usuário 
-            consulte um profissional de saúde qualificado.
-            7. Em caso de perguntas relacionadas a emergências, enfatize a importância
-            de procurar ajuda médica imediata e forneça informações básicas sobre como
-            entrar em contato com os serviços de emergência.
-            
-            Formate suas respostas da seguinte maneira:
-            - Use frases completas e estruturadas.
-            - Empregue marcadores ou numeração para listas ou instruções passo a passo.
-            - Para tópicos complexos, comece com uma visão geral breve seguida de 
-            informações mais detalhadas.
-            - Use exemplos práticos ou analogias para explicar conceitos difíceis, quando
-            relevante.
-            
-            Lembre-se destes pontos importantes:
-            - Nunca forneça diagnósticos médicos nem substitua o conselho de 
-            profissionais de saúde.
-            - Mantenha estrita confidencialidade e não solicite nem discuta informações 
-            pessoais de pacientes.
-            - Se você não souber algo sobre um tópico específico, admita honestamente e
-            sugira a consulta de um profissional qualificado.
-            - Fique atualizado sobre as políticas de saúde pública do Brasil e os 
-            procedimentos padrão da UBS.
-            - Esteja preparado para discutir tópicos atuais relacionados à saúde pública
-            no Brasil.
-            
-            Após fornecer sua resposta, pergunte sempre se o usuário tem mais perguntas
-            ou se precisa de mais assistência. Use o seguinte formato:
-            
-            Posso ajudar com mais alguma coisa, [Nome do Usuário]?
-            
-            Continue esse ciclo de interação até o usuário indicar que não tem mais
-            perguntas.
-            Ao finalizar a conversa, use uma saudação de encerramento educada, como:
-            
-            Obrigado por utilizar nossos serviços, [Nome do Usuário]. Tenha um ótimo
-            dia e não hesite em voltar se precisar de mais assistência.
-            Lembre-se de se comunicar sempre em português formal brasileiro durante
-            toda a interação.
-            """
-            if "messages" not in st.session_state:
-                st.session_state.messages = [{"role": "system", "content": prompt_model}]
-
-            def clear_all_session_state():
-                """
-                Limpa todos os estados da sessão.
-                """
-                for key in list(st.session_state.keys()):
-                    del st.session_state[key]
-
-            if st.sidebar.button('Limpar Estado da Sessão', type="primary", key='limpar_sessao_sidebar'):
-
-                clear_all_session_state()
-                st.rerun()
-
-            col1, col2 = st.columns(2)
-            with col1:
-                st.write('')
-            with col2:
-                st.markdown("")
-
-            for message in st.session_state.messages:
-                if message["role"] != "system":
-                    avatar = '🤖' if message["role"] == "assistant" else '👤'
-                    with st.chat_message(message["role"], avatar=avatar):
-                        st.markdown(message["content"])
-
-            def generate_chat_responses(chat_complete) -> Generator[str, None, None]:
-                """
-                Gera conteúdo de resposta do chat a partir da resposta da API Groq.
-                """
-                for chunk in chat_complete:
-                    if chunk.choices[0].delta.content:
-                        yield chunk.choices[0].delta.content
-
-            if prompt := st.chat_input("Digite seu prompt aqui..."):
-                st.session_state.messages.append({
-                    "role": "user",
-                    "content": prompt
-                })
-
-                with st.chat_message("user", avatar='👤'):
-                    st.markdown(prompt)
-
-                # Buscar resposta da API Groq
-                try:
-                    chat_complete = client.chat.completions.create(
-                        model='llama-3.2-90b-text-preview',
-                        messages=st.session_state.messages,
-                        max_tokens=8000,
-                        stream=True
-                    )
-
-                    # Usar a função geradora
-                    with st.chat_message("assistant", avatar="🤖"):
-                        response_placeholder = st.empty()
-                        full_response = ""
-                        for chunk in generate_chat_responses(chat_complete):
-                            full_response += chunk
-                            response_placeholder.markdown(full_response + "▌")
-                        response_placeholder.markdown(full_response)
-
-                    # Adicionar a resposta completa ao st.session_state.messages
-                    st.session_state.messages.append({
-                        "role": "assistant",
-                        "content": full_response
-                    })
-
-                except Exception as e:
-                    st.error(f"Ocorreu um erro: {e}", icon="🚨")
+        Formate suas respostas da seguinte maneira:
+        - Use frases completas e estruturadas.
+        - Empregue marcadores ou numeração para listas ou instruções passo a passo.
+        - Para tópicos complexos, comece com uma visão geral breve seguida de 
+        informações mais detalhadas.
+        - Use exemplos práticos ou analogias para explicar conceitos difíceis, quando
+        relevante.
         
-            elif nested_menu == "Subpágina 3":
-                st.header("Subpágina 3")
-                st.write("Conteúdo da Subpágina 3")
+        Lembre-se destes pontos importantes:
+        - Nunca forneça diagnósticos médicos nem substitua o conselho de 
+        profissionais de saúde.
+        - Mantenha estrita confidencialidade e não solicite nem discuta informações 
+        pessoais de pacientes.
+        - Se você não souber algo sobre um tópico específico, admita honestamente e
+        sugira a consulta de um profissional qualificado.
+        - Fique atualizado sobre as políticas de saúde pública do Brasil e os 
+        procedimentos padrão da UBS.
+        - Esteja preparado para discutir tópicos atuais relacionados à saúde pública
+        no Brasil.
         
-    elif main_choice == "Sobre":
-        st.header("Sobre")
+        Após fornecer sua resposta, pergunte sempre se o usuário tem mais perguntas
+        ou se precisa de mais assistência. Use o seguinte formato:
+        
+        Posso ajudar com mais alguma coisa, [Nome do Usuário]?
+        
+        Continue esse ciclo de interação até o usuário indicar que não tem mais
+        perguntas.
+        Ao finalizar a conversa, use uma saudação de encerramento educada, como:
+        
+        Obrigado por utilizar nossos serviços, [Nome do Usuário]. Tenha um ótimo
+        dia e não hesite em voltar se precisar de mais assistência.
+        Lembre-se de se comunicar sempre em português formal brasileiro durante
+        toda a interação.
+        """
+        if "messages" not in st.session_state:
+            st.session_state.messages = [{"role": "system", "content": prompt_model}]
+
+        def clear_all_session_state():
+            """
+            Limpa todos os estados da sessão.
+            """
+            for key in list(st.session_state.keys()):
+                del st.session_state[key]
+
+        if st.sidebar.button('Limpar Estado da Sessão', type="primary", key='limpar_sessao_sidebar'):
+
+            clear_all_session_state()
+            st.rerun()
 
+        col1, col2 = st.columns(2)
+        with col1:
+            st.write('')
+        with col2:
+            st.markdown("")
 
+        for message in st.session_state.messages:
+            if message["role"] != "system":
+                avatar = '🤖' if message["role"] == "assistant" else '👤'
+                with st.chat_message(message["role"], avatar=avatar):
+                    st.markdown(message["content"])
 
+        def generate_chat_responses(chat_complete) -> Generator[str, None, None]:
+            """
+            Gera conteúdo de resposta do chat a partir da resposta da API Groq.
+            """
+            for chunk in chat_complete:
+                if chunk.choices[0].delta.content:
+                    yield chunk.choices[0].delta.content
+
+        if prompt := st.chat_input("Digite seu prompt aqui..."):
+            st.session_state.messages.append({
+                "role": "user",
+                "content": prompt
+            })
+
+            with st.chat_message("user", avatar='👤'):
+                st.markdown(prompt)
+
+            # Buscar resposta da API Groq
+            try:
+                chat_complete = client.chat.completions.create(
+                    model='llama-3.2-90b-text-preview',
+                    messages=st.session_state.messages,
+                    max_tokens=8000,
+                    stream=True
+                )
+
+                # Usar a função geradora
+                with st.chat_message("assistant", avatar="🤖"):
+                    response_placeholder = st.empty()
+                    full_response = ""
+                    for chunk in generate_chat_responses(chat_complete):
+                        full_response += chunk
+                        response_placeholder.markdown(full_response + "▌")
+                    response_placeholder.markdown(full_response)
+
+                # Adicionar a resposta completa ao st.session_state.messages
+                st.session_state.messages.append({
+                    "role": "assistant",
+                    "content": full_response
+                })
+
+            except Exception as e:
+                st.error(f"Ocorreu um erro: {e}", icon="🚨")
+    
+    elif nested_menu == "Geral":
+        st.header("Geral")
+        st.write("Conteúdo da Subpágina 3")
 
 if __name__ == "__main__":
     main()