Update modules/ui/ui.py

modules/ui/ui.py

@@ -589,15 +589,34 @@ def display_morphosyntax_analysis_interface(nlp_models, lang_code):
             # Mostrar resultados
             display_morphosyntax_results(st.session_state.morphosyntax_result, lang_code, t)
 
-    elif 'morphosyntax_result' in st.session_state:
+            # 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'])
+        else:
+            st.warning(t['warning_message'])
+    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)
     else:
-        st.warning(t['warning_message'])
+        st.info(t['initial_message'])  # Añade esta traducción a tu diccionario            
 
 def display_morphosyntax_results(result, lang_code, t):
+    if result is None:
+        st.warning(t['no_results'])  # Añade esta traducción a tu diccionario
+        return
+
     doc = result['doc']
-    advanced_analysis = result['advanced_analysis']
     
     # Mostrar leyenda (código existente)
     st.markdown(f"##### {t['legend']}")