Update modules/studentact/current_situation_interface.py

modules/studentact/current_situation_interface.py

@@ -33,6 +33,37 @@ plt.rcParams['axes.spines.right'] = False
 logger = logging.getLogger(__name__)
 ####################################
 
+TEXT_TYPES = {
+    'academic_article': {
+        'name': 'Artículo Académico',
+        'thresholds': {
+            'vocabulary': {'min': 0.70, 'target': 0.85},
+            'structure': {'min': 0.75, 'target': 0.90},
+            'cohesion': {'min': 0.65, 'target': 0.80},
+            'clarity': {'min': 0.70, 'target': 0.85}
+        }
+    },
+    'student_essay': {
+        'name': 'Trabajo Universitario',
+        'thresholds': {
+            'vocabulary': {'min': 0.60, 'target': 0.75},
+            'structure': {'min': 0.65, 'target': 0.80},
+            'cohesion': {'min': 0.55, 'target': 0.70},
+            'clarity': {'min': 0.60, 'target': 0.75}
+        }
+    },
+    'general_communication': {
+        'name': 'Comunicación General',
+        'thresholds': {
+            'vocabulary': {'min': 0.50, 'target': 0.65},
+            'structure': {'min': 0.55, 'target': 0.70},
+            'cohesion': {'min': 0.45, 'target': 0.60},
+            'clarity': {'min': 0.50, 'target': 0.65}
+        }
+    }
+}
+####################################
+
 def display_current_situation_interface(lang_code, nlp_models, t):
     """
     Interfaz simplificada con gráfico de radar para visualizar métricas.
@@ -48,6 +79,17 @@ def display_current_situation_interface(lang_code, nlp_models, t):
         st.session_state.current_metrics = None
 
     # st.markdown("## Análisis Inicial de Escritura")
+
+    # Añadir selector de tipo de texto
+    text_type = st.selectbox(
+        "Tipo de texto a analizar:",
+        options=list(TEXT_TYPES.keys()),
+        format_func=lambda x: TEXT_TYPES[x]['name'],
+        help="Selecciona el tipo de texto para ajustar los criterios de evaluación"
+    )
+    
+    # Guardar el tipo seleccionado en session_state
+    st.session_state.current_text_type = text_type
     
     try:
         # Container principal con dos columnas
@@ -117,16 +159,40 @@ def display_results(metrics):
         # Crear dos columnas para métricas y gráfico
         metrics_col, graph_col = st.columns([1, 1.5])
         
+        # Configuración de métricas con sus umbrales
         metrics_config = [
-            ("Vocabulario", metrics['vocabulary']['normalized_score'], "Riqueza y variedad del vocabulario"),
-            ("Estructura", metrics['structure']['normalized_score'], "Organización y complejidad de oraciones"),
-            ("Cohesión", metrics['cohesion']['normalized_score'], "Conexión y fluidez entre ideas"),
-            ("Claridad", metrics['clarity']['normalized_score'], "Facilidad de comprensión del texto")
+            {
+                'label': "Vocabulario",
+                'key': 'vocabulary',
+                'value': metrics['vocabulary']['normalized_score'],
+                'help': "Riqueza y variedad del vocabulario",
+                'thresholds': {'min': 0.60, 'target': 0.75}
+            },
+            {
+                'label': "Estructura",
+                'key': 'structure',
+                'value': metrics['structure']['normalized_score'],
+                'help': "Organización y complejidad de oraciones",
+                'thresholds': {'min': 0.65, 'target': 0.80}
+            },
+            {
+                'label': "Cohesión",
+                'key': 'cohesion',
+                'value': metrics['cohesion']['normalized_score'],
+                'help': "Conexión y fluidez entre ideas",
+                'thresholds': {'min': 0.55, 'target': 0.70}
+            },
+            {
+                'label': "Claridad",
+                'key': 'clarity',
+                'value': metrics['clarity']['normalized_score'],
+                'help': "Facilidad de comprensión del texto",
+                'thresholds': {'min': 0.60, 'target': 0.75}
+            }
         ]
 
         # Mostrar métricas verticalmente
         with metrics_col:
-            # Contenedor con bordes para las métricas
             st.markdown("""
                 <style>
                 .metric-container {
@@ -139,52 +205,70 @@ def display_results(metrics):
                 </style>
             """, unsafe_allow_html=True)
             
-            for label, value, help_text in metrics_config:
+            for metric in metrics_config:
                 with st.container():
+                    # Determinar estado basado en umbrales
+                    value = metric['value']
+                    if value < metric['thresholds']['min']:
+                        status = "⚠️ Por mejorar"
+                        color = "inverse"
+                    elif value < metric['thresholds']['target']:
+                        status = "📈 Aceptable"
+                        color = "off"
+                    else:
+                        status = "✅ Óptimo"
+                        color = "normal"
+                    
                     st.metric(
-                        label,
+                        metric['label'],
                         f"{value:.2f}",
-                        "Meta: 1.00",
-                        delta_color="off",
-                        help=help_text
+                        f"{status} (Meta: {metric['thresholds']['target']:.2f})",
+                        delta_color=color,
+                        help=metric['help']
                     )
                     st.markdown("<div style='margin-bottom: 1rem;'></div>", unsafe_allow_html=True)
 
         # Gráfico radar en la columna derecha
         with graph_col:
             # Preparar datos para el gráfico
-            categories = [m[0] for m in metrics_config]
-            values_user = [m[1] for m in metrics_config]
-            values_pattern = [1.0] * len(categories)
+            categories = [m['label'] for m in metrics_config]
+            values_user = [m['value'] for m in metrics_config]
+            min_values = [m['thresholds']['min'] for m in metrics_config]
+            target_values = [m['thresholds']['target'] for m in metrics_config]
 
             # Crear y configurar gráfico
-            fig = plt.figure(figsize=(8, 8))  # Aumentado el tamaño
+            fig = plt.figure(figsize=(8, 8))
             ax = fig.add_subplot(111, projection='polar')
 
-            # Configurar gráfico radar
+            # Configurar radar
             angles = [n / float(len(categories)) * 2 * np.pi for n in range(len(categories))]
             angles += angles[:1]
             values_user += values_user[:1]
-            values_pattern += values_pattern[:1]
+            min_values += min_values[:1]
+            target_values += target_values[:1]
 
             # Configurar ejes
             ax.set_xticks(angles[:-1])
-            ax.set_xticklabels(categories, fontsize=10)  # Aumentado el tamaño de fuente
+            ax.set_xticklabels(categories, fontsize=10)
             circle_ticks = np.arange(0, 1.1, 0.2)
             ax.set_yticks(circle_ticks)
             ax.set_yticklabels([f'{tick:.1f}' for tick in circle_ticks], fontsize=8)
             ax.set_ylim(0, 1)
 
-            # Dibujar gráfico con colores mejorados
-            ax.plot(angles, values_pattern, '#2ecc71', linestyle='--', linewidth=1, label='Patrón', alpha=0.5)
-            ax.fill(angles, values_pattern, '#2ecc71', alpha=0.1)
+            # Dibujar áreas de umbrales
+            ax.plot(angles, min_values, '#e74c3c', linestyle='--', linewidth=1, label='Mínimo', alpha=0.5)
+            ax.plot(angles, target_values, '#2ecc71', linestyle='--', linewidth=1, label='Meta', alpha=0.5)
+            ax.fill_between(angles, target_values, [1]*len(angles), color='#2ecc71', alpha=0.1)
+            ax.fill_between(angles, [0]*len(angles), min_values, color='#e74c3c', alpha=0.1)
+
+            # Dibujar valores del usuario
             ax.plot(angles, values_user, '#3498db', linewidth=2, label='Tu escritura')
             ax.fill(angles, values_user, '#3498db', alpha=0.2)
-            
+
             # Ajustar leyenda
             ax.legend(
-                loc='upper right', 
-                bbox_to_anchor=(0.1, 0.1), 
+                loc='upper right',
+                bbox_to_anchor=(0.1, 0.1),
                 fontsize=10,
                 frameon=True,
                 facecolor='white',