Update app.py

app.py

@@ -3,6 +3,65 @@ import numpy as np
 from transformers import pipeline
 import pandas as pd
 
+# CSS personalizado para melhorar a aparência
+CUSTOM_CSS = """
+.container {
+    max-width: 1200px;
+    margin: 0 auto;
+    padding: 20px;
+}
+.header-box {
+    background: linear-gradient(135deg, #1e3c72 0%, #2a5298 100%);
+    color: white;
+    padding: 2rem;
+    border-radius: 10px;
+    margin-bottom: 2rem;
+    text-align: center;
+}
+.question-box {
+    background: #f8f9fa;
+    border: 1px solid #e9ecef;
+    border-radius: 8px;
+    padding: 1.5rem;
+    margin-bottom: 1rem;
+    box-shadow: 0 2px 4px rgba(0,0,0,0.05);
+}
+.category-title {
+    color: #2a5298;
+    border-bottom: 2px solid #2a5298;
+    padding-bottom: 0.5rem;
+    margin-bottom: 1rem;
+}
+.results-container {
+    background: white;
+    border-radius: 10px;
+    padding: 2rem;
+    box-shadow: 0 4px 6px rgba(0,0,0,0.1);
+}
+.profile-card {
+    background: #f8f9fa;
+    border-radius: 8px;
+    padding: 1.5rem;
+    margin-bottom: 1rem;
+    text-align: center;
+}
+.slider-container {
+    margin: 10px 0;
+}
+.button-primary {
+    background: #2a5298 !important;
+    color: white !important;
+    padding: 12px 24px !important;
+    font-weight: bold !important;
+}
+.results-title {
+    font-size: 1.5rem;
+    color: #2a5298;
+    text-align: center;
+    margin-bottom: 1rem;
+}
+"""
+
 class DiscAnalyzer:
     def __init__(self):
         self.disc_profiles = {
@@ -12,6 +71,90 @@ class DiscAnalyzer:
             'C': 'Conformidade'
         }
         
+        self.profile_descriptions = {
+            'D': """
+## Perfil Dominância (D)
+
+### Características Principais:
+- Direto e decisivo
+- Focado em resultados
+- Assume riscos calculados
+- Forte senso de liderança
+
+### Pontos Fortes:
+- Tomada de decisão rápida
+- Determinação
+- Orientação para resultados
+- Iniciativa
+
+### Sugestões de Desenvolvimento:
+- Desenvolver paciência
+- Considerar opiniões alheias
+- Praticar escuta ativa
+""",
+            'I': """
+## Perfil Influência (I)
+
+### Características Principais:
+- Comunicativo e entusiasta
+- Sociável e otimista
+- Persuasivo
+- Foco em relacionamentos
+
+### Pontos Fortes:
+- Networking
+- Comunicação
+- Motivação de equipes
+- Criatividade
+
+### Sugestões de Desenvolvimento:
+- Melhorar organização
+- Focar em detalhes
+- Gerenciar melhor o tempo
+""",
+            'S': """
+## Perfil Estabilidade (S)
+
+### Características Principais:
+- Paciente e consistente
+- Bom ouvinte
+- Leal e confiável
+- Trabalho em equipe
+
+### Pontos Fortes:
+- Cooperação
+- Confiabilidade
+- Suporte a equipe
+- Planejamento
+
+### Sugestões de Desenvolvimento:
+- Ser mais assertivo
+- Adaptar-se a mudanças
+- Expressar opiniões
+""",
+            'C': """
+## Perfil Conformidade (C)
+
+### Características Principais:
+- Analítico e preciso
+- Foco em qualidade
+- Organizado
+- Atenção aos detalhes
+
+### Pontos Fortes:
+- Análise crítica
+- Precisão
+- Organização
+- Controle de qualidade
+
+### Sugestões de Desenvolvimento:
+- Ser mais flexível
+- Tomar decisões mais rápidas
+- Desenvolver habilidades sociais
+"""
+        }
+        
+        # Inicializar modelo
         self.generator = pipeline(
             "text-generation",
             model="facebook/opt-125m",
@@ -19,12 +162,18 @@ class DiscAnalyzer:
         )
 
     def generate_llm_report(self, scores, perfil_principal):
+        """
+        Gera relatório personalizado usando modelo do Hugging Face baseado nos scores DISC.
+        Combina a saída do modelo com descrições predefinidas para maior qualidade.
+        """
+        base_description = self.profile_descriptions[perfil_principal]
+        
         prompt = f"""Analise o perfil DISC:
 D:{scores['D']:.1f}% I:{scores['I']:.1f}% S:{scores['S']:.1f}% C:{scores['C']:.1f}%
 Perfil Principal: {self.disc_profiles[perfil_principal]}
 """
         try:
