Update app.py

app.py

@@ -6,39 +6,132 @@ import numpy as np
 # Inicializar o modelo
 model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')
 
-# Base de conhecimento DISC para o modelo
-perfis_base = {
-    'D': '''
-    Perfil Dominância (D) é caracterizado por ser direto, decisivo, orientado a resultados.
-    Pessoas com alto D são determinadas, competitivas, assertivas e gostam de desafios.
-    Tendem a ser líderes naturais, focados em objetivos e rápidos na tomada de decisão.
-    Trabalham bem sob pressão e buscam autoridade e responsabilidade.
-    ''',
-    
-    'I': '''
-    Perfil Influência (I) é caracterizado por ser comunicativo, entusiasta e sociável.
-    Pessoas com alto I são otimistas, expressivas, persuasivas e gostam de interagir.
-    Tendem a ser carismáticas, motivadoras e habilidosas em networking.
-    Trabalham bem em equipe e são orientadas para pessoas e relacionamentos.
-    ''',
-    
-    'S': '''
-    Perfil Estabilidade (S) é caracterizado por ser paciente, consistente e cooperativo.
-    Pessoas com alto S são confiáveis, leais, bons ouvintes e prestativas.
-    Tendem a ser excelentes em suporte, mediação e manutenção da harmonia.
-    Trabalham bem em ambientes estáveis e valorizam relacionamentos duradouros.
-    ''',
-    
-    'C': '''
-    Perfil Conformidade (C) é caracterizado por ser analítico, preciso e sistemático.
-    Pessoas com alto C são detalhistas, organizadas e focadas em qualidade.
-    Tendem a ser especialistas, metódicas e orientadas por dados e processos.
-    Trabalham bem com regras e procedimentos, valorizando a precisão.
-    '''
+# Base de conhecimento para análise
+DISC_BASE = {
+    'D': {
+        'caracteristicas_principais': [
+            "liderança natural",
+            "foco em resultados",
+            "tomada rápida de decisão",
+            "assertividade",
+            "orientação para desafios"
+        ],
+        'pontos_fortes': [
+            "determinação",
+            "eficiência",
+            "iniciativa",
+            "coragem",
+            "competitividade"
+        ],
+        'areas_desenvolvimento': [
+            "paciência",
+            "empatia",
+            "diplomacia",
+            "escuta ativa",
+            "trabalho em equipe"
+        ],
+        'ambiente_ideal': [
+            "dinâmico",
+            "desafiador",
+            "com autonomia",
+            "orientado a resultados",
+            "competitivo"
+        ]
+    },
+    'I': {
+        'caracteristicas_principais': [
+            "comunicação expressiva",
+            "entusiasmo",
+            "sociabilidade",
+            "otimismo",
+            "persuasão"
+        ],
+        'pontos_fortes': [
+            "carisma",
+            "networking",
+            "criatividade",
+            "motivação",
+            "adaptabilidade social"
+        ],
+        'areas_desenvolvimento': [
+            "organização",
+            "foco",
+            "gestão do tempo",
+            "atenção a detalhes",
+            "conclusão de tarefas"
+        ],
+        'ambiente_ideal': [
+            "social",
+            "colaborativo",
+            "dinâmico",
+            "com reconhecimento",
+            "flexível"
+        ]
+    },
+    'S': {
+        'caracteristicas_principais': [
+            "estabilidade",
+            "cooperação",
+            "paciência",
+            "lealdade",
+            "consistência"
+        ],
+        'pontos_fortes': [
+            "confiabilidade",
+            "mediação",
+            "suporte",
+            "harmonia",
+            "trabalho em equipe"
+        ],
+        'areas_desenvolvimento': [
+            "proatividade",
+            "assertividade",
+            "adaptação a mudanças",
+            "tomada de decisão",
+            "expressão de opiniões"
+        ],
+        'ambiente_ideal': [
+            "estável",
+            "previsível",
+            "harmonioso",
+            "colaborativo",
+            "estruturado"
+        ]
+    },
+    'C': {
+        'caracteristicas_principais': [
+            "análise",
+            "precisão",
+            "organização",
+            "sistemático",
+            "atenção a detalhes"
+        ],
+        'pontos_fortes': [
+            "qualidade",
+            "planejamento",
+            "metodologia",
+            "processos",
+            "compliance"
+        ],
+        'areas_desenvolvimento': [
+            "flexibilidade",
+            "rapidez decisória",
+            "relacionamentos",
+            "expressão emocional",
+            "tolerância a riscos"
+        ],
+        'ambiente_ideal': [
+            "estruturado",
+            "organizado",
+            "com regras claras",
+            "focado em qualidade",
+            "metódico"
+        ]
+    }
 }
 
 # Questionário DISC
-disc_questions = [
+DISC_QUESTIONS = [
     {
         "pergunta": "Em situações de trabalho em equipe, eu geralmente:",
         "opcoes": [
@@ -86,16 +179,69 @@ disc_questions = [
     }
 ]
 
+def get_caracteristicas_principais(perfil):
+    """Retorna características principais do perfil."""
+    return ", ".join(DISC_BASE[perfil]['caracteristicas_principais'][:3])
+
+def get_caracteristicas_secundarias(perfil):
+    """Retorna características secundárias do perfil."""
+    return ", ".join(DISC_BASE[perfil]['caracteristicas_principais'][3:])
+
+def get_pontos_fortes(perfil):
+    """Retorna pontos fortes do perfil."""
+    return ", ".join(DISC_BASE[perfil]['pontos_fortes'])
+
+def get_contextos_ideais(perfil):
+    """Retorna contextos ideais para o perfil."""
+    return ", ".join(DISC_BASE[perfil]['ambiente_ideal'])
+
+def get_contribuicoes_principais(perfil):
+    """Retorna principais contribuições do perfil."""
+    contribuicoes = {
+        'D': "liderar projetos e tomar decisões estratégicas",
+        'I': "motivar equipes e criar engajamento",
+        'S': "manter a estabilidade e promover cooperação",
+        'C': "garantir qualidade e precisão nos processos"
+    }
+    return contribuicoes[perfil]
+
+def get_aplicacoes_praticas(perfil):
+    """Retorna aplicações práticas para o perfil."""
+    aplicacoes = {
+        'D': "gestão de projetos, tomada de decisão e resolução de problemas",
+        'I': "comunicação, vendas e desenvolvimento de equipes",
+        'S': "suporte, mediação e gestão de relacionamentos",
+        'C': "análise, planejamento e controle de qualidade"
+    }
+    return aplicacoes[perfil]
+
+def get_areas_desenvolvimento(perfil):
+    """Retorna áreas de desenvolvimento do perfil."""
+    return ", ".join(DISC_BASE[perfil]['areas_desenvolvimento'])
+
+def get_estrategias_desenvolvimento(perfil):
+    """Retorna estratégias de desenvolvimento para o perfil."""
+    estrategias = {
+        'D': "desenvolvimento de habilidades interpessoais e paciência",
+        'I': "aprimoramento do foco e da organização",
+        'S': "desenvolvimento de proatividade e adaptabilidade",
+        'C': "trabalho na flexibilidade e rapidez decisória"
+    }
+    return estrategias[perfil]
+
+def get_recomendacoes_praticas(perfil_principal, perfil_secundario):
+    """Retorna recomendações práticas baseadas na combinação de perfis."""
+    return f"desenvolvimento equilibrado entre as características {perfil_principal} e {perfil_secundario}"
+
 def calcular_perfil(respostas):
-    # Contar frequência de cada perfil
+    """Calcula os percentuais de cada perfil DISC."""
     contagem = {'D': 0, 'I': 0, 'S': 0, 'C': 0}
     
     for resposta in respostas:
-        if resposta:  # Verifica se a resposta não é None
-            perfil = resposta.split()[0]  # Pega a primeira letra (D, I, S ou C)
+        if resposta:
+            perfil = resposta.split()[0]
             contagem[perfil] += 1
     
-    # Calcular porcentagens
     total = sum(contagem.values())
     if total > 0:
         percentuais = {k: (v/total) * 100 for k, v in contagem.items()}
@@ -104,191 +250,63 @@ def calcular_perfil(respostas):
     
     return percentuais
 
-def gerar_paragrafo_modelo(perfil, percentual, contexto):
-    # Criar texto base para o modelo analisar
-    texto_base = f"{perfis_base[perfil]} Intensidade: {percentual:.1f}%. Contexto: {contexto}"
-    
-    # Gerar embedding do texto base
-    embedding_base = model.encode([texto_base])[0]
+def gerar_relatorio(percentuais):
+    """Gera o relatório completo baseado nos percentuais DISC."""
+    perfis_ordenados = dict(sorted(percentuais.items(), key=lambda x: x[1], reverse=True))
+    perfil_dominante = list(perfis_ordenados.keys())[0]
+    perfil_secundario = list(perfis_ordenados.keys())[1]
     
-    # Templates para cada contexto
-    templates = {
-        "visao_geral": [
-            "Este perfil demonstra forte orientação para [ASPECTO], com ênfase em [CARACTERÍSTICA].",
-            "A análise indica uma tendência natural para [ASPECTO], destacando-se em [CARACTERÍSTICA].",
-            "O comportamento predominante revela foco em [ASPECTO], com notável [CARACTERÍSTICA]."
-        ],
-        "pontos_fortes": [
-            "Seus pontos fortes incluem [ASPECTOS], sendo particularmente efetivo em [SITUAÇÃO].",
-            "Destaca-se principalmente por [ASPECTOS], com excelente capacidade para [SITUAÇÃO].",
-            "Suas principais competências envolvem [ASPECTOS], sendo muito eficaz em [SITUAÇÃO]."
-        ],
-        "recomendacoes": [
-            "Para desenvolvimento, pode focar em [ASPECTOS], buscando equilibrar [CARACTERÍSTICA].",
-            "As oportunidades de crescimento incluem [ASPECTOS], visando aprimorar [CARACTERÍSTICA].",
-            "O caminho de desenvolvimento sugere atenção a [ASPECTOS], para fortalecer [CARACTERÍSTICA]."
-        ]
-    }
+    # Gerar embedding para análise personalizada
+    texto_base = f"""
+    Perfil dominante {perfil_dominante}: {', '.join(DISC_BASE[perfil_dominante]['caracteristicas_principais'])}
+    Perfil secundário {perfil_secundario}: {', '.join(DISC_BASE[perfil_secundario]['caracteristicas_principais'])}
+    Intensidade principal: {perfis_ordenados[perfil_dominante]:.1f}%
+    Intensidade secundária: {perfis_ordenados[perfil_secundario]:.1f}%
+    """
     
-    # Selecionar template mais apropriado baseado na similaridade
-    melhor_template = max(templates[contexto],
-                         key=lambda t: cosine_similarity([embedding_base],
-                                                       [model.encode([t])[0]])[0][0])
+    embedding_base = model.encode([texto_base])[0]
     
-    return melhor_template.replace("[ASPECTO]", 
-                                 get_aspecto_principal(perfil, contexto)
-                                ).replace("[CARACTERÍSTICA]",
-                                        get_caracteristica_principal(perfil, contexto)
-                                ).replace("[ASPECTOS]",
-                                        get_aspectos(perfil, contexto)
-                                ).replace("[SITUAÇÃO]",
-                                        get_situacao(perfil, contexto))
-
-def get_aspecto_principal(perfil, contexto):
-    aspectos = {
-        'D': {
-            'visao_geral': "resultados e ação",
-            'pontos_fortes': "liderança e decisão",
-            'recomendacoes': "relacionamentos interpessoais"
-        },
-        'I': {
-            'visao_geral': "comunicação e influência",
-            'pontos_fortes': "persuasão e entusiasmo",
-            'recomendacoes': "organização e foco"
-        },
-        'S': {
-            'visao_geral': "estabilidade e cooperação",
-            'pontos_fortes': "suporte e consistência",
-            'recomendacoes': "adaptabilidade e assertividade"
-        },
-        'C': {
-            'visao_geral': "análise e precisão",
-            'pontos_fortes': "qualidade e processos",
-            'recomendacoes': "flexibilidade e velocidade"
-        }
-    }
-    return aspectos[perfil][contexto]
+    # Gerar parágrafos personalizados
+    p1 = f"""Seu perfil comportamental apresenta uma predominância do tipo {perfil_dominante} ({perfis_ordenados[perfil_dominante]:.1f}%), 
+    caracterizado por {get_caracteristicas_principais(perfil_dominante)}, com influência secundária do tipo {perfil_secundario} 
+    ({perfis_ordenados[perfil_secundario]:.1f}%), que adiciona elementos de {get_caracteristicas_principais(perfil_secundario)}."""
 
-def get_caracteristica_principal(perfil, contexto):
-    caracteristicas = {
-        'D': {
-            'visao_geral': "determinação e competitividade",
-            'pontos_fortes': "tomada rápida de decisão",
-            'recomendacoes': "desenvolvimento de empatia"
-        },
-        'I': {
-            'visao_geral': "sociabilidade e otimismo",
-            'pontos_fortes': "networking e motivação",
-            'recomendacoes': "gestão de tempo"
-        },
-        'S': {
-            'visao_geral': "confiabilidade e paciência",
-            'pontos_fortes': "trabalho em equipe",
-            'recomendacoes': "gestão de mudanças"
-        },
-        'C': {
-            'visao_geral': "organização e metodologia",
-            'pontos_fortes': "atenção aos detalhes",
-            'recomendacoes': "tomada de decisão rápida"
-        }
-    }
-    return caracteristicas[perfil][contexto]
+    p2 = f"""Seus principais pontos fortes incluem {get_pontos_fortes(perfil_dominante)}. 
+    Você tende a se destacar em ambientes que sejam {get_contextos_ideais(perfil_dominante)}, 
+    onde pode contribuir efetivamente através de {get_contribuicoes_principais(perfil_dominante)}. 
+    Suas habilidades são particularmente valiosas em situações que envolvam {get_aplicacoes_praticas(perfil_dominante)}."""
 
-def get_aspectos(perfil, contexto):
-    aspectos = {
-        'D': {
-            'visao_geral': "liderança, competitividade e iniciativa",
-            'pontos_fortes': "direcionamento, foco em resultados e autonomia",
-            'recomendacoes': "escuta ativa, paciência e colaboração"
-        },
-        'I': {
-            'visao_geral': "comunicação, entusiasmo e sociabilidade",
-            'pontos_fortes': "persuasão, networking e criatividade",
-            'recomendacoes': "organização, foco e conclusão de tarefas"
-        },
-        'S': {
-            'visao_geral': "estabilidade, cooperação e harmonia",
-            'pontos_fortes': "consistência, lealdade e suporte",
-            'recomendacoes': "proatividade, assertividade e adaptabilidade"
-        },
-        'C': {
-            'visao_geral': "análise, precisão e qualidade",
-            'pontos_fortes': "organização, planejamento e controle",
-            'recomendacoes': "flexibilidade, agilidade e relacionamentos"
-        }
-    }
-    return aspectos[perfil][contexto]
+    p3 = f"""Para potencializar seu desenvolvimento, considere focar em {get_areas_desenvolvimento(perfil_dominante)}. 
+    Recomenda-se atenção especial ao {get_estrategias_desenvolvimento(perfil_dominante)}, 
+    buscando um {get_recomendacoes_praticas(perfil_dominante, perfil_secundario)}."""
 
-def get_situacao(perfil, contexto):
-    situacoes = {
-        'D': {
-            'visao_geral': "ambientes desafiadores",
-            'pontos_fortes': "situações que exigem decisões rápidas",
-            'recomendacoes': "trabalho em equipe colaborativo"
-        },
-        'I': {
-            'visao_geral': "ambientes sociais",
-            'pontos_fortes': "situações que exigem persuasão",
-            'recomendacoes': "tarefas que exigem foco individual"
-        },
-        'S': {
-            'visao_geral': "ambientes estáveis",
-            'pontos_fortes': "situações que exigem cooperação",
-            'recomendacoes': "ambientes de mudança constante"
-        },
-        'C': {
-            'visao_geral': "ambientes estruturados",
-            'pontos_fortes': "situações que exigem precisão",
-            'recomendacoes': "decisões sob pressão"
-        }
-    }
-    return situacoes[perfil][contexto]
-
-def gerar_relatorio(percentuais):
-    # Ordenar perfis por porcentagem
-    perfis_ordenados = dict(sorted(percentuais.items(), key=lambda x: x[1], reverse=True))
-    perfil_dominante = list(perfis_ordenados.keys())[0]
-    
-    # Gerar três parágrafos usando o modelo
-    p1 = gerar_paragrafo_modelo(perfil_dominante, perfis_ordenados[perfil_dominante], "visao_geral")
-    p2 = gerar_paragrafo_modelo(perfil_dominante, perfis_ordenados[perfil_dominante], "pontos_fortes")
-    p3 = gerar_paragrafo_modelo(perfil_dominante, perfis_ordenados[perfil_dominante], "recomendacoes")
-    
-    # Combinar parágrafos com dados quantitativos
-    relatorio = f"""Análise de Perfil DISC
+    return f"""RELATÓRIO DE ANÁLISE COMPORTAMENTAL DISC
 
 {p1}
-Distribuição quantitativa: {', '.join([f'{k}: {v:.1f}%' for k, v in perfis_ordenados.items()])}
 
 {p2}
-Este perfil representa {perfis_ordenados[perfil_dominante]:.1f}% das suas respostas.
 
 {p3}
-"""
-    
-    return relatorio
+
+Distribuição completa dos perfis DISC:
+{', '.join([f'{k}: {v:.1f}%' for k, v in perfis_ordenados.items()])}"""
 
 def processar_questionario(*args):
-    # Recebe as respostas como argumentos individuais e converte para lista
+    """Processa as respostas do questionário e gera o relatório."""
     respostas = list(args)
-    
-    # Calcular percentuais
     percentuais = calcular_perfil(respostas)
-    
-    # Gerar relatório
-    relatorio = gerar_relatorio(percentuais)
-    
-    return relatorio
+    return gerar_relatorio(percentuais)
 
 # Interface Gradio
 with gr.Blocks(title="Análise de Perfil DISC") as iface:
     gr.Markdown("# Análise de Perfil Comportamental DISC")
+    gr.Markdown("### Responda às questões abaixo conforme seu comportamento mais natural")
     
     # Lista para armazenar os componentes de rádio
     radio_components = []
     
     # Criar questões
-    for i, questao in enumerate(disc_questions):
+    for i, questao in enumerate(DISC_QUESTIONS):
         gr.Markdown(f"### {questao['pergunta']}")
         radio = gr.Radio(
             choices=[f"{perfil} - {desc}" for perfil, desc in questao['opcoes']],