Update modules/morphosyntax/morphosyntax_interface.py

modules/morphosyntax/morphosyntax_interface.py

@@ -62,34 +62,39 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
             </style>
         """, unsafe_allow_html=True)
 
-        # Estado consistente con keys únicas
+        # Estado consistente
         if 'morpho_state' not in st.session_state:
             st.session_state.morpho_state = {
                 'original_text': '',
                 'original_analysis': None,
                 'iteration_text': '',
                 'iteration_analysis': None,
-                'analysis_count': 0
+                'analysis_count': 0,
+                'show_original': False,
+                'show_iteration': False
             }
 
-        # Texto original - Key única basada en timestamp
-        text_key = f"original_text_{st.session_state.morpho_state['analysis_count']}"
-        text_input = st.text_area(
-            "",
-            value=st.session_state.morpho_state['original_text'],
-            key=text_key,
-            height=100
-        )
-
-        # Botón analizar original
-        col1, col2, col3 = st.columns([2,1,2])
-        with col1:
-            analyze_button = st.button(
-                "Analizar Texto Original",
-                type="primary",
-                use_container_width=True
+        # Contenedor para entrada original
+        with st.container():
+            text_key = f"original_text_{st.session_state.morpho_state['analysis_count']}"
+            text_input = st.text_area(
+                "",
+                value=st.session_state.morpho_state['original_text'],
+                key=text_key,
+                height=100
             )
 
+            col1, col2, col3 = st.columns([2,1,2])
+            with col1:
+                analyze_button = st.button(
+                    "Analizar Texto Original",
+                    type="primary",
+                    use_container_width=True
+                )
+
+        # Contenedor para visualizaciones
+        diagram_container = st.container()
+        
         # Procesar texto original
         if analyze_button and text_input.strip():
             try:
@@ -99,12 +104,11 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
                     nlp_models[lang_code]
                 )
                 
-                # Actualizar estado original
                 st.session_state.morpho_state.update({
                     'original_text': text_input,
                     'original_analysis': {'doc': doc, 'analysis': analysis},
                     'iteration_text': text_input,
-                    'analysis_count': st.session_state.morpho_state['analysis_count'] + 1
+                    'show_original': True
                 })
 
                 if store_student_morphosyntax_result(
@@ -112,80 +116,88 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
                     text=text_input,
                     arc_diagrams=analysis['arc_diagrams']
                 ):
-                    # Mostrar diagrama original
-                    st.subheader("Análisis Original")
-                    display_morphosyntax_results({'doc': doc, 'analysis': analysis}, lang_code, morpho_t)
-
-                    st.markdown("---")
-
-                    # Campo para iteración
-                    iteration_key = f"iteration_{st.session_state.morpho_state['analysis_count']}"
-                    iteration_text = st.text_area(
-                        "",
-                        value=st.session_state.morpho_state['iteration_text'],
-                        key=iteration_key,
-                        height=100
+                    with diagram_container:
+                        st.subheader("Análisis Original")
+                        display_morphosyntax_results(
+                            st.session_state.morpho_state['original_analysis'],
+                            lang_code,
+                            morpho_t
+                        )
+
+        # Contenedor para iteración
+        if st.session_state.morpho_state['show_original']:
+            iteration_container = st.container()
+            with iteration_container:
+                st.markdown("---")
+                iteration_key = f"iteration_{st.session_state.morpho_state['analysis_count']}"
+                iteration_text = st.text_area(
+                    "",
+                    value=st.session_state.morpho_state['iteration_text'],
+                    key=iteration_key,
+                    height=100
+                )
+
+                col1, col2, col3 = st.columns([2,1,2])
+                with col1:
+                    iterate_button = st.button(
+                        "Analizar Cambios",
+                        type="primary", 
+                        use_container_width=True
                     )
 
-                    # Botón analizar iteración  
-                    col1, col2, col3 = st.columns([2,1,2])
-                    with col1:
-                        iterate_button = st.button(
-                            "Analizar Cambios",
-                            type="primary", 
-                            use_container_width=True
+                if iterate_button and iteration_text.strip():
+                    try:
+                        doc_iter = nlp_models[lang_code](iteration_text)
+                        analysis_iter = perform_advanced_morphosyntactic_analysis(
+                            iteration_text,
+                            nlp_models[lang_code]
                         )
 
-                    # Procesar iteración
-                    if iterate_button and iteration_text.strip():
-                        try:
-                            doc_iter = nlp_models[lang_code](iteration_text)
-                            analysis_iter = perform_advanced_morphosyntactic_analysis(
-                                iteration_text,
-                                nlp_models[lang_code]
-                            )
-
-                            # Actualizar estado de iteración
-                            st.session_state.morpho_state.update({
-                                'iteration_text': iteration_text,
-                                'iteration_analysis': {'doc': doc_iter, 'analysis': analysis_iter}
-                            })
-
-                            # Guardar iteración
-                            if store_student_morphosyntax_result(
-                                username=st.session_state.username,
-                                text=iteration_text,
-                                arc_diagrams=analysis_iter['arc_diagrams']
-                            ):
-                                # Mostrar ambos diagramas
+                        st.session_state.morpho_state.update({
+                            'iteration_text': iteration_text,
+                            'iteration_analysis': {'doc': doc_iter, 'analysis': analysis_iter},
+                            'show_iteration': True
+                        })
+
+                        if store_student_morphosyntax_result(
+                            username=st.session_state.username,
+                            text=iteration_text,
+                            arc_diagrams=analysis_iter['arc_diagrams']
+                        ):
+                            # Redibujar ambos diagramas
+                            with diagram_container:
+                                st.empty()  # Limpiar contenedor
+                                
+                                # Mostrar original
                                 st.subheader("Análisis Original")
                                 display_morphosyntax_results(
-                                    st.session_state.morpho_state['original_analysis'], 
-                                    lang_code, 
+                                    st.session_state.morpho_state['original_analysis'],
+                                    lang_code,
                                     morpho_t
                                 )
                                 
                                 st.markdown("---")
                                 
+                                # Mostrar iteración
                                 st.subheader("Análisis de Cambios")
                                 display_morphosyntax_results(
-                                    {'doc': doc_iter, 'analysis': analysis_iter}, 
-                                    lang_code, 
+                                    {'doc': doc_iter, 'analysis': analysis_iter},
+                                    lang_code,
                                     morpho_t
                                 )
 
-                        except Exception as e:
-                            st.error("Error al procesar iteración")
-                            logger.error(f"Error: {str(e)}")
-
-            except Exception as e:
-                st.error("Error al procesar texto original")
-                logger.error(f"Error: {str(e)}")
+                    except Exception as e:
+                        st.error("Error al procesar iteración")
+                        logger.error(f"Error: {str(e)}")
 
     except Exception as e:
         st.error("Error general en la interfaz")
         logger.error(f"Error: {str(e)}")
 
+
+
+############################################################################
+
 def display_morphosyntax_results(result, lang_code, morpho_t):
     """
     Muestra solo el diagrama de arco.