Update modules/morphosyntax/morphosyntax_process.py

modules/morphosyntax/morphosyntax_process.py

@@ -13,35 +13,104 @@ from ..text_analysis.morpho_analysis import (
 
 from ..database.morphosintax_mongo_db import store_student_morphosyntax_result
 
-from ..chatbot.chatbot import process_chat_input
+import logging
+logger = logging.getLogger(__name__)
 
-def process_morphosyntactic_input(user_input, lang_code, nlp_models, t):
-    if user_input.startswith('/analisis_morfosintactico'):
-        text_to_analyze = user_input.split('[', 1)[1].rsplit(']', 1)[0]
-        result = perform_advanced_morphosyntactic_analysis(text_to_analyze, nlp_models[lang_code])
-
-        if result is None or 'arc_diagrams' not in result:
-            response = t.get('morphosyntactic_analysis_error', 'Error in morphosyntactic analysis')
-            return response, None, None
-
-        response = t.get('morphosyntactic_analysis_completed', 'Morphosyntactic analysis completed')
-        visualizations = result['arc_diagrams']
+def process_morphosyntactic_input(text, lang_code, nlp_models, t):
+    """
+    Procesa el texto ingresado para realizar el análisis morfosintáctico.
+    
+    Args:
+        text: Texto a analizar
+        lang_code: Código del idioma
+        nlp_models: Diccionario de modelos spaCy
+        t: Diccionario de traducciones
 
+    Returns:
+        tuple: (análisis, visualizaciones, texto_resaltado, mensaje)
+    """
+    try:
+        # Realizar el análisis morfosintáctico
+        doc = nlp_models[lang_code](text)
+        
+        # Obtener el análisis avanzado
+        analysis = perform_advanced_morphosyntactic_analysis(text, nlp_models[lang_code])
+        
+        # Generar visualizaciones
+        arc_diagrams = generate_arc_diagram(doc, lang_code)
+        
+        # Obtener palabras repetidas y texto resaltado
+        word_colors = get_repeated_words_colors(doc)
+        highlighted_text = highlight_repeated_words(doc, word_colors)
+        
+        # Guardar el análisis en la base de datos
         store_student_morphosyntax_result(
             st.session_state.username,
-            text_to_analyze,
-            visualizations
+            text,
+            {
+                'arc_diagrams': arc_diagrams,
+                'pos_analysis': analysis['pos_analysis'],
+                'morphological_analysis': analysis['morphological_analysis'],
+                'sentence_structure': analysis['sentence_structure']
+            }
         )
+        
+        return {
+            'analysis': analysis,
+            'visualizations': arc_diagrams,
+            'highlighted_text': highlighted_text,
+            'success': True,
+            'message': t.get('MORPHOSYNTACTIC', {}).get('success_message', 'Analysis completed successfully')
+        }
+        
+    except Exception as e:
+        logger.error(f"Error en el análisis morfosintáctico: {str(e)}")
+        return {
+            'analysis': None,
+            'visualizations': None,
+            'highlighted_text': None,
+            'success': False,
+            'message': t.get('MORPHOSYNTACTIC', {}).get('error_message', f'Error in analysis: {str(e)}')
+        }
 
-        return response, visualizations, result
-    else:
-        chatbot = st.session_state.morphosyntax_chat_input
-        response = chatbot.generate_response(user_input, lang_code)
-        return response, None, None
-
+def format_analysis_results(analysis_result, t):
+    """
+    Formatea los resultados del análisis para su visualización.
+    
+    Args:
+        analysis_result: Resultado del análisis morfosintáctico
+        t: Diccionario de traducciones
+    
+    Returns:
+        dict: Resultados formateados para visualización
+    """
+    morpho_t = t.get('MORPHOSYNTACTIC', {})
+    
+    if not analysis_result['success']:
+        return {
+            'formatted_text': analysis_result['message'],
+            'visualizations': None
+        }
+        
+    formatted_sections = []
+    
+    # Formato para análisis POS
+    if 'pos_analysis' in analysis_result['analysis']:
+        pos_section = [f"### {morpho_t.get('pos_analysis', 'Part of Speech Analysis')}"]
+        for pos_item in analysis_result['analysis']['pos_analysis']:
+            pos_section.append(
+                f"- {morpho_t.get(pos_item['pos'], pos_item['pos'])}: "
+                f"{pos_item['count']} ({pos_item['percentage']}%)\n  "
+                f"Ejemplos: {', '.join(pos_item['examples'])}"
+            )
+        formatted_sections.append('\n'.join(pos_section))
+    
+    # Agregar otras secciones de formato según sea necesario
+    
+    return {
+        'formatted_text': '\n\n'.join(formatted_sections),
+        'visualizations': analysis_result['visualizations'],
+        'highlighted_text': analysis_result['highlighted_text']
+    }
 
-def format_analysis_response(result, t, lang_code):
-    # Formatea la respuesta del análisis en texto legible
-    # Usa las traducciones (t) para los títulos y etiquetas
-    # ...
-    pass
+__all__ = ['process_morphosyntactic_input', 'format_analysis_results']