Update modules/morphosyntax/morphosyntax_interface.py

modules/morphosyntax/morphosyntax_interface.py

@@ -99,7 +99,7 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
                     nlp_models[lang_code]
                 )
                 
-                # Actualizar estado
+                # Actualizar estado original
                 st.session_state.morpho_state.update({
                     'original_text': text_input,
                     'original_analysis': {'doc': doc, 'analysis': analysis},
@@ -107,13 +107,13 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
                     'analysis_count': st.session_state.morpho_state['analysis_count'] + 1
                 })
 
-                # Guardar en base de datos
                 if store_student_morphosyntax_result(
                     username=st.session_state.username,
                     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("---")
@@ -139,25 +139,40 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
                     # Procesar iteración
                     if iterate_button and iteration_text.strip():
                         try:
-                            doc = nlp_models[lang_code](iteration_text)
-                            analysis = perform_advanced_morphosyntactic_analysis(
+                            doc_iter = nlp_models[lang_code](iteration_text)
+                            analysis_iter = perform_advanced_morphosyntactic_analysis(
                                 iteration_text,
                                 nlp_models[lang_code]
                             )
 
-                            # Actualizar estado
+                            # Actualizar estado de iteración
                             st.session_state.morpho_state.update({
                                 'iteration_text': iteration_text,
-                                'iteration_analysis': {'doc': doc, 'analysis': analysis}
+                                'iteration_analysis': {'doc': doc_iter, 'analysis': analysis_iter}
                             })
 
-                            # Guardar y mostrar
+                            # Guardar iteración
                             if store_student_morphosyntax_result(
                                 username=st.session_state.username,
                                 text=iteration_text,
-                                arc_diagrams=analysis['arc_diagrams']
+                                arc_diagrams=analysis_iter['arc_diagrams']
                             ):
-                                display_morphosyntax_results({'doc': doc, 'analysis': analysis}, lang_code, morpho_t)
+                                # Mostrar ambos diagramas
+                                st.subheader("Análisis Original")
+                                display_morphosyntax_results(
+                                    st.session_state.morpho_state['original_analysis'], 
+                                    lang_code, 
+                                    morpho_t
+                                )
+                                
+                                st.markdown("---")
+                                
+                                st.subheader("Análisis de Cambios")
+                                display_morphosyntax_results(
+                                    {'doc': doc_iter, 'analysis': analysis_iter}, 
+                                    lang_code, 
+                                    morpho_t
+                                )
 
                         except Exception as e:
                             st.error("Error al procesar iteración")
@@ -171,7 +186,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):
     """
     Muestra solo el diagrama de arco.