Update modules/morphosyntax/morphosyntax_interface.py

modules/morphosyntax/morphosyntax_interface.py

@@ -5,12 +5,9 @@ import re
 import logging
 from spacy import displacy
 
-# Importa tu pipeline de spacy, por ejemplo:
-# nlp_models = {"es": spacy.load("es_core_news_sm")}
-
-# Supongamos que estas funciones existen en tus otros módulos:
+# Se asume que la función perform_advanced_morphosyntactic_analysis
+# y los métodos store_student_morphosyntax_base/iteration existen.
 from ..morphosyntax.morphosyntax_process import perform_advanced_morphosyntactic_analysis
-
 from ..database.morphosyntax_iterative_mongo_db import (
     store_student_morphosyntax_base,
     store_student_morphosyntax_iteration,
@@ -25,11 +22,11 @@ def initialize_arc_analysis_state():
     """
     if "arc_analysis_state" not in st.session_state:
         st.session_state.arc_analysis_state = {
-            "base_id": None,            # ObjectId del documento base
-            "base_text": "",            # Texto inicial
-            "base_diagram": None,       # HTML del diagrama base
-            "iteration_text": "",       # Texto de iteración
-            "iteration_diagram": None,  # HTML del diagrama de iteración
+            "base_id": None,
+            "base_text": "",
+            "base_diagram": None,
+            "iteration_text": "",
+            "iteration_diagram": None,
         }
         logger.info("Estado de análisis de arcos inicializado.")
 
@@ -45,13 +42,14 @@ def reset_arc_analysis_state():
         "iteration_text": "",
         "iteration_diagram": None,
     }
+    logger.info("Estado de arcos reseteado.")
 
 ###########################################################################
 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 que se
-    renderice con `st.write(..., unsafe_allow_html=True)`.
+    No imprime directamente en pantalla; regresa el HTML para 
+    usar con `st.write(..., unsafe_allow_html=True)`.
     """
     try:
         diagram_html = ""
@@ -62,10 +60,10 @@ def display_arc_diagram(doc):
                 options={
                     "distance": 100,
                     "arrow_spacing": 20,
-                    "word_spacing": 30,
+                    "word_spacing": 30
                 }
             )
-            # Ajustar tamaño y posición
+            # Ajustar tamaños
             svg_html = svg_html.replace('height="375"', 'height="200"')
             svg_html = re.sub(
                 r'<svg[^>]*>',
@@ -77,9 +75,10 @@ def display_arc_diagram(doc):
                 lambda m: f'<g transform="translate({m.group(1)},50)"',
                 svg_html
             )
-            # Envolver en contenedor con estilo
+            # Envolver en contenedor
             diagram_html += f'<div class="arc-diagram-container">{svg_html}</div>'
         return diagram_html
+
     except Exception as e:
         logger.error(f"Error en display_arc_diagram: {str(e)}")
         return "<p style='color:red;'>Error generando diagrama</p>"
@@ -90,7 +89,7 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
     Interfaz principal para la visualización de diagramas de arco
     (Texto Base vs Iteraciones).
     """
-    # CSS para layout estable
+    # CSS para layout vertical y estable
     st.markdown("""
         <style>
         .stTextArea textarea {
@@ -113,31 +112,33 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
         </style>
     """, unsafe_allow_html=True)
 
-    # Inicializar el estado si no existe
+    # 1) Inicializar estados
     initialize_arc_analysis_state()
-    arc_state = st.session_state.arc_analysis_state  # Para abreviar
+    arc_state = st.session_state.arc_analysis_state
 
-    # Creamos pestañas: "Texto Base" y "Iteraciones"
+    # 2) Creamos pestañas: "Texto Base" y "Iteraciones"
     tabs = st.tabs(["Texto Base", "Iteraciones"])
 
-    # ------------------- PESTAÑA 1: Texto Base --------------------------
+    # =================== PESTAÑA 1: Texto Base ==========================
     with tabs[0]:
         st.subheader("Análisis de Texto Base")
 
-        # Botón para resetear todo
+        # Botón para iniciar nuevo análisis
         if st.button("Nuevo Análisis", key="btn_reset_base"):
+            # Solo limpiamos el estado; si requieres forzar reload,
+            # descomenta la siguiente línea:
+            # st.experimental_rerun()
             reset_arc_analysis_state()
-            st.experimental_rerun()
 
-        # Input para texto base
+        # Textarea de texto base
         arc_state["base_text"] = st.text_area(
             "Ingrese su texto inicial",
             value=arc_state["base_text"],
             key="base_text_input",
-            height=120
+            height=150
         )
 
-        # Botón de análisis base
+        # Botón para analizar texto base
         if st.button("Analizar Texto Base", key="btn_analyze_base"):
             if not arc_state["base_text"].strip():
                 st.warning("Ingrese un texto para analizar.")
@@ -146,10 +147,10 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
                     # Procesar con spaCy
                     doc = nlp_models[lang_code](arc_state["base_text"])
                     # Generar HTML del arco
-                    base_arc_html = display_arc_diagram(doc)
-                    arc_state["base_diagram"] = base_arc_html
+                    arc_html = display_arc_diagram(doc)
+                    arc_state["base_diagram"] = arc_html
 
-                    # Guardar en BD usando tu análisis avanzado
+                    # Guardar en Mongo
                     analysis = perform_advanced_morphosyntactic_analysis(
                         arc_state["base_text"],
                         nlp_models[lang_code]
@@ -161,45 +162,43 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
                     )
                     if base_id:
                         arc_state["base_id"] = base_id
-                        st.success(f"Análisis base guardado con ID: {base_id}")
+                        st.success(f"Análisis base guardado. ID: {base_id}")
 
-                except Exception as e:
+                except Exception as exc:
                     st.error("Error procesando texto base")
-                    logger.error(f"Error en análisis base: {str(e)}")
+                    logger.error(f"Error en análisis base: {str(exc)}")
 
-        # Mostrar el arco base, si existe
+        # Mostrar diagrama base
         if arc_state["base_diagram"]:
             st.markdown("<hr class='divider'>", unsafe_allow_html=True)
             st.markdown("#### Diagrama de Arco (Texto Base)")
             st.write(arc_state["base_diagram"], unsafe_allow_html=True)
 
-    # ------------------- PESTAÑA 2: Iteraciones -------------------------
+    # ================== PESTAÑA 2: Iteraciones ==========================
     with tabs[1]:
         st.subheader("Análisis de Cambios / Iteraciones")
 
-        # Si no se ha analizado nada en la primera pestaña
+        # Verificar que exista texto base analizado
         if not arc_state["base_id"]:
             st.info("Primero analiza un texto base en la pestaña anterior.")
             return
 
-        # Mostrar el texto base como referencia, en modo solo lectura
+        # Mostrar texto base como referencia (solo lectura)
         st.text_area(
             "Texto Base (solo lectura)",
             value=arc_state["base_text"],
-            key="base_text_ref",
             height=80,
             disabled=True
         )
 
-        # Texto de iteración
+        # Caja de texto para la iteración
         arc_state["iteration_text"] = st.text_area(
             "Texto de Iteración",
             value=arc_state["iteration_text"],
-            key="iteration_text_input",
-            height=120
+            height=150
         )
 
-        # Botón para analizar iteración
+        # Botón 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.")
@@ -207,11 +206,11 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
                 try:
                     # Procesar con spaCy
                     doc_iter = nlp_models[lang_code](arc_state["iteration_text"])
-                    iteration_arc_html = display_arc_diagram(doc_iter)
-                    arc_state["iteration_diagram"] = iteration_arc_html
+                    arc_html_iter = display_arc_diagram(doc_iter)
+                    arc_state["iteration_diagram"] = arc_html_iter
 
-                    # Guardar en BD
-                    iteration_analysis = perform_advanced_morphosyntactic_analysis(
+                    # Guardar en Mongo
+                    analysis_iter = perform_advanced_morphosyntactic_analysis(
                         arc_state["iteration_text"],
                         nlp_models[lang_code]
                     )
@@ -220,29 +219,29 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
                         base_id=arc_state["base_id"],
                         original_text=arc_state["base_text"],
                         iteration_text=arc_state["iteration_text"],
-                        arc_diagrams=iteration_analysis["arc_diagrams"]
+                        arc_diagrams=analysis_iter["arc_diagrams"]
                     )
                     if iteration_id:
-                        st.success(f"Iteración guardada con ID: {iteration_id}")
+                        st.success(f"Iteración guardada. ID: {iteration_id}")
 
-                except Exception as e:
+                except Exception as exc:
                     st.error("Error procesando iteración")
-                    logger.error(f"Error en iteración: {str(e)}")
+                    logger.error(f"Error en iteración: {str(exc)}")
 
-        # Mostrar el arco de iteración, si existe
+        # Mostrar diagrama de iteración
         if arc_state["iteration_diagram"]:
             st.markdown("<hr class='divider'>", unsafe_allow_html=True)
             st.markdown("#### Diagrama de Arco (Iteración)")
             st.write(arc_state["iteration_diagram"], unsafe_allow_html=True)
 
-        # Mostrar comparación (opcional)
+        # Comparación vertical (uno abajo del otro)
         if arc_state["base_diagram"] and arc_state["iteration_diagram"]:
             st.markdown("<hr class='divider'>", unsafe_allow_html=True)
-            st.markdown("### Comparación de Diagrama Base vs. Iteración")
-            col_base, col_iter = st.columns(2)
-            with col_base:
-                st.markdown("**Diagrama Base**")
-                st.write(arc_state["base_diagram"], unsafe_allow_html=True)
-            with col_iter:
-                st.markdown("**Diagrama Iterado**")
-                st.write(arc_state["iteration_diagram"], unsafe_allow_html=True)
+            st.markdown("### Comparación Vertical: Base vs. Iteración")
+            
+            st.markdown("**Diagrama Base**")
+            st.write(arc_state["base_diagram"], unsafe_allow_html=True)
+            
+            st.markdown("---")
+            st.markdown("**Diagrama Iterado**")
+            st.write(arc_state["iteration_diagram"], unsafe_allow_html=True)