Update modules/morphosyntax/morphosyntax_interface.py

modules/morphosyntax/morphosyntax_interface.py

@@ -112,7 +112,9 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
             </style>
         """, unsafe_allow_html=True)
 
-        # Inicializar estado
+        # Inicializar estado y mantener tab activo
+        st.session_state.tab_states['morpho_active'] = True
+        st.session_state.selected_tab = 1
         initialize_arc_analysis_state()
 
         # Crear subtabs
@@ -188,100 +190,85 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
                     st.error("Error al procesar texto original")
                     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("---")
-                
-                # Usar un formulario con callback
-                def on_iteration_submit():
-                    st.session_state.iteration_submitted = True
-                
-                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.get('iteration_text', ''),
-                        key=iteration_key,
-                        height=100
-                    )
-                    
-                    # Guardar en estado temporal
-                    if iteration_text:
-                        st.session_state.temp_iteration_text = iteration_text
+            # Sección de iteración
+            if st.session_state.arc_analysis_state.get('original_analysis'):
+                with iteration_input_container:
+                    st.markdown("---")
                     
-                    iterate_button = st.form_submit_button(
-                        "Analizar Cambios",
-                        type="primary",
-                        use_container_width=True,
-                        on_click=on_iteration_submit
-                    )
-
-                # Procesar iteración cuando se envía el formulario
-                if hasattr(st.session_state, 'iteration_submitted') and st.session_state.iteration_submitted:
-                    iteration_text = getattr(st.session_state, 'temp_iteration_text', '')
-                    
-                    if iteration_text.strip():
-                        try:
-                            # Verificar si ya existe en caché
-                            cache_key = f"iter_{iteration_text}"
-                            cached_result = get_cached_analysis(cache_key)
-                            
-                            if cached_result:
-                                logger.info("Usando resultado cacheado")
-                                doc_iter = cached_result['doc']
-                                analysis_iter = cached_result['analysis']
-                            else:
-                                logger.info("Generando nuevo análisis")
-                                doc_iter = nlp_models[lang_code](iteration_text)
-                                analysis_iter = perform_advanced_morphosyntactic_analysis(
-                                    iteration_text,
-                                    nlp_models[lang_code]
-                                )
-                                # Guardar en caché
-                                cache_analysis_results(cache_key, {
-                                    'doc': doc_iter,
-                                    'analysis': analysis_iter
-                                })
-
-                            # Actualizar estado
-                            st.session_state.arc_analysis_state.update({
-                                'iteration_text': iteration_text,
-                                'iteration_analysis': {'doc': doc_iter, 'analysis': analysis_iter}
-                            })
-
-                            if store_student_morphosyntax_result(
-                                username=st.session_state.username,
-                                text=iteration_text,
-                                arc_diagrams=analysis_iter['arc_diagrams']
-                            ):
-                                # Mostrar ambos análisis
-                                col1, col2 = st.columns(2)
-                                
-                                with col1:
-                                    display_original_analysis(
-                                        original_diagram_container,
-                                        st.session_state.arc_analysis_state['original_analysis'],
-                                        lang_code,
-                                        morpho_t
-                                    )
+                    # Formulario de iteración
+                    with st.form(key="iteration_form"):
+                        iteration_text = st.text_area(
+                            "",
+                            value=st.session_state.arc_analysis_state.get('iteration_text', ''),
+                            key=f"iteration_{st.session_state.arc_analysis_state['analysis_count']}",
+                            height=100
+                        )
+                        
+                        # Botón de análisis dentro del formulario
+                        submitted = st.form_submit_button(
+                            "Analizar Cambios",
+                            type="primary",
+                            use_container_width=True
+                        )
+                        
+                        # Cuando se envía el formulario
+                        if submitted and iteration_text.strip():
+                            try:
+                                # Verificar si ya existe en caché
+                                cache_key = f"iter_{iteration_text}"
+                                cached_result = get_cached_analysis(cache_key)
                                 
-                                with col2:
-                                    display_iteration_analysis(
-                                        iteration_diagram_container,
-                                        {'doc': doc_iter, 'analysis': analysis_iter},
-                                        lang_code,
-                                        morpho_t
+                                if cached_result:
+                                    logger.info("Usando resultado cacheado")
+                                    doc_iter = cached_result['doc']
+                                    analysis_iter = cached_result['analysis']
+                                else:
+                                    logger.info("Generando nuevo análisis")
+                                    doc_iter = nlp_models[lang_code](iteration_text)
+                                    analysis_iter = perform_advanced_morphosyntactic_analysis(
+                                        iteration_text,
+                                        nlp_models[lang_code]
                                     )
+                                    # Guardar en caché
+                                    cache_analysis_results(cache_key, {
+                                        'doc': doc_iter,
+                                        'analysis': analysis_iter
+                                    })
+
+                                # Actualizar estado
+                                st.session_state.arc_analysis_state.update({
+                                    'iteration_text': iteration_text,
+                                    'iteration_analysis': {'doc': doc_iter, 'analysis': analysis_iter}
+                                })
 
-                            # Limpiar flag de envío
-                            st.session_state.iteration_submitted = False
-                            
-                        except Exception as e:
-                            st.error("Error al procesar iteración")
-                            logger.error(f"Error al procesar iteración: {str(e)}")
-                            # Limpiar flag de envío en caso de error
-                            st.session_state.iteration_submitted = False
+                                if store_student_morphosyntax_result(
+                                    username=st.session_state.username,
+                                    text=iteration_text,
+                                    arc_diagrams=analysis_iter['arc_diagrams']
+                                ):
+                                    # Mostrar análisis en columnas
+                                    col1, col2 = st.columns(2)
+                                    
+                                    # Mostrar ambos análisis
+                                    with col1:
+                                        display_original_analysis(
+                                            original_diagram_container,
+                                            st.session_state.arc_analysis_state['original_analysis'],
+                                            lang_code,
+                                            morpho_t
+                                        )
+                                    
+                                    with col2:
+                                        display_iteration_analysis(
+                                            iteration_diagram_container,
+                                            {'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 al procesar iteración: {str(e)}")
 
         # Otros subtabs...
         with subtabs[1]: