Update modules/morphosyntax/morphosyntax_interface.py

modules/morphosyntax/morphosyntax_interface.py

@@ -41,6 +41,8 @@ import pandas as pd
 import base64
 import re
 
+############################################################################
+
 def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
     try:
         # CSS para mejorar la estabilidad y diseño
@@ -76,22 +78,30 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
             </style>
         """, unsafe_allow_html=True)
 
-        # Inicializar el estado
+        # Verificar si el estado existe y tiene todas las claves necesarias
         if 'morphosyntax_state' not in st.session_state:
-            st.session_state.morphosyntax_state = {
-                'has_original': False,
-                'original_text': '',
-                'original_analysis': None,
-                'current_text': '',
-                'iterations': [],
-                'analysis_count': 0
-            }
+            st.session_state['morphosyntax_state'] = {}
+        
+        # Asegurarse de que todas las claves existan con valores por defecto
+        default_state = {
+            'has_original': False,
+            'original_text': '',
+            'original_analysis': None,
+            'current_text': '',
+            'iterations': [],
+            'analysis_count': 0
+        }
+        
+        # Actualizar solo las claves faltantes
+        for key, value in default_state.items():
+            if key not in st.session_state.morphosyntax_state:
+                st.session_state.morphosyntax_state[key] = value
 
-        # Texto original (si aún no se ha ingresado)
+        # Mostrar interfaz inicial o de iteración según el estado
         if not st.session_state.morphosyntax_state['has_original']:
-            st.markdown("### Original Text")
+            st.markdown("### Texto Original")
             original_text = st.text_area(
-                "Enter your initial text",
+                "Ingrese su texto inicial",
                 key="original_text_input",
                 height=150
             )
@@ -99,67 +109,72 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
             col1, col2, col3 = st.columns([2,1,2])
             with col1:
                 if st.button(
-                    "Set as Original Text", 
+                    "Establecer como Texto Original", 
                     type="primary",
                     use_container_width=True,
                     disabled=not bool(original_text.strip())
                 ):
-                    # Procesar texto original
-                    doc = nlp_models[lang_code](original_text)
-                    analysis = perform_advanced_morphosyntactic_analysis(
-                        original_text,
-                        nlp_models[lang_code]
-                    )
-                    
-                    # Guardar estado original
-                    st.session_state.morphosyntax_state.update({
-                        'has_original': True,
-                        'original_text': original_text,
-                        'original_analysis': {
-                            'doc': doc,
-                            'advanced_analysis': analysis
-                        }
-                    })
+                    try:
+                        # Procesar texto original
+                        doc = nlp_models[lang_code](original_text)
+                        analysis = perform_advanced_morphosyntactic_analysis(
+                            original_text,
+                            nlp_models[lang_code]
+                        )
+                        
+                        # Guardar estado original
+                        st.session_state.morphosyntax_state.update({
+                            'has_original': True,
+                            'original_text': original_text,
+                            'original_analysis': {
+                                'doc': doc,
+                                'advanced_analysis': analysis
+                            },
+                            'current_text': original_text
+                        })
+                        
+                        # Guardar en base de datos
+                        if store_student_morphosyntax_result(
+                            username=st.session_state.username,
+                            text=original_text,
+                            arc_diagrams=analysis['arc_diagrams']
+                        ):
+                            st.success("Texto original guardado exitosamente")
+                            st.rerun()
+                        else:
+                            st.error("Error al guardar el texto original")
                     
-                    # Guardar en base de datos
-                    store_student_morphosyntax_result(
-                        username=st.session_state.username,
-                        text=original_text,
-                        arc_diagrams=analysis['arc_diagrams']
-                    )
-                    st.rerun()
+                    except Exception as e:
+                        logger.error(f"Error procesando texto original: {str(e)}")
+                        st.error("Error al procesar el texto original")
 
-        # Mostrar contenido original y área de iteración
         else:
-            # Sección Original
-            st.markdown("### Original Version")
-            with st.container():
-                col1, col2 = st.columns([1,1])
-                
-                with col1:
-                    st.markdown("#### Original Text")
-                    st.markdown(f"<div class='original-content'>{st.session_state.morphosyntax_state['original_text']}</div>", 
-                              unsafe_allow_html=True)
-                
-                with col2:
-                    st.markdown("#### Original Analysis")
-                    display_morphosyntax_results(
-                        st.session_state.morphosyntax_state['original_analysis'],
-                        lang_code,
-                        morpho_t
-                    )
+            # Mostrar versión original y área de iteración
+            st.markdown("### Versión Original")
+            
+            # Columnas para texto original y su análisis
+            col1, col2 = st.columns([1,1])
+            with col1:
+                st.markdown("#### Texto Original")
+                st.markdown(f"<div class='original-content'>{st.session_state.morphosyntax_state['original_text']}</div>", 
+                          unsafe_allow_html=True)
+            
+            with col2:
+                st.markdown("#### Análisis Original")
+                display_morphosyntax_results(
+                    st.session_state.morphosyntax_state['original_analysis'],
+                    lang_code,
+                    morpho_t
+                )
 
-            # Separador
+            # Área de iteración
             st.markdown("---")
-
-            # Sección de Iteración
-            st.markdown("### Current Iteration")
+            st.markdown("### Iteración Actual")
             
-            # Campo para la nueva versión
+            # Campo para nueva versión
             iteration_text = st.text_area(
-                "Modify your text",
-                value=st.session_state.morphosyntax_state.get('current_text', 
-                                                            st.session_state.morphosyntax_state['original_text']),
+                "Modifique su texto",
+                value=st.session_state.morphosyntax_state['current_text'],
                 key=f"iteration_input_{st.session_state.morphosyntax_state['analysis_count']}",
                 height=150
             )
@@ -168,7 +183,7 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
             col1, col2, col3 = st.columns([2,1,2])
             with col1:
                 analyze_button = st.button(
-                    "Analyze Changes",
+                    "Analizar Cambios",
                     type="primary",
                     icon="🔍",
                     key=f"analyze_{st.session_state.morphosyntax_state['analysis_count']}",
@@ -176,52 +191,58 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
                     use_container_width=True
                 )
 
-            # Procesar nueva iteración
+            # Procesar iteración
             if analyze_button and iteration_text.strip():
-                with st.spinner("Processing changes..."):
-                    # Realizar análisis
-                    doc = nlp_models[lang_code](iteration_text)
-                    analysis = perform_advanced_morphosyntactic_analysis(
-                        iteration_text,
-                        nlp_models[lang_code]
-                    )
-                    
-                    # Actualizar estado
-                    current_analysis = {
-                        'doc': doc,
-                        'advanced_analysis': analysis
-                    }
-                    
-                    # Guardar iteración
-                    st.session_state.morphosyntax_state['iterations'].append({
-                        'text': iteration_text,
-                        'analysis': current_analysis,
-                        'timestamp': pd.Timestamp.now()
-                    })
-                    
-                    st.session_state.morphosyntax_state['current_text'] = iteration_text
-                    st.session_state.morphosyntax_state['analysis_count'] += 1
-                    
-                    # Guardar en base de datos
-                    store_student_morphosyntax_result(
-                        username=st.session_state.username,
-                        text=iteration_text,
-                        arc_diagrams=analysis['arc_diagrams']
-                    )
-                    
-                    # Mostrar análisis actual
-                    st.markdown("#### Current Analysis")
-                    display_morphosyntax_results(
-                        current_analysis,
-                        lang_code,
-                        morpho_t
-                    )
+                try:
+                    with st.spinner("Procesando cambios..."):
+                        doc = nlp_models[lang_code](iteration_text)
+                        analysis = perform_advanced_morphosyntactic_analysis(
+                            iteration_text,
+                            nlp_models[lang_code]
+                        )
+                        
+                        current_analysis = {
+                            'doc': doc,
+                            'advanced_analysis': analysis
+                        }
+                        
+                        # Guardar iteración
+                        st.session_state.morphosyntax_state['iterations'].append({
+                            'text': iteration_text,
+                            'analysis': current_analysis,
+                            'timestamp': pd.Timestamp.now()
+                        })
+                        
+                        st.session_state.morphosyntax_state['current_text'] = iteration_text
+                        st.session_state.morphosyntax_state['analysis_count'] += 1
+                        
+                        # Guardar en base de datos
+                        if store_student_morphosyntax_result(
+                            username=st.session_state.username,
+                            text=iteration_text,
+                            arc_diagrams=analysis['arc_diagrams']
+                        ):
+                            st.success("Análisis guardado exitosamente")
+                            
+                            # Mostrar análisis actual
+                            st.markdown("#### Análisis Actual")
+                            display_morphosyntax_results(
+                                current_analysis,
+                                lang_code,
+                                morpho_t
+                            )
+                        else:
+                            st.error("Error al guardar el análisis")
+                
+                except Exception as e:
+                    logger.error(f"Error procesando iteración: {str(e)}")
+                    st.error("Error al procesar los cambios")
 
-            # Mostrar historial de iteraciones
+            # Mostrar historial
             if st.session_state.morphosyntax_state['iterations']:
-                with st.expander("View Iteration History"):
+                with st.expander("Ver Historial de Iteraciones"):
                     for idx, iteration in enumerate(reversed(st.session_state.morphosyntax_state['iterations']), 1):
-                        st.markdown(f"**Iteration {idx} - {iteration['timestamp'].strftime('%H:%M:%S')}**")
+                        st.markdown(f"**Iteración {idx} - {iteration['timestamp'].strftime('%H:%M:%S')}**")
                         st.markdown(f"```\n{iteration['text']}\n```")
                         display_morphosyntax_results(
                             iteration['analysis'],
@@ -234,8 +255,6 @@ def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
         logger.error(f"Error general en display_morphosyntax_interface: {str(e)}")
         st.error("Se produjo un error. Por favor, intente de nuevo.")
 
-# El resto del código permanece igual...
-