Update modules/morphosyntax/morphosyntax_interface.py

modules/morphosyntax/morphosyntax_interface.py

@@ -31,6 +31,20 @@ import logging
 logger = logging.getLogger(__name__)
 
 ###########################################################################
+def initialize_arc_analysis_state():
+    """Inicializa el estado del análisis de arcos si no existe"""
+    if 'arc_analysis_state' not in st.session_state:
+        st.session_state.arc_analysis_state = {
+            'original_text': '',
+            'original_analysis': None,
+            'iteration_text': '',
+            'iteration_analysis': None,
+            'analysis_count': 0
+        }
+        logger.info("Estado de análisis de arcos inicializado")
+
+###########################################################################
+
 def reset_morpho_state():
     """Resetea el estado del análisis morfosintáctico"""
     if 'arc_analysis_state' in st.session_state:
@@ -76,9 +90,8 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
             </style>
         """, unsafe_allow_html=True)
 
-        # Estado para subtabs
-        if 'morpho_subtab' not in st.session_state:
-            st.session_state.morpho_subtab = 0
+        # Inicializar estado
+        initialize_arc_analysis_state()
 
         # Crear subtabs
         subtabs = st.tabs([
@@ -96,8 +109,6 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
                     reset_morpho_state()
                     st.rerun()
 
-            # Estado para análisis...
-
             # Contenedores separados para cada tipo de análisis
             original_input_container = st.container()
             original_diagram_container = st.container()
@@ -119,7 +130,8 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
                     analyze_button = st.button(
                         "Analizar Texto Original",
                         type="primary",
-                        key=f"analyze_original_{st.session_state.arc_analysis_state['analysis_count']}"
+                        key=f"analyze_original_{st.session_state.arc_analysis_state['analysis_count']}",
+                        use_container_width=True
                     )
 
             # Procesar texto original
@@ -131,7 +143,6 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
                         nlp_models[lang_code]
                     )
                     
-                    # Actualizar estado
                     st.session_state.arc_analysis_state.update({
                         'original_text': text_input,
                         'original_analysis': {'doc': doc, 'analysis': analysis},
@@ -145,23 +156,22 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
                         arc_diagrams=analysis['arc_diagrams']
                     ):
                         display_original_analysis(
-                            original_diagram_container, 
+                            original_diagram_container,
                             st.session_state.arc_analysis_state['original_analysis'],
                             lang_code,
                             morpho_t
                         )
+
                 except Exception as e:
                     st.error("Error al procesar texto original")
-                    logger.error(f"Error: {str(e)}")
+                    logger.error(f"Error al procesar texto original: {str(e)}")
 
             # Sección de iteración
             if st.session_state.arc_analysis_state.get('original_analysis'):
                 with iteration_input_container:
                     st.markdown("---")
-                    iteration_key = f"iteration_{st.session_state.arc_analysis_state['analysis_count']}"
-                    
-                    # Usar un formulario para evitar recargas innecesarias
                     with st.form(key="iteration_form"):
+                        iteration_key = f"iteration_{st.session_state.arc_analysis_state['analysis_count']}"
                         iteration_text = st.text_area(
                             "",
                             value=st.session_state.arc_analysis_state['iteration_text'],
@@ -182,7 +192,6 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
                                 nlp_models[lang_code]
                             )
 
-                            # Actualizar solo el análisis de iteración
                             st.session_state.arc_analysis_state.update({
                                 'iteration_text': iteration_text,
                                 'iteration_analysis': {'doc': doc_iter, 'analysis': analysis_iter}
@@ -193,6 +202,14 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
                                 text=iteration_text,
                                 arc_diagrams=analysis_iter['arc_diagrams']
                             ):
+                                # Mostrar análisis original y de cambios
+                                display_original_analysis(
+                                    original_diagram_container,
+                                    st.session_state.arc_analysis_state['original_analysis'],
+                                    lang_code,
+                                    morpho_t
+                                )
+                                
                                 display_iteration_analysis(
                                     iteration_diagram_container,
                                     {'doc': doc_iter, 'analysis': analysis_iter},
@@ -202,14 +219,18 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
 
                         except Exception as e:
                             st.error("Error al procesar iteración")
-                            logger.error(f"Error: {str(e)}")
+                            logger.error(f"Error al procesar iteración: {str(e)}")
 
         # Otros subtabs...
+        with subtabs[1]:
+            st.info("Análisis de Categorías en desarrollo...")
+            
+        with subtabs[2]:
+            st.info("Análisis Morfológico en desarrollo...")
 
     except Exception as e:
         st.error("Error general en la interfaz")
-        logger.error(f"Error: {str(e)}")
-
+        logger.error(f"Error general en la interfaz: {str(e)}")
 
 ############################################################################