Update views/pdf_chat.py

views/pdf_chat.py

@@ -1,7 +1,320 @@
-import streamlit as st
-
-
-st.title("Converse com seu PDF")
-
-
-st.write('## Em breve!')
+import os
+from typing import Generator
+
+import streamlit as st
+from groq import Groq
+from streamlit_extras.add_vertical_space import add_vertical_space
+from streamlit_option_menu import option_menu
+
+st.title("Utilidades")
+# Função principal
+def main():
+    # Configuração da página principal
+    st.set_page_config(page_title="Utilidades", layout="wide")
+
+    # Menu principal (multipage)
+    main_menu = ["Auxílio", "Calculadoras", "Sobre"]
+    main_choice = st.sidebar.selectbox("Menu Principal", main_menu)
+
+    if main_choice == "Home":
+        st.title("Página Inicial")
+        st.write("Bem-vindo à página inicial!")
+
+    elif main_choice == "Página Aninhada":
+        st.title("Página com Menu Aninhado")
+
+        # Menu aninhado usando streamlit-option-menu na horizontal
+        nested_menu = option_menu(
+            menu_title=None,
+            options=[
+                "Incidências de radiografia", "RAG", "Geral"
+            ],
+            icons=['lungs', 'gear', 'envelope'],
+            menu_icon="cast",
+            default_index=0,
+            orientation="horizontal",
+        )
+
+        # Conteúdo das subpáginas
+        if nested_menu == "Incidências de radiografia":
+            st.header("Subpágina 1")
+
+            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)
+        
+            if "messages" not in st.session_state:
+                st.session_state.messages = []
+        
+            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.
+            
+            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="🚨")
+        
+            elif nested_menu == "Subpágina 3":
+                st.header("Subpágina 3")
+                st.write("Conteúdo da Subpágina 3")
+        
+    elif main_choice == "Sobre":
+        st.header("Sobre")
+
+
+
+
+if __name__ == "__main__":
+    main()