Spaces:
Running
Running
| #modules/morphosyntax/morphosyntax_interface.py | |
| import streamlit as st | |
| from streamlit_float import * | |
| from streamlit_antd_components import * | |
| from streamlit.components.v1 import html | |
| import spacy | |
| from spacy import displacy | |
| import spacy_streamlit | |
| import pandas as pd | |
| import base64 | |
| import re | |
| from .morphosyntax_process import ( | |
| process_morphosyntactic_input, | |
| format_analysis_results, | |
| perform_advanced_morphosyntactic_analysis, | |
| get_repeated_words_colors, | |
| highlight_repeated_words, | |
| POS_COLORS, | |
| POS_TRANSLATIONS | |
| ) | |
| from ..utils.widget_utils import generate_unique_key | |
| from ..database.morphosintax_mongo_db import store_student_morphosyntax_result | |
| from ..database.chat_mongo_db import store_chat_history, get_chat_history | |
| import logging | |
| logger = logging.getLogger(__name__) | |
| def display_morphosyntax_interface(lang_code, nlp_models, morpho_t): | |
| try: | |
| # Inicializar el estado si no existe | |
| if 'morphosyntax_state' not in st.session_state: | |
| st.session_state.morphosyntax_state = { | |
| 'analysis_count': 0, | |
| 'current_text': '', # Almacenar el texto actual | |
| 'last_analysis': None, | |
| 'needs_update': False # Flag para actualización | |
| } | |
| # Campo de entrada de texto que mantiene su valor | |
| text_key = "morpho_text_input" | |
| # Función para manejar cambios en el texto | |
| def on_text_change(): | |
| st.session_state.morphosyntax_state['current_text'] = st.session_state[text_key] | |
| st.session_state.morphosyntax_state['needs_update'] = True | |
| # Recuperar el texto anterior si existe | |
| default_text = st.session_state.morphosyntax_state.get('current_text', '') | |
| sentence_input = st.text_area( | |
| morpho_t.get('morpho_input_label', 'Enter text to analyze'), | |
| value=default_text, # Usar el texto guardado | |
| height=150, | |
| key=text_key, | |
| on_change=on_text_change, | |
| placeholder=morpho_t.get('morpho_input_placeholder', 'Enter your text here...') | |
| ) | |
| # Botón de análisis | |
| col1, col2, col3 = st.columns([2,1,2]) | |
| with col1: | |
| analyze_button = st.button( | |
| morpho_t.get('morpho_analyze_button', 'Analyze Morphosyntax'), | |
| key=f"morpho_button_{st.session_state.morphosyntax_state['analysis_count']}", | |
| type="primary", | |
| icon="🔍", | |
| disabled=not bool(sentence_input.strip()), | |
| use_container_width=True | |
| ) | |
| # Procesar análisis solo cuando sea necesario | |
| if (analyze_button or st.session_state.morphosyntax_state['needs_update']) and sentence_input.strip(): | |
| try: | |
| with st.spinner(morpho_t.get('processing', 'Processing...')): | |
| doc = nlp_models[lang_code](sentence_input) | |
| advanced_analysis = perform_advanced_morphosyntactic_analysis( | |
| sentence_input, | |
| nlp_models[lang_code] | |
| ) | |
| st.session_state.morphosyntax_result = { | |
| 'doc': doc, | |
| 'advanced_analysis': advanced_analysis | |
| } | |
| # Solo guardar en DB si fue un click en el botón | |
| if analyze_button: | |
| if store_student_morphosyntax_result( | |
| username=st.session_state.username, | |
| text=sentence_input, | |
| arc_diagrams=advanced_analysis['arc_diagrams'] | |
| ): | |
| st.success(morpho_t.get('success_message', 'Analysis saved successfully')) | |
| st.session_state.morphosyntax_state['analysis_count'] += 1 | |
| st.session_state.morphosyntax_state['needs_update'] = False | |
| # Mostrar resultados en un contenedor específico | |
| with st.container(): | |
| display_morphosyntax_results( | |
| st.session_state.morphosyntax_result, | |
| lang_code, | |
| morpho_t | |
| ) | |
| except Exception as e: | |
| logger.error(f"Error en análisis morfosintáctico: {str(e)}") | |
| st.error(morpho_t.get('error_processing', f'Error processing text: {str(e)}')) | |
| # Mostrar resultados previos si existen | |
| elif 'morphosyntax_result' in st.session_state and st.session_state.morphosyntax_result: | |
| with st.container(): | |
| display_morphosyntax_results( | |
| st.session_state.morphosyntax_result, | |
| lang_code, | |
| morpho_t | |
| ) | |
| except Exception as e: | |
| logger.error(f"Error general en display_morphosyntax_interface: {str(e)}") | |
| st.error("Se produjo un error. Por favor, intente de nuevo.") | |
| def display_morphosyntax_results(result, lang_code, morpho_t): | |
| """ | |
| Muestra solo el análisis sintáctico con diagramas de arco. | |
| """ | |
| if result is None: | |
| st.warning(morpho_t.get('no_results', 'No results available')) | |
| return | |
| doc = result['doc'] | |
| # Análisis sintáctico (diagramas de arco) | |
| st.markdown(f"### {morpho_t.get('arc_diagram', 'Syntactic analysis: Arc diagram')}") | |
| with st.container(): | |
| sentences = list(doc.sents) | |
| for i, sent in enumerate(sentences): | |
| with st.container(): | |
| st.subheader(f"{morpho_t.get('sentence', 'Sentence')} {i+1}") | |
| try: | |
| html = displacy.render(sent, style="dep", options={ | |
| "distance": 100, | |
| "arrow_spacing": 20, | |
| "word_spacing": 30 | |
| }) | |
| # Ajustar dimensiones del SVG | |
| html = html.replace('height="375"', 'height="200"') | |
| html = re.sub(r'<svg[^>]*>', lambda m: m.group(0).replace('height="450"', 'height="300"'), html) | |
| html = re.sub(r'<g [^>]*transform="translate\((\d+),(\d+)\)"', | |
| lambda m: f'<g transform="translate({m.group(1)},50)"', html) | |
| # Envolver en un div con clase para estilos | |
| html = f'<div class="arc-diagram-container">{html}</div>' | |
| st.write(html, unsafe_allow_html=True) | |
| except Exception as e: | |
| logger.error(f"Error rendering sentence {i}: {str(e)}") | |
| st.error(f"Error displaying diagram for sentence {i+1}") | |