Update modules/morphosyntax/morphosyntax_interface.py

modules/morphosyntax/morphosyntax_interface.py

@@ -94,6 +94,7 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
 
         # Contenedor para visualizaciones
         diagram_container = st.container()
+        iteration_container = st.container()
         
         # Procesar texto original
         if analyze_button and text_input.strip():
@@ -108,7 +109,8 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
                     'original_text': text_input,
                     'original_analysis': {'doc': doc, 'analysis': analysis},
                     'iteration_text': text_input,
-                    'show_original': True
+                    'show_original': True,
+                    'analysis_count': st.session_state.morpho_state['analysis_count'] + 1
                 })
 
                 if store_student_morphosyntax_result(
@@ -123,10 +125,12 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
                             lang_code,
                             morpho_t
                         )
+            except Exception as e:
+                st.error("Error al procesar texto original")
+                logger.error(f"Error: {str(e)}")
 
-        # Contenedor para iteración
-        if st.session_state.morpho_state['show_original']:
-            iteration_container = st.container()
+        # Mostrar sección de iteración si hay análisis original
+        if st.session_state.morpho_state.get('show_original', False):
             with iteration_container:
                 st.markdown("---")
                 iteration_key = f"iteration_{st.session_state.morpho_state['analysis_count']}"
@@ -164,7 +168,6 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
                             text=iteration_text,
                             arc_diagrams=analysis_iter['arc_diagrams']
                         ):
-                            # Redibujar ambos diagramas
                             with diagram_container:
                                 st.empty()  # Limpiar contenedor
                                 
@@ -194,8 +197,6 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
         st.error("Error general en la interfaz")
         logger.error(f"Error: {str(e)}")
 
-
-
 ############################################################################
 
 def display_morphosyntax_results(result, lang_code, morpho_t):