Update modules/morphosyntax/morphosyntax_interface.py

modules/morphosyntax/morphosyntax_interface.py

@@ -1,5 +1,6 @@
 #modules/morphosyntax/morphosyntax_interface.py
 import streamlit as st
+import spacy_streamlit
 from streamlit_float import *
 from streamlit_antd_components import *
 from streamlit.components.v1 import html
@@ -20,52 +21,61 @@ def display_morphosyntax_interface(lang_code, nlp_models, t):
     Interfaz para el análisis morfosintáctico
     """
     morpho_t = t.get('MORPHOSYNTACTIC', {})
-    st.title(morpho_t.get('title', 'AIdeaText - Morphological 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:
         st.session_state[input_key] = ""
 
     sentence_input = st.text_area(
-        morpho_t.get('morpho_input_label', 'Enter text to analyze:'),
+        t['input_label'],
         height=150,
-        placeholder=morpho_t.get('morpho_input_placeholder', 'Enter your text here...'),
+        placeholder=t['input_placeholder'],
         value=st.session_state[input_key],
-        key=f"text_area_{lang_code}"
+        key=f"text_area_{lang_code}",
+        on_change=lambda: setattr(st.session_state, input_key, st.session_state[f"text_area_{lang_code}"])
     )
 
-    if st.button(morpho_t.get('analyze_button', 'Analyze'), key=f"analyze_button_{lang_code}"):
-        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("---")
+    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)
+
+            # 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)
+
+            # 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(result['message'])
+                st.error(t['error_message'])
         else:
-            st.warning(morpho_t.get('warning_message', 'Please enter a text to analyze.'))
+            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 de exportación
     if st.button(morpho_t.get('export_button', 'Export Analysis')):