Update modules/morphosyntax/morphosyntax_interface.py

modules/morphosyntax/morphosyntax_interface.py

@@ -4,7 +4,9 @@ from streamlit_float import *
 from streamlit_antd_components import *
 from streamlit.components.v1 import html
 import base64
+
 from .morphosyntax_process import process_morphosyntactic_input, format_analysis_results
+
 from ..utils.widget_utils import generate_unique_key
 from ..database.morphosintax_mongo_db import store_student_morphosyntax_result
 from ..database.chat_db import store_chat_history
@@ -13,19 +15,12 @@ from ..database.morphosintaxis_export import export_user_interactions
 import logging
 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
     """
-    # Asegurarnos de que estamos accediendo al diccionario MORPHOSYNTACTIC
     morpho_t = t.get('MORPHOSYNTACTIC', {})
-    
-    st.title(morpho_t.get('title', 'AIdeaText - Morphological and Syntactic Analysis'))
+    st.title(morpho_t.get('title', 'AIdeaText - Morphological Analysis'))
 
     input_key = f"morphosyntax_input_{lang_code}"
     if input_key not in st.session_state:
@@ -36,49 +31,51 @@ def display_morphosyntax_interface(lang_code, nlp_models, t):
         height=150,
         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}"])
+        key=f"text_area_{lang_code}"
     )
 
     if st.button(morpho_t.get('analyze_button', 'Analyze'), 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])
-            
-            # 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)
-            
-            # 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(morpho_t.get('success_message', 'Analysis saved successfully.'))
+        if sentence_input:
+            # Usar el proceso morfosintáctico actualizado
+            result = process_morphosyntactic_input(
+                sentence_input,
+                lang_code,
+                nlp_models,
+                t
+            )
+
+            if result['success']:
+                # Formatear y mostrar resultados
+                formatted_results = format_analysis_results(result, t)
+                
+                # Mostrar texto resaltado si está disponible
+                if formatted_results['highlighted_text']:
+                    st.markdown(formatted_results['highlighted_text'], unsafe_allow_html=True)
+                
+                # Mostrar el análisis formateado
+                st.markdown(formatted_results['formatted_text'])
+                
+                # Mostrar visualizaciones
+                if formatted_results['visualizations']:
+                    for i, viz in enumerate(formatted_results['visualizations']):
+                        st.markdown(f"**{morpho_t.get('sentence', 'Sentence')} {i+1}**")
+                        st.components.v1.html(viz, height=370, scrolling=True)
+                        if i < len(formatted_results['visualizations']) - 1:
+                            st.markdown("---")
             else:
-                st.error(morpho_t.get('error_message', 'Error saving analysis.'))
+                st.error(result['message'])
         else:
             st.warning(morpho_t.get('warning_message', 'Please enter a text to analyze.'))
-    
-    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, morpho_t)
-    else:
-        st.info(morpho_t.get('morpho_initial_message', 'Enter text to begin analysis.'))
 
+    # Botón de exportación
+    if st.button(morpho_t.get('export_button', 'Export Analysis')):
+        pdf_buffer = export_user_interactions(st.session_state.username, 'morphosyntax')
+        st.download_button(
+            label=morpho_t.get('download_pdf', 'Download PDF'),
+            data=pdf_buffer,
+            file_name="morphosyntax_analysis.pdf",
+            mime="application/pdf"
+        )
 
 '''
     if user_input: