Update modules/morphosyntax/morphosyntax_interface.py

modules/morphosyntax/morphosyntax_interface.py

@@ -14,87 +14,174 @@ from ..database.morphosintaxis_export import export_user_interactions
 import logging
 logger = logging.getLogger(__name__)
 
-def display_morphosyntax_interface(lang_code, nlp_models, t):
-    #st.title("Análisis Morfosintáctico")
+def display_morphosyntax_analysis_interface(nlp_models, lang_code):
+    translations = {
+        'es': {
+            'title': "AIdeaText - Análisis morfológico y sintáctico",
+            'input_label': "Ingrese un texto para analizar (máximo 5,000 palabras",
+            'input_placeholder': "Esta funcionalidad le ayudará con dos competencias:\n" 
+                                 "[1] \"Escribe diversos tipos de textos en su lengua materna\"\n"
+                                 "[2] \"Lee diversos tipos de textos escritos en su lengua materna\"\n\n"
+                                 "Ingrese su texto aquí para analizar...",
+            'analyze_button': "Analizar texto",
+            'repeated_words': "Palabras repetidas",
+            'legend': "Leyenda: Categorías gramaticales",
+            'arc_diagram': "Análisis sintáctico: Diagrama de arco",
+            '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.",
+            'initial_message': "Ingrese un texto y presione 'Analizar texto' para comenzar.",
+            'no_results': "No hay resultados disponibles. Por favor, realice un análisis primero.",
+            'pos_analysis': "Análisis de categorías gramaticales",
+            'morphological_analysis': "Análisis morfológico",
+            'sentence_structure': "Estructura de oraciones",
+            'word': "Palabra",
+            'count': "Cantidad",
+            'percentage': "Porcentaje",
+            'examples': "Ejemplos",
+            'lemma': "Lema",
+            'tag': "Etiqueta",
+            'dep': "Dependencia",
+            'morph': "Morfología",
+            'root': "Raíz",
+            'subjects': "Sujetos",
+            'objects': "Objetos",
+            'verbs': "Verbos",
+            'grammatical_category': "Categoría gramatical",
+            'dependency': "Dependencia",
+            'morphology': "Morfología"
+        },
+        'en': {
+            'title': "AIdeaText - Morphological and Syntactic Analysis",
+            'input_label': "Enter a text to analyze (max 5,000 words):",
+            'input_placeholder': "This functionality will help you with two competencies:\n"
+                             "[1] \"Write various types of texts in your native language\"\n"
+                             "[2] \"Read various types of written texts in your native language\"\n\n"
+                             "Enter your text here to analyze...",
+            'analyze_button': "Analyze text",
+            'repeated_words': "Repeated words",
+            'legend': "Legend: Grammatical categories",
+            'arc_diagram': "Syntactic analysis: Arc diagram",
+            '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.",
+            'initial_message': "Enter a text and press 'Analyze text' to start.",
+            'no_results': "No results available. Please perform an analysis first.",
+            'pos_analysis': "Part of Speech Analysis",
+            'morphological_analysis': "Morphological Analysis",
+            'sentence_structure': "Sentence Structure",
+            'word': "Word",
+            'count': "Count",
+            'percentage': "Percentage",
+            'examples': "Examples",
+            'lemma': "Lemma",
+            'tag': "Tag",
+            'dep': "Dependency",
+            'morph': "Morphology",
+            'root': "Root",
+            'subjects': "Subjects",
+            'objects': "Objects",
+            'verbs': "Verbs", 
+            'grammatical_category': "Grammatical category",
+            'dependency': "Dependency",
+            'morphology': "Morphology"
+        },
+        'fr': {
+            'title': "AIdeaText - Analyse morphologique et syntaxique",
+            'input_label': "Entrez un texte à analyser (max 5 000 mots) :",
+            'input_placeholder': "Cette fonctionnalité vous aidera avec deux compétences :\n"
+                             "[1] \"Écrire divers types de textes dans votre langue maternelle\"\n"
+                             "[2] \"Lire divers types de textes écrits dans votre langue maternelle\"\n\n"
+                             "Entrez votre texte ici pour l'analyser...",
+            'analyze_button': "Analyser le texte",
+            'repeated_words': "Mots répétés",
+            'legend': "Légende : Catégories grammaticales",
+            'arc_diagram': "Analyse syntaxique : Diagramme en arc",
+            '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.",
+            'initial_message': "Entrez un texte et appuyez sur 'Analyser le texte' pour commencer.",
+            'no_results': "Aucun résultat disponible. Veuillez d'abord effectuer une analyse.",
+            'pos_analysis': "Analyse des parties du discours",
+            'morphological_analysis': "Analyse morphologique",
+            'sentence_structure': "Structure des phrases",
+            'word': "Mot",
+            'count': "Nombre",
+            'percentage': "Pourcentage",
+            'examples': "Exemples",
+            'lemma': "Lemme",
+            'tag': "Étiquette",
+            'dep': "Dépendance",
+            'morph': "Morphologie",
+            'root': "Racine",
+            'subjects': "Sujets",
+            'objects': "Objets",
+            'verbs': "Verbes",
+            'grammatical_category': "Catégorie grammaticale",
+            'dependency': "Dépendance",
+            'morphology': "Morphologie"
+        }
+    }
 
-    # Contenedor para el historial del chat
-    chat_container = st.container()
+    t = translations[lang_code]
 
-    # Input del usuario (siempre visible en la parte inferior)
-    user_input = st.chat_input(t['morpho_input_label'])
+    input_key = f"morphosyntax_input_{lang_code}"
 
-    # Inicializar el historial del chat si no existe
-    if 'morphosyntax_chat_history' not in st.session_state:
-        st.session_state.morphosyntax_chat_history = []
+    if input_key not in st.session_state:
+        st.session_state[input_key] = ""
 
-    # Procesar el input del usuario
-    if user_input:
-        # Añadir el mensaje del usuario al historial
-        st.session_state.morphosyntax_chat_history.append({"role": "user", "content": user_input})
-        store_chat_history(st.session_state.username, [{"role": "user", "content": user_input}], "morphosyntax")
+    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=lambda: setattr(st.session_state, input_key, st.session_state[f"text_area_{lang_code}"])
+    )
 
-        response, visualizations, result = process_morphosyntactic_input(user_input, lang_code, nlp_models, t)
+    if st.button(t['analyze_button'], key=f"analyze_button_{lang_code}"):
+        current_input = st.session_state[input_key]
+        if current_input:
+            doc = nlp_models[lang_code](current_input)
 
-        # Añadir la respuesta al historial
-        assistant_message = {
-            "role": "assistant",
-            "content": response,
-            "visualizations": visualizations if visualizations else []
-        }
-        st.session_state.morphosyntax_chat_history.append(assistant_message)
-        store_chat_history(st.session_state.username, [assistant_message], "morphosyntax")
+            # Análisis morfosintáctico avanzado
+            advanced_analysis = perform_advanced_morphosyntactic_analysis(current_input, nlp_models[lang_code])    
+
+            # 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, t)
 
-        # Si es un análisis, guardarlo en la base de datos
-        if user_input.startswith('/analisis_morfosintactico') and result:
-            store_student_morphosyntax_result(
+            # Guardar resultados
+            if store_morphosyntax_result(
                 st.session_state.username,
-                user_input.split('[', 1)[1].rsplit(']', 1)[0],  # texto analizado
-                visualizations
-            )
-
-    # Mostrar el historial del chat
-    with chat_container:
-        for message in st.session_state.morphosyntax_chat_history:
-            with st.chat_message(message["role"]):
-                st.write(message["content"])
-                if "visualizations" in message and message["visualizations"]:
-                    for i, viz in enumerate(message["visualizations"]):
-                        st.markdown(f"**Oración {i+1} del párrafo analizado**")
-                        st.components.v1.html(
-                            f"""
-                            <div style="width: 100%; overflow-x: auto; white-space: nowrap;">
-                                <div style="min-width: 1200px;">
-                                    {viz}
-                                </div>
-                            </div>
-                            """,
-                            height=370,
-                            scrolling=True
-                        )
-                        if i < len(message["visualizations"]) - 1:
-                            st.markdown("---")  # Separador entre diagramas
+                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.info(t['initial_message'])  # Añade esta traducción a tu diccionario    
 
-    # Botón para limpiar el historial del chat
-    if st.button(t['clear_chat'], key=generate_unique_key('morphosyntax', 'clear_chat')):
-        st.session_state.morphosyntax_chat_history = []
-        st.rerun()
 
-    # Botón de exportación
-    if st.button("Exportar Interacciones"):
-        pdf_buffer = export_user_interactions(st.session_state.username, 'morphosyntax')
-        st.download_button(
-            label="Descargar PDF",
-            data=pdf_buffer,
-            file_name="interacciones_morfosintaxis.pdf",
-            mime="application/pdf"
-        )
-
-def display_visualizations(visualizations, t):
-    for i, viz in enumerate(visualizations):
-        st.markdown(f"**{t['sentence']} {i+1}**")
-        st.components.v1.html(viz, height=370, scrolling=True)
-        if i < len(visualizations) - 1:
-            st.markdown("---")
 
 '''
     if user_input: