Update modules/morphosyntax/morphosyntax_interface.py

modules/morphosyntax/morphosyntax_interface.py

@@ -28,62 +28,84 @@ logger = logging.getLogger(__name__)
 def display_morphosyntax_interface(lang_code, nlp_models, t):
     """
     Interfaz para el análisis morfosintáctico
+    Args:
+        lang_code: Código del idioma actual
+        nlp_models: Modelos de spaCy cargados
+        t: Diccionario de traducciones
     """
+    # Obtener el diccionario de traducciones morfosintácticas
     morpho_t = t.get('MORPHOSYNTACTIC', {})
-    #st.title(morpho_t.get('title', 'AIdeaText - Morphological Analysis'))
-
+    
+    # Inicializar el estado de la entrada
     input_key = f"morphosyntax_input_{lang_code}"
-
     if input_key not in st.session_state:
         st.session_state[input_key] = ""
 
+    # Campo de entrada de texto
     sentence_input = st.text_area(
-        t['morpho_input_label'],
+        morpho_t.get('morpho_input_label', 'Enter text to analyze'),
         height=150,
-        placeholder=t['input_placeholder'],
+        placeholder=morpho_t.get('morpho_input_placeholder', 'Enter your text here...'),
         value=st.session_state[input_key],
         key=f"text_area_{lang_code}",
         on_change=lambda: setattr(st.session_state, input_key, st.session_state[f"text_area_{lang_code}"])
     )
 
-    if st.button(t['analyze_button'], key=f"analyze_button_{lang_code}"):
+    # Botón de análisis
+    if st.button(morpho_t.get('analyze_button', 'Analyze text'), key=f"analyze_button_{lang_code}"):
         current_input = st.session_state[input_key]
         if current_input:
-            doc = nlp_models[lang_code](current_input)
-
-            # Análisis morfosintáctico avanzado
-            advanced_analysis = perform_advanced_morphosyntactic_analysis(current_input, nlp_models[lang_code])    
+            try:
+                # Procesar el texto
+                doc = nlp_models[lang_code](current_input)
+                
+                # Realizar análisis morfosintáctico
+                advanced_analysis = perform_advanced_morphosyntactic_analysis(
+                    current_input, 
+                    nlp_models[lang_code]
+                )    
+
+                # Guardar resultado en el estado de la sesión
+                st.session_state.morphosyntax_result = {
+                    'doc': doc,
+                    'advanced_analysis': advanced_analysis
+                }
 
-            # Guardar el resultado en el estado de la sesión
-            st.session_state.morphosyntax_result = {
-                'doc': doc,
-                'advanced_analysis': advanced_analysis
-            }
+                # Mostrar resultados
+                display_morphosyntax_results(
+                    st.session_state.morphosyntax_result, 
+                    lang_code, 
+                    morpho_t  # Pasar morpho_t en lugar de t
+                )
+
+                # Guardar en la base de datos
+                if store_morphosyntax_result(
+                    st.session_state.username,
+                    current_input,
+                    get_repeated_words_colors(doc),
+                    advanced_analysis['arc_diagram'],
+                    advanced_analysis['pos_analysis'],
+                    advanced_analysis['morphological_analysis'],
+                    advanced_analysis['sentence_structure']
+                ):
+                    st.success(morpho_t.get('success_message', 'Analysis saved successfully'))
+                else:
+                    st.error(morpho_t.get('error_message', 'Error saving analysis'))
 
-            # Mostrar resultados
-            display_morphosyntax_results(st.session_state.morphosyntax_result, lang_code, t)
-
-            # Guardar resultados
-            if store_morphosyntax_result(
-                st.session_state.username,
-                current_input,
-                get_repeated_words_colors(doc),
-                advanced_analysis['arc_diagram'],
-                advanced_analysis['pos_analysis'],
-                advanced_analysis['morphological_analysis'],
-                advanced_analysis['sentence_structure']
-            ):
-                st.success(t['success_message'])
-            else:
-                st.error(t['error_message'])
+            except Exception as e:
+                st.error(morpho_t.get('error_processing', f'Error processing text: {str(e)}'))
         else:
-            st.warning(t['warning_message'])
+            st.warning(morpho_t.get('warning_message', 'Please enter a text to analyze'))
+
+    # Mostrar resultados previos si existen
     elif 'morphosyntax_result' in st.session_state and st.session_state.morphosyntax_result is not None:
-        
-        # Si hay un resultado guardado, mostrarlo
-        display_morphosyntax_results(st.session_state.morphosyntax_result, lang_code, t)
+        display_morphosyntax_results(
+            st.session_state.morphosyntax_result, 
+            lang_code, 
+            morpho_t  # Pasar morpho_t en lugar de t
+        )
     else:
-        st.info(t['initial_message'])  # Añade esta traducción a tu diccionario            
+        st.info(morpho_t.get('morpho_initial_message', 'Enter text to begin analysis'))       
 
 def display_morphosyntax_results(result, lang_code, t):
     if result is None: