Update modules/ui.py

modules/ui.py

@@ -12,7 +12,16 @@ from spacy import displacy
 #Importaciones locales
 #Importaciones locales de autenticación y base de datos 
 from .auth import authenticate_user, register_user
-from .database import get_student_data, store_morphosyntax_result, store_semantic_result, store_chat_history, create_admin_user, create_student_user
+
+from .database import (
+    get_student_data, 
+    store_morphosyntax_result, 
+    store_semantic_result, 
+    store_discourse_analysis_result,
+    store_chat_history, 
+    create_admin_user, 
+    create_student_user
+)
 
 #Importaciones locales de uiadmin
 from .admin_ui import admin_page
@@ -257,7 +266,10 @@ def display_morphosyntax_analysis_interface(nlp_models, lang_code):
             'repeated_words': "Palabras repetidas",
             'legend': "Leyenda: Categorías gramaticales",
             'arc_diagram': "Análisis sintáctico: Diagrama de arco",
-            'sentence': "Oración"
+            'sentence': "Oración",
+            'success_message': "Análisis guardado correctamente.",
+            'error_message': "Hubo un problema al guardar el análisis. Por favor, inténtelo de nuevo.",
+            'warning_message': "Por favor, ingrese un texto para analizar."
         },
         'en': {
             'title': "AIdeaText - Morphological and Syntactic Analysis",
@@ -267,7 +279,10 @@ def display_morphosyntax_analysis_interface(nlp_models, lang_code):
             'repeated_words': "Repeated words",
             'legend': "Legend: Grammatical categories",
             'arc_diagram': "Syntactic analysis: Arc diagram",
-            'sentence': "Sentence"
+            'sentence': "Sentence",
+            'success_message': "Analysis saved successfully.",
+            'error_message': "There was a problem saving the analysis. Please try again.",
+            'warning_message': "Please enter a text to analyze."
         },
         'fr': {
             'title': "AIdeaText - Analyse morphologique et syntaxique",
@@ -277,7 +292,10 @@ def display_morphosyntax_analysis_interface(nlp_models, lang_code):
             'repeated_words': "Mots répétés",
             'legend': "Légende : Catégories grammaticales",
             'arc_diagram': "Analyse syntaxique : Diagramme en arc",
-            'sentence': "Phrase"
+            'sentence': "Phrase",
+            'success_message': "Analyse enregistrée avec succès.",
+            'error_message': "Un problème est survenu lors de l'enregistrement de l'analyse. Veuillez réessayer.",
+            'warning_message': "Veuillez entrer un texte à analyser."
         }
     }
 
@@ -285,21 +303,16 @@ def display_morphosyntax_analysis_interface(nlp_models, lang_code):
 
     input_key = f"morphosyntax_input_{lang_code}"
 
-    # Inicializar la clave en session_state si no existe
     if input_key not in st.session_state:
         st.session_state[input_key] = ""
 
-    # Función para actualizar el estado del input
-    def update_input():
-        st.session_state[input_key] = st.session_state[f"text_area_{lang_code}"]
-
     sentence_input = st.text_area(
         t['input_label'],
         height=150,
         placeholder=t['input_placeholder'],
         value=st.session_state[input_key],
         key=f"text_area_{lang_code}",
-        on_change=update_input
+        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}"):
@@ -339,11 +352,11 @@ def display_morphosyntax_analysis_interface(nlp_models, lang_code):
                 word_colors,
                 arc_diagrams,
             ):
-                st.success("Análisis guardado correctamente.")
+                st.success(t['success_message'])
             else:
-                st.error("Hubo un problema al guardar el análisis. Por favor, inténtelo de nuevo.")
+                st.error(t['error_message'])
         else:
-            st.warning("Por favor, ingrese un texto para analizar.")
+            st.warning(t['warning_message'])
 
 ###############################################################################################################
 def display_semantic_analysis_interface(nlp_models, lang_code):
@@ -351,93 +364,189 @@ def display_semantic_analysis_interface(nlp_models, lang_code):
         'es': {
             'title': "AIdeaText - Análisis semántico",
             'file_uploader': "Cargar archivo de texto",
+            'input_label': "O ingrese el texto aquí:",
             'analyze_button': "Analizar texto",
             'semantic_relations': "Relaciones Semánticas Relevantes",
+            'success_message': "Análisis semántico guardado correctamente.",
+            'error_message': "Hubo un problema al guardar el análisis semántico. Por favor, inténtelo de nuevo.",
+            'warning_message': "Por favor, ingrese un texto o cargue un archivo para analizar."
         },
         'en': {
             'title': "AIdeaText - Semantic Analysis",
             'file_uploader': "Upload text file",
+            'input_label': "Or enter the text here:",
             'analyze_button': "Analyze text",
             'semantic_relations': "Relevant Semantic Relations",
+            'success_message': "Semantic analysis saved successfully.",
+            'error_message': "There was a problem saving the semantic analysis. Please try again.",
+            'warning_message': "Please enter a text or upload a file to analyze."
         },
         'fr': {
             'title': "AIdeaText - Analyse sémantique",
             'file_uploader': "Télécharger le fichier texte",
+            'input_label': "Ou entrez le texte ici :",
             'analyze_button': "Analyser le texte",
             'semantic_relations': "Relations Sémantiques Pertinentes",
+            'success_message': "Analyse sémantique enregistrée avec succès.",
+            'error_message': "Un problème est survenu lors de l'enregistrement de l'analyse sémantique. Veuillez réessayer.",
+            'warning_message': "Veuillez entrer un texte ou télécharger un fichier à analyser."
         }
     }
 
     t = translations[lang_code]
-
     st.header(t['title'])
 
+    # Inicializar el estado de la sesión para el texto semántico
+    if 'semantic_text' not in st.session_state:
+        st.session_state.semantic_text = ""
+
+    # Opción para cargar archivo
     uploaded_file = st.file_uploader(t['file_uploader'], type=['txt'])
 
-    if uploaded_file is not None:
-        text_content = uploaded_file.getvalue().decode('utf-8')
+    # Opción para ingresar texto directamente
+    text_input = st.text_area(
+        t['input_label'], 
+        value=st.session_state.semantic_text,
+        height=300,
+        key="semantic_text_input"
+    )
 
-        if st.button(t['analyze_button']):
-            relations_graph = perform_semantic_analysis(text_content, nlp_models[lang_code], lang_code)
+    if st.button(t['analyze_button']):
+        # Priorizar el archivo cargado sobre el texto ingresado
+        if uploaded_file is not None:
+            text_content = uploaded_file.getvalue().decode('utf-8')
+        elif text_input:
+            text_content = text_input
+        else:
+            st.warning(t['warning_message'])
+            return
 
-            with st.expander(t['semantic_relations'], expanded=True):
-                st.pyplot(relations_graph)
+        # Guardar el texto en el estado de la sesión
+        st.session_state.semantic_text = text_content
 
-##################################################################################################
+        # Realizar el análisis
+        relations_graph = perform_semantic_analysis(text_content, nlp_models[lang_code], lang_code)
+
+        # Mostrar el gráfico de relaciones semánticas
+        with st.expander(t['semantic_relations'], expanded=True):
+            st.pyplot(relations_graph)
 
+        # Guardar el resultado del análisis
+        if store_semantic_result(st.session_state.username, text_content, relations_graph):
+            st.success(t['success_message'])
+        else:
+            st.error(t['error_message'])
+
+##################################################################################################
 def display_discourse_analysis_interface(nlp_models, lang_code):
     translations = {
         'es': {
             'title': "AIdeaText - Análisis del discurso",
             'file_uploader1': "Cargar archivo de texto 1 (Patrón)",
             'file_uploader2': "Cargar archivo de texto 2 (Comparación)",
+            'input_label1': "O ingrese el texto 1 aquí:",
+            'input_label2': "O ingrese el texto 2 aquí:",
             'analyze_button': "Analizar textos",
             'comparison': "Comparación de Relaciones Semánticas",
+            'success_message': "Análisis del discurso guardado correctamente.",
+            'error_message': "Hubo un problema al guardar el análisis del discurso. Por favor, inténtelo de nuevo.",
+            'warning_message': "Por favor, ingrese ambos textos o cargue ambos archivos para analizar."
         },
         'en': {
             'title': "AIdeaText - Discourse Analysis",
             'file_uploader1': "Upload text file 1 (Pattern)",
             'file_uploader2': "Upload text file 2 (Comparison)",
+            'input_label1': "Or enter text 1 here:",
+            'input_label2': "Or enter text 2 here:",
             'analyze_button': "Analyze texts",
             'comparison': "Comparison of Semantic Relations",
+            'success_message': "Discourse analysis saved successfully.",
+            'error_message': "There was a problem saving the discourse analysis. Please try again.",
+            'warning_message': "Please enter both texts or upload both files to analyze."
         },
         'fr': {
             'title': "AIdeaText - Analyse du discours",
             'file_uploader1': "Télécharger le fichier texte 1 (Modèle)",
             'file_uploader2': "Télécharger le fichier texte 2 (Comparaison)",
+            'input_label1': "Ou entrez le texte 1 ici :",
+            'input_label2': "Ou entrez le texte 2 ici :",
             'analyze_button': "Analyser les textes",
             'comparison': "Comparaison des Relations Sémantiques",
+            'success_message': "Analyse du discours enregistrée avec succès.",
+            'error_message': "Un problème est survenu lors de l'enregistrement de l'analyse du discours. Veuillez réessayer.",
+            'warning_message': "Veuillez entrer les deux textes ou télécharger les deux fichiers à analyser."
         }
     }
 
     t = translations[lang_code]
-
     st.header(t['title'])
 
+    # Inicializar el estado de la sesión para los textos del discurso
+    if 'discourse_text1' not in st.session_state:
+        st.session_state.discourse_text1 = ""
+    if 'discourse_text2' not in st.session_state:
+        st.session_state.discourse_text2 = ""
+
     col1, col2 = st.columns(2)
 
     with col1:
         uploaded_file1 = st.file_uploader(t['file_uploader1'], type=['txt'])
+        text_input1 = st.text_area(
+            t['input_label1'], 
+            value=st.session_state.discourse_text1,
+            height=300,
+            key="discourse_text_input1"
+        )
 
     with col2:
         uploaded_file2 = st.file_uploader(t['file_uploader2'], type=['txt'])
+        text_input2 = st.text_area(
+            t['input_label2'], 
+            value=st.session_state.discourse_text2,
+            height=300,
+            key="discourse_text_input2"
+        )
+
+    if st.button(t['analyze_button']):
+        # Priorizar los archivos cargados sobre el texto ingresado
+        if uploaded_file1 is not None:
+            text_content1 = uploaded_file1.getvalue().decode('utf-8')
+        elif text_input1:
+            text_content1 = text_input1
+        else:
+            text_content1 = None
 
-    if uploaded_file1 is not None and uploaded_file2 is not None:
-        text_content1 = uploaded_file1.getvalue().decode('utf-8')
-        text_content2 = uploaded_file2.getvalue().decode('utf-8')
-
-        if st.button(t['analyze_button']):
-            graph1, graph2 = perform_discourse_analysis(text_content1, text_content2, nlp_models[lang_code], lang_code)
-
-            st.subheader(t['comparison'])
-            
-            col1, col2 = st.columns(2)
-            
-            with col1:
-                st.pyplot(graph1)
-            
-            with col2:
-                st.pyplot(graph2)
+        if uploaded_file2 is not None:
+            text_content2 = uploaded_file2.getvalue().decode('utf-8')
+        elif text_input2:
+            text_content2 = text_input2
+        else:
+            text_content2 = None
+
+        if text_content1 is None or text_content2 is None:
+            st.warning(t['warning_message'])
+            return
+
+        # Guardar los textos en el estado de la sesión
+        st.session_state.discourse_text1 = text_content1
+        st.session_state.discourse_text2 = text_content2
+
+        # Realizar el análisis
+        graph1, graph2 = perform_discourse_analysis(text_content1, text_content2, nlp_models[lang_code], lang_code)
+
+        # Mostrar los gráficos de comparación
+        st.subheader(t['comparison'])
+        col1, col2 = st.columns(2)
+        with col1:
+            st.pyplot(graph1)
+        with col2:
+            st.pyplot(graph2)
+
+        # Guardar el resultado del análisis
+        if store_discourse_analysis_result(st.session_state.username, text_content1 + "\n\n" + text_content2, graph1, graph2):
+            st.success(t['success_message'])
+        else:
+            st.error(t['error_message'])
 
 ##################################################################################################