-            output = self.generator(
+            llm_output = self.generator(
                 prompt,
                 max_length=300,
                 num_return_sequences=1,
@@ -32,11 +181,25 @@ Perfil Principal: {self.disc_profiles[perfil_principal]}
                 top_p=0.9,
                 do_sample=True
             )
-            return output[0]['generated_text'].replace(prompt, "").strip()
+            
+            ai_insights = llm_output[0]['generated_text'].replace(prompt, "").strip()
+            
+            # Combinar descrição base com insights da IA
+            complete_report = f"""
+{base_description}
+
+### Análise Personalizada:
+{ai_insights}
+"""
+            return complete_report
+            
         except Exception as e:
-            return f"Erro ao gerar relatório: {str(e)}"
+            return base_description
 
     def avaliar_disc(self, *args):
+        """
+        Avalia o perfil DISC baseado nas respostas do questionário.
+        """
         responses = [int(arg) for arg in args]
 
         scores = {
@@ -57,7 +220,7 @@ Perfil Principal: {self.disc_profiles[perfil_principal]}
         })
 
         return (
-            f"**{self.disc_profiles[perfil_principal]} ({perfil_principal})**",
+            f"### Seu perfil principal é: {self.disc_profiles[perfil_principal]} ({perfil_principal})",
             relatorio_llm,
             df,
             gr.update(visible=True),  # Mostrar seção de resultados
@@ -65,6 +228,9 @@ Perfil Principal: {self.disc_profiles[perfil_principal]}
         )
 
 def create_disc_interface():
+    """
+    Cria a interface Gradio para o analisador DISC.
+    """
     analyzer = DiscAnalyzer()
     
     with gr.Blocks(
@@ -73,22 +239,24 @@ def create_disc_interface():
             primary_hue="blue",
             secondary_hue="purple",
         ),
+        css=CUSTOM_CSS
     ) as interface:
         # Cabeçalho
-        with gr.Box():
+        with gr.Box(elem_classes=["header-box"]):
             gr.Markdown(
                 """
-                # 🎯 Analisador de Perfil DISC com IA
-                #### Descubra seu estilo comportamental com análise personalizada
+                # 🎯 Análise de Perfil DISC Profissional
+                #### Descubra suas características comportamentais através de IA
                 """
             )
         
         # Container do questionário
-        with gr.Box() as questionnaire_container:
+        with gr.Box(elem_classes=["container"]) as questionnaire_container:
             gr.Markdown(
                 """
                 ### 📝 Questionário DISC
                 Avalie cada afirmação de acordo com sua identificação:
+                
                 - 1 = Discordo totalmente
                 - 2 = Discordo parcialmente
                 - 3 = Concordo parcialmente
@@ -126,33 +294,36 @@ def create_disc_interface():
 
             sliders = []
             for categoria, perguntas in questions.items():
-                with gr.Box():
-                    gr.Markdown(f"#### {categoria}")
+                with gr.Box(elem_classes=["question-box"]):
+                    gr.Markdown(f"#### {categoria}", elem_classes=["category-title"])
                     for i, pergunta in enumerate(perguntas, 1):
-                        slider = gr.Slider(
-                            minimum=1,
-                            maximum=4,
-                            value=2,
-                            step=1,
-                            label=pergunta,
-                            interactive=True
-                        )
-                        sliders.append(slider)
+                        with gr.Box(elem_classes=["slider-container"]):
+                            slider = gr.Slider(
+                                minimum=1,
+                                maximum=4,
+                                value=2,
+                                step=1,
+                                label=pergunta,
+                                interactive=True
+                            )
+                            sliders.append(slider)
 
             analyze_btn = gr.Button(
                 "📊 Analisar Perfil",
                 variant="primary",
+                elem_classes=["button-primary"],
                 size="lg"
             )
 
         # Container dos resultados (inicialmente oculto)
-        with gr.Box(visible=False) as results_container:
-            gr.Markdown("### 📊 Resultados da Análise")
+        with gr.Box(visible=False, elem_classes=["results-container"]) as results_container:
+            gr.Markdown("### 📊 Resultados da Análise", elem_classes=["results-title"])
             
             with gr.Row():
                 with gr.Column(scale=1):
                     perfil_output = gr.Markdown(
-                        label="Perfil Principal"
+                        label="Perfil Principal",
+                        elem_classes=["profile-card"]
                     )
                 
             with gr.Row():
@@ -167,7 +338,8 @@ def create_disc_interface():
                     )
                 with gr.Column(scale=3):
                     relatorio_output = gr.Markdown(
-                        label="Análise Detalhada"
+                        label="Análise Detalhada",
+                        elem_classes=["report-content"]
                     )
             
             # Botão para novo teste