Update modules/morphosyntax/morphosyntax_interface.py

modules/morphosyntax/morphosyntax_interface.py

@@ -1,5 +1,6 @@
 #modules/morphosyntax/morphosyntax_interface.py
 
+#Importaciones generales
 import streamlit as st
 from streamlit_float import *
 from streamlit_antd_components import *
@@ -11,6 +12,7 @@ import pandas as pd
 import base64
 import re
 
+#Importaciones locales
 from .morphosyntax_process import (
     process_morphosyntactic_input,
     format_analysis_results,
@@ -29,17 +31,29 @@ import logging
 logger = logging.getLogger(__name__)
 
 ###########################################################################
+def reset_morpho_state():
+    """Resetea el estado del análisis morfosintáctico"""
+    if 'arc_analysis_state' in st.session_state:
+        st.session_state.arc_analysis_state = {
+            'original_text': '',
+            'original_analysis': None,
+            'iteration_text': '',
+            'iteration_analysis': None,
+            'analysis_count': 0
+        }
+############################################################################
 
-import streamlit as st
-from streamlit_float import *
-from streamlit_antd_components import *
-from streamlit.components.v1 import html
-import spacy
-from spacy import displacy
-import spacy_streamlit
-import pandas as pd
-import base64
-import re
+def display_original_analysis(container, analysis, lang_code, morpho_t):
+    """Muestra el análisis original en el contenedor especificado"""
+    with container:
+        st.subheader("Análisis Original")
+        display_morphosyntax_results(analysis, lang_code, morpho_t)
+
+def display_iteration_analysis(container, analysis, lang_code, morpho_t):
+    """Muestra el análisis de cambios en el contenedor especificado"""
+    with container:
+        st.subheader("Análisis de Cambios")
+        display_morphosyntax_results(analysis, lang_code, morpho_t)
 
 ############################################################################
 
@@ -75,19 +89,23 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
 
         # Tab de Diagramas de Arco
         with subtabs[0]:
-            # Estado para análisis
-            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
-                }
-
-            # Contenedor para texto original
-            original_container = st.container()
-            with original_container:
+            # Botón de reset
+            col1, col2, col3 = st.columns([2,1,2])
+            with col1:
+                if st.button("Nuevo Análisis", type="secondary", use_container_width=True):
+                    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()
+            iteration_input_container = st.container()
+            iteration_diagram_container = st.container()
+
+            # Sección de análisis original
+            with original_input_container:
                 text_key = f"original_text_{st.session_state.arc_analysis_state['analysis_count']}"
                 text_input = st.text_area(
                     "",
@@ -101,13 +119,9 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
                     analyze_button = st.button(
                         "Analizar Texto Original",
                         type="primary",
-                        use_container_width=True,
                         key=f"analyze_original_{st.session_state.arc_analysis_state['analysis_count']}"
                     )
 
-            # Contenedor para diagramas
-            diagrams_container = st.container()
-            
             # Procesar texto original
             if analyze_button and text_input.strip():
                 try:
@@ -117,6 +131,7 @@ 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},
@@ -129,38 +144,34 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
                         text=text_input,
                         arc_diagrams=analysis['arc_diagrams']
                     ):
-                        with diagrams_container:
-                            st.subheader("Análisis Original")
-                            display_morphosyntax_results(
-                                st.session_state.arc_analysis_state['original_analysis'],
-                                lang_code,
-                                morpho_t
-                            )
+                        display_original_analysis(
+                            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)}")
 
-            # Contenedor para iteración
+            # Sección de iteración
             if st.session_state.arc_analysis_state.get('original_analysis'):
-                iteration_container = st.container()
-                with iteration_container:
+                with iteration_input_container:
                     st.markdown("---")
                     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'],
-                        key=iteration_key,
-                        height=100,
-                        on_change=lambda: None  # Prevenir recarga en cambio de foco
-                    )
-
-                    col1, col2, col3 = st.columns([2,1,2])
-                    with col1:
-                        iterate_button = st.button(
+                    
+                    # Usar un formulario para evitar recargas innecesarias
+                    with st.form(key="iteration_form"):
+                        iteration_text = st.text_area(
+                            "",
+                            value=st.session_state.arc_analysis_state['iteration_text'],
+                            key=iteration_key,
+                            height=100
+                        )
+                        iterate_button = st.form_submit_button(
                             "Analizar Cambios",
-                            type="primary", 
-                            use_container_width=True,
-                            key=f"iterate_{st.session_state.arc_analysis_state['analysis_count']}"
+                            type="primary",
+                            use_container_width=True
                         )
 
                     if iterate_button and iteration_text.strip():
@@ -171,6 +182,7 @@ 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}
@@ -181,40 +193,24 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
                                 text=iteration_text,
                                 arc_diagrams=analysis_iter['arc_diagrams']
                             ):
-                                with diagrams_container:
-                                    # Mostrar original
-                                    st.subheader("Análisis Original")
-                                    display_morphosyntax_results(
-                                        st.session_state.arc_analysis_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,
-                                        morpho_t
-                                    )
+                                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: {str(e)}")
 
-        # Otros subtabs aquí...
-        with subtabs[1]:
-            st.info("Análisis de Categorías en desarrollo...")
-            
-        with subtabs[2]:
-            st.info("Análisis Morfológico en desarrollo...")
+        # Otros subtabs...
 
     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):