Update modules/morphosyntax/morphosyntax_interface.py

modules/morphosyntax/morphosyntax_interface.py

@@ -5,8 +5,7 @@ import re
 import logging
 from spacy import displacy
 
-# Se asume que la función perform_advanced_morphosyntactic_analysis
-# y los métodos store_student_morphosyntax_base/iteration existen.
+# Funciones de análisis y DB que ya tienes en tus módulos
 from ..morphosyntax.morphosyntax_process import perform_advanced_morphosyntactic_analysis
 from ..database.morphosyntax_iterative_mongo_db import (
     store_student_morphosyntax_base,
@@ -15,12 +14,9 @@ from ..database.morphosyntax_iterative_mongo_db import (
 
 logger = logging.getLogger(__name__)
 
-
 ###########################################################################
 def initialize_arc_analysis_state():
-    """
-    Inicializa el estado de análisis de arcos (base e iteraciones) si no existe.
-    """
+    """Inicializa el estado de análisis de arcos (base e iteraciones) si no existe."""
     if "arc_analysis_state" not in st.session_state:
         st.session_state.arc_analysis_state = {
             "base_id": None,
@@ -33,9 +29,7 @@ def initialize_arc_analysis_state():
 
 ###########################################################################
 def reset_arc_analysis_state():
-    """
-    Resetea completamente el estado de análisis de arcos.
-    """
+    """Resetea completamente el estado de análisis de arcos."""
     st.session_state.arc_analysis_state = {
         "base_id": None,
         "base_text": "",
@@ -49,8 +43,7 @@ def reset_arc_analysis_state():
 def display_arc_diagram(doc):
     """
     Genera y retorna el HTML del diagrama de arco para un `Doc` de spaCy.
-    No imprime directamente en pantalla; regresa el HTML para 
-    usar con `st.write(..., unsafe_allow_html=True)`.
+    No imprime directamente; retorna el HTML para usar con `st.write(...)`.
     """
     try:
         diagram_html = ""
@@ -126,7 +119,7 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
 
         # Botón para iniciar nuevo análisis
         if st.button("Nuevo Análisis", key="btn_reset_base"):
-            # Si requieres recargar la app por completo:
+            # Si requieres recargar la app por completo, podrías descomentar:
             # st.experimental_rerun()
             reset_arc_analysis_state()
 
@@ -146,9 +139,8 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
                 try:
                     # Procesar con spaCy
                     doc = nlp_models[lang_code](arc_state["base_text"])
-                    # Generar HTML del arco
-                    arc_html = display_arc_diagram(doc)
-                    arc_state["base_diagram"] = arc_html
+                    base_arc_html = display_arc_diagram(doc)
+                    arc_state["base_diagram"] = base_arc_html
 
                     # Guardar en Mongo
                     analysis = perform_advanced_morphosyntactic_analysis(
@@ -178,27 +170,28 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
     with tabs[1]:
         st.subheader("Análisis de Cambios / Iteraciones")
 
-        # Verificar que exista texto base analizado
+        # Verificar que exista un texto base
         if not arc_state["base_id"]:
             st.info("Primero analiza un texto base en la pestaña anterior.")
             return
 
-        # Mostrar texto base como referencia (solo lectura)
-        st.text_area(
-            "Texto Base (solo lectura)",
-            value=arc_state["base_text"],
-            height=80,
-            disabled=True
-        )
+        # --- 1) Mostrar SIEMPRE el diagrama base arriba ---
+        st.markdown("#### Diagrama de Arco (Texto Base)")
+        if arc_state["base_diagram"]:
+            st.write(arc_state["base_diagram"], unsafe_allow_html=True)
+        else:
+            st.info("No hay diagrama base disponible.")
 
-        # Caja de texto para la iteración
+        # --- 2) Caja de texto para la iteración ---
+        st.markdown("<hr class='divider'>", unsafe_allow_html=True)
+        st.subheader("Texto de Iteración")
         arc_state["iteration_text"] = st.text_area(
-            "Texto de Iteración",
+            "Ingrese su nueva versión / iteración",
             value=arc_state["iteration_text"],
             height=150
         )
 
-        # Botón analizar iteración
+        # Botón para analizar iteración
         if st.button("Analizar Cambios", key="btn_analyze_iteration"):
             if not arc_state["iteration_text"].strip():
                 st.warning("Ingrese texto de iteración.")
@@ -228,17 +221,8 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
                     st.error("Error procesando iteración")
                     logger.error(f"Error en iteración: {str(exc)}")
 
-        # Comparación vertical si tenemos ambos diagramas
-        if arc_state["base_diagram"] and arc_state["iteration_diagram"]:
+        # --- 3) Mostrar diagrama de iteración debajo ---
+        if arc_state["iteration_diagram"]:
             st.markdown("<hr class='divider'>", unsafe_allow_html=True)
-            st.markdown("### Comparación Vertical: Base vs. Iteración")
-
-            # Diagrama base
-            st.markdown("**Diagrama Base**")
-            st.write(arc_state["base_diagram"], unsafe_allow_html=True)
-
-            st.markdown("---")
-
-            # Diagrama iterado
-            st.markdown("**Diagrama Iterado**")
+            st.markdown("#### Diagrama de Arco (Iteración)")
             st.write(arc_state["iteration_diagram"], unsafe_allow_html=True)