Update modules/morphosyntax/morphosyntax_interface.py

modules/morphosyntax/morphosyntax_interface.py

@@ -81,16 +81,27 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
             </style>
         """, unsafe_allow_html=True)
 
-        # Inicialización del estado con valores por defecto
+        # Inicialización más robusta del estado
         if 'morphosyntax_state' not in st.session_state:
             st.session_state.morphosyntax_state = {
                 'original_text': '',
                 'current_text': '',
-                'iterations': [],
+                'original_analysis': None,
                 'analysis_count': 0,
+                'iterations': []  # Inicialización explícita de iterations como lista vacía
+            }
+        else:
+            # Asegurar que todas las claves existan
+            required_keys = {
+                'original_text': '',
+                'current_text': '',
                 'original_analysis': None,
-                'editing_original': False
+                'analysis_count': 0,
+                'iterations': []
             }
+            for key, default_value in required_keys.items():
+                if key not in st.session_state.morphosyntax_state:
+                    st.session_state.morphosyntax_state[key] = default_value
 
         with st.container():
             # Sección de texto original
@@ -99,11 +110,11 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
             # Input para texto original
             original_text = st.text_area(
                 "Ingrese una oración", 
-                value=st.session_state.morphosyntax_state.get('original_text', ''),
+                value=st.session_state.morphosyntax_state['original_text'],
                 key="original_text_input",
                 placeholder="Ingresar solo una oración hasta el punto y aparte. Si es punto seguido, dejar así.",
                 height=100,
-                disabled=False  # Siempre editable
+                disabled=False
             )
 
             # Botón para analizar texto original
@@ -116,7 +127,7 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
                     disabled=not bool(original_text.strip())
                 )
 
-            # Procesar texto original si se presiona el botón
+            # Procesar texto original
             if analyze_original and original_text.strip():
                 try:
                     with st.spinner("Procesando texto original..."):
@@ -126,14 +137,15 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
                             nlp_models[lang_code]
                         )
                         
-                        # Actualizar estado
+                        # Actualizar estado de forma segura
                         st.session_state.morphosyntax_state.update({
                             'original_text': original_text,
+                            'current_text': original_text,
                             'original_analysis': {
                                 'doc': doc,
                                 'advanced_analysis': analysis
                             },
-                            'current_text': original_text
+                            'iterations': []  # Reiniciar iteraciones al cambiar texto original
                         })
                         
                         # Guardar en base de datos
@@ -149,8 +161,8 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
                     logger.error(f"Error procesando texto original: {str(e)}")
                     st.error("Error al procesar el texto original")
 
-            # Mostrar diagrama de arco del texto original si existe
-            if st.session_state.morphosyntax_state.get('original_analysis'):
+            # Mostrar diagrama original
+            if st.session_state.morphosyntax_state['original_analysis']:
                 display_morphosyntax_results(
                     st.session_state.morphosyntax_state['original_analysis'],
                     lang_code,
@@ -164,7 +176,7 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
             # Campo para nueva versión
             iteration_text = st.text_area(
                 "Modifique la oración",
-                value=st.session_state.morphosyntax_state.get('current_text', ''),
+                value=st.session_state.morphosyntax_state['current_text'],
                 key=f"iteration_input_{st.session_state.morphosyntax_state['analysis_count']}",
                 placeholder="Ingresar solo una oración hasta el punto y aparte. Si es punto seguido, dejar así.",
                 height=100
@@ -197,15 +209,21 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
                             'advanced_analysis': analysis
                         }
                         
-                        # Guardar iteración
-                        st.session_state.morphosyntax_state['iterations'].append({
+                        # Crear nueva iteración
+                        new_iteration = {
                             'text': iteration_text,
                             'analysis': current_analysis,
                             'timestamp': pd.Timestamp.now()
-                        })
+                        }
                         
-                        st.session_state.morphosyntax_state['current_text'] = iteration_text
-                        st.session_state.morphosyntax_state['analysis_count'] += 1
+                        # Actualizar estado de forma segura
+                        iterations = st.session_state.morphosyntax_state.get('iterations', [])
+                        iterations.append(new_iteration)
+                        st.session_state.morphosyntax_state.update({
+                            'current_text': iteration_text,
+                            'analysis_count': st.session_state.morphosyntax_state['analysis_count'] + 1,
+                            'iterations': iterations
+                        })
                         
                         if store_student_morphosyntax_result(
                             username=st.session_state.username,
@@ -225,7 +243,7 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
                     st.error("Error al procesar los cambios")
 
             # Mostrar historial de iteraciones
-            if st.session_state.morphosyntax_state['iterations']:
+            if st.session_state.morphosyntax_state.get('iterations', []):
                 with st.expander("Historial de Iteraciones", expanded=False):
                     for idx, iteration in enumerate(reversed(st.session_state.morphosyntax_state['iterations'])):
                         st.markdown(f"**Iteración {idx + 1} ({iteration['timestamp'].strftime('%H:%M:%S')})**")
@@ -251,7 +269,6 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
 
 
 
-
 #########################################################################3
 def display_morphosyntax_results(result, lang_code, morpho_t):
     """