Update app.py

app.py

@@ -303,13 +303,17 @@ def main_app():
                         st.pyplot(fig)
                         
                     # Store analysis results
-                    store_analysis_result(
+                    # Guardar el resultado
+                    if store_analysis_result(
                         st.session_state.username,
                         sentence_input,
-                        highlighted_text,
+                        word_colors,
                         arc_diagrams,
                         fig
-                    )
+                    ):
+                        st.success("Análisis guardado correctamente.")
+                    else:
+                        st.error("Hubo un problema al guardar el análisis. Por favor, inténtelo de nuevo.")
 
         elif st.session_state.role == "Profesor":
             # Teacher interface code