Update app.py

app.py

@@ -3,88 +3,210 @@ from sentence_transformers import SentenceTransformer
 from sklearn.metrics.pairwise import cosine_similarity
 import numpy as np
 
-# Carregar o modelo
+# Modelo para embeddings
 model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')
 
-# Definir perfis DISC base para comparação
-disc_profiles = {
-    'D (Dominância)': '''
-    - Direto e decisivo
-    - Focado em resultados
-    - Assume riscos
-    - Gosta de desafios
-    - Toma iniciativa
-    ''',
-    
-    'I (Influência)': '''
-    - Otimista e entusiasta
-    - Comunicativo
-    - Sociável
-    - Persuasivo
-    - Expressivo
-    ''',
-    
-    'S (Estabilidade)': '''
-    - Paciente e cooperativo
-    - Leal e confiável
-    - Calmo e consistente
-    - Bom ouvinte
-    - Trabalha bem em equipe
-    ''',
-    
-    'C (Conformidade)': '''
-    - Analítico e preciso
-    - Sistemático
-    - Focado em qualidade
-    - Atento aos detalhes
-    - Segue regras e procedimentos
-    '''
+# Questionário DISC
+disc_questions = [
+    {
+        "pergunta": "Em situações de trabalho em equipe, eu geralmente:",
+        "opcoes": [
+            ("D", "Assumo a liderança e tomo decisões rápidas"),
+            ("I", "Motivo o grupo e mantenho o ambiente animado"),
+            ("S", "Apoio os colegas e mantenho a harmonia"),
+            ("C", "Analiso detalhadamente antes de agir")
+        ]
+    },
+    {
+        "pergunta": "Quando enfrento um problema, eu:",
+        "opcoes": [
+            ("D", "Enfrento diretamente e busco soluções imediatas"),
+            ("I", "Discuto com outros e busco diferentes perspectivas"),
+            ("S", "Mantenho a calma e sigo um processo estabelecido"),
+            ("C", "Pesquiso todas as informações disponíveis primeiro")
+        ]
+    },
+    {
+        "pergunta": "Em reuniões, eu costumo:",
+        "opcoes": [
+            ("D", "Ir direto ao ponto e focar em resultados"),
+            ("I", "Participar ativamente e compartilhar ideias"),
+            ("S", "Ouvir atentamente e contribuir quando solicitado"),
+            ("C", "Tomar notas e questionar os detalhes")
+        ]
+    },
+    # Adicione mais perguntas conforme necessário
+]
+
+# Características de cada perfil para o relatório
+perfis_caracteristicas = {
+    'D': {
+        'pontos_fortes': [
+            "Capacidade de liderança",
+            "Foco em resultados",
+            "Tomada rápida de decisão",
+            "Assertividade",
+            "Iniciativa"
+        ],
+        'areas_desenvolvimento': [
+            "Paciência com outros",
+            "Sensibilidade às necessidades alheias",
+            "Diplomacia",
+            "Atenção aos detalhes"
+        ],
+        'ambiente_ideal': [
+            "Competitivo",
+            "Com autonomia",
+            "Orientado a resultados",
+            "Com desafios constantes"
+        ]
+    },
+    'I': {
+        'pontos_fortes': [
+            "Comunicação",
+            "Entusiasmo",
+            "Networking",
+            "Criatividade",
+            "Persuasão"
+        ],
+        'areas_desenvolvimento': [
+            "Organização",
+            "Foco",
+            "Conclusão de tarefas",
+            "Atenção aos detalhes"
+        ],
+        'ambiente_ideal': [
+            "Social",
+            "Colaborativo",
+            "Dinâmico",
+            "Com reconhecimento público"
+        ]
+    },
+    'S': {
+        'pontos_fortes': [
+            "Estabilidade",
+            "Confiabilidade",
+            "Trabalho em equipe",
+            "Paciência",
+            "Lealdade"
+        ],
+        'areas_desenvolvimento': [
+            "Adaptação a mudanças",
+            "Assertividade",
+            "Iniciativa",
+            "Tomada de decisão"
+        ],
+        'ambiente_ideal': [
+            "Estável",
+            "Previsível",
+            "Cooperativo",
+            "Com rotinas claras"
+        ]
+    },
+    'C': {
+        'pontos_fortes': [
+            "Precisão",
+            "Análise",
+            "Organização",
+            "Qualidade",
+            "Planejamento"
+        ],
+        'areas_desenvolvimento': [
+            "Flexibilidade",
+            "Velocidade",
+            "Expressão emocional",
+            "Tomada de risco"
+        ],
+        'ambiente_ideal': [
+            "Estruturado",
+            "Com regras claras",
+            "Focado em qualidade",
+            "Com tempo para análise"
+        ]
+    }
 }
 
-def analyze_disc(user_text):
-    # Gerar embeddings para o texto do usuário
-    user_embedding = model.encode([user_text])[0]
-    
-    # Gerar embeddings para cada perfil DISC
-    results = {}
-    for profile_name, profile_desc in disc_profiles.items():
-        profile_embedding = model.encode([profile_desc])[0]
-        
-        # Calcular similaridade
-        similarity = cosine_similarity([user_embedding], [profile_embedding])[0][0]
-        results[profile_name] = float(similarity)
-    
-    # Normalizar resultados para soma = 100%
-    total = sum(results.values())
-    normalized_results = {k: (v/total) * 100 for k, v in results.items()}
-    
-    # Ordenar resultados
-    sorted_results = dict(sorted(normalized_results.items(), key=lambda x: x[1], reverse=True))
-    
-    # Formatar resultado
-    output = "Análise do Perfil DISC:\n\n"
-    for profile, score in sorted_results.items():
-        output += f"{profile}: {score:.1f}%\n"
-        
-    return output
+def calcular_perfil(respostas):
+    # Contar frequência de cada perfil
+    contagem = {'D': 0, 'I': 0, 'S': 0, 'C': 0}
+    for resposta in respostas:
+        contagem[resposta] += 1
+    
+    # Calcular porcentagens
+    total = sum(contagem.values())
+    percentuais = {k: (v/total) * 100 for k, v in contagem.items()}
+    
+    return percentuais
+
+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]
+    
+    # Cabeçalho do relatório
+    relatorio = "RELATÓRIO DE PERFIL COMPORTAMENTAL DISC\n\n"
+    
+    # Resultados percentuais
+    relatorio += "DISTRIBUIÇÃO DOS PERFIS:\n"
+    for perfil, porcentagem in perfis_ordenados.items():
+        relatorio += f"{perfil}: {porcentagem:.1f}%\n"
+    
+    relatorio += f"\nPERFIL PREDOMINANTE: {perfil_dominante}\n\n"
+    
+    # Características do perfil dominante
+    relatorio += "PONTOS FORTES:\n"
+    for ponto in perfis_caracteristicas[perfil_dominante]['pontos_fortes']:
+        relatorio += f"- {ponto}\n"
+    
+    relatorio += "\nÁREAS DE DESENVOLVIMENTO:\n"
+    for area in perfis_caracteristicas[perfil_dominante]['areas_desenvolvimento']:
+        relatorio += f"- {area}\n"
+    
+    relatorio += "\nAMBIENTE IDEAL DE TRABALHO:\n"
+    for caracteristica in perfis_caracteristicas[perfil_dominante]['ambiente_ideal']:
+        relatorio += f"- {caracteristica}\n"
+    
+    # Recomendações personalizadas baseadas no modelo
+    texto_base = " ".join(perfis_caracteristicas[perfil_dominante]['pontos_fortes'])
+    embedding_base = model.encode([texto_base])[0]
+    
+    return relatorio
+
+def processar_questionario(respostas):
+    # Converter respostas em perfis
+    perfis_selecionados = [opcao.split()[0] for opcao in respostas]
+    
+    # Calcular percentuais
+    percentuais = calcular_perfil(perfis_selecionados)
+    
+    # Gerar relatório
+    relatorio = gerar_relatorio(percentuais)
+    
+    return relatorio
 
 # Interface Gradio
-iface = gr.Interface(
-    fn=analyze_disc,
-    inputs=gr.Textbox(
-        lines=5,
-        placeholder="Digite uma descrição do comportamento da pessoa..."
-    ),
-    outputs=gr.Textbox(),
-    title="Analisador de Perfil DISC",
-    description="Analise o perfil DISC baseado em uma descrição comportamental usando IA",
-    examples=[
-        ["Pessoa focada em resultados, toma decisões rápidas e gosta de desafios"],
-        ["Muito comunicativo, alegre e gosta de trabalhar com pessoas"],
-        ["Calmo, paciente e trabalha bem em equipe"],
-        ["Detalhista, analítico e segue procedimentos com precisão"]
-    ]
-)
+with gr.Blocks(title="Questionário DISC") as iface:
+    gr.Markdown("# Análise de Perfil Comportamental DISC")
+    
+    with gr.Group():
+        respostas = []
+        for i, questao in enumerate(disc_questions):
+            gr.Markdown(f"### {questao['pergunta']}")
+            respostas.append(
+                gr.Radio(
+                    choices=[f"{perfil} - {desc}" for perfil, desc in questao['opcoes']],
+                    label=f"Questão {i+1}"
+                )
+            )
+    
+    btn = gr.Button("Gerar Relatório")
+    output = gr.Textbox(label="Relatório", lines=20)
+    
+    btn.click(
+        fn=processar_questionario,
+        inputs=respostas,
+        outputs=output
+    )
 
 # Iniciar a interface
 iface.launch()