#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__)

###########################################################################

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

############################################################################

def display_morphosyntax_interface(lang_code, nlp_models, morpho_t):
    try:
        # CSS mejorado para estabilidad y layout horizontal
        st.markdown("""
            <style>
            .stTextArea textarea {
                font-size: 1rem;
                line-height: 1.5;
                padding: 0.5rem;
                border-radius: 0.375rem;
                border: 1px solid #e2e8f0;
                background-color: white;
                min-height: 100px !important;
                height: 100px !important;
            }
            .block-container {
                padding-top: 0.5rem !important;
                padding-bottom: 0.5rem !important;
                margin: 0 !important;
            }
            .main-content {
                display: flex;
                flex-direction: column;
                gap: 1rem;
                padding: 0.5rem;
            }
            .arc-diagram-container {
                width: 100%;
                overflow-x: auto;
                background-color: white;
                padding: 0.5rem;
                border-radius: 0.375rem;
                box-shadow: 0 1px 2px rgba(0,0,0,0.1);
                margin-top: 0.5rem;
            }
            .edit-container {
                position: sticky;
                top: 0;
                background-color: white;
                z-index: 100;
                padding: 0.5rem 0;
            }
            </style>
        """, unsafe_allow_html=True)

        # Inicialización del estado con valores por defecto
        if 'morphosyntax_state' not in st.session_state:
            st.session_state['morphosyntax_state'] = {}
        
        default_state = {
            'has_original': False,
            'original_text': '',
            'original_analysis': None,
            'current_text': '',
            'iterations': [],
            'analysis_count': 0,
            'current_tab': 0  # Para mantener el tab activo
        }
        
        for key, value in default_state.items():
            if key not in st.session_state.morphosyntax_state:
                st.session_state.morphosyntax_state[key] = value

        with st.container():
            # Área de edición del texto original
            if not st.session_state.morphosyntax_state['has_original']:
                st.markdown("### Texto Original")
                original_text = st.text_area(
                    "Ingrese una oración",
                    key="original_text_input",
                    placeholder="Ingresar solo una oración hasta el punto y aparte. Si es punto seguido, dejar así.",
                    height=100
                )
                
                col1, col2, col3 = st.columns([2,1,2])
                with col1:
                    if st.button(
                        "Analizar Texto Original", 
                        type="primary",
                        use_container_width=True,
                        disabled=not bool(original_text.strip())
                    ):
                        try:
                            # Procesar texto original
                            doc = nlp_models[lang_code](original_text)
                            analysis = perform_advanced_morphosyntactic_analysis(
                                original_text,
                                nlp_models[lang_code]
                            )
                            
                            # Actualizar estado
                            st.session_state.morphosyntax_state.update({
                                'has_original': True,
                                'original_text': original_text,
                                'original_analysis': {
                                    'doc': doc,
                                    'advanced_analysis': analysis
                                },
                                'current_text': original_text
                            })
                            
                            if store_student_morphosyntax_result(
                                username=st.session_state.username,
                                text=original_text,
                                arc_diagrams=analysis['arc_diagrams']
                            ):
                                st.success("Texto analizado exitosamente")
                                st.rerun()
                            else:
                                st.error("Error al guardar el análisis")
                        except Exception as e:
                            logger.error(f"Error procesando texto original: {str(e)}")
                            st.error("Error al procesar el texto")
            else:
                # Layout horizontal para edición y visualización
                st.markdown("### Versión Original")
                
                # Permitir editar texto original
                edit_original = st.button("Editar texto original")
                if edit_original:
                    st.session_state.morphosyntax_state['has_original'] = False
                    st.rerun()
                
                col1, col2 = st.columns([1, 1])
                with col1:
                    st.text_area(
                        "Texto Original",
                        value=st.session_state.morphosyntax_state['original_text'],
                        disabled=True,
                        height=100
                    )
                
                with col2:
                    display_morphosyntax_results(
                        st.session_state.morphosyntax_state['original_analysis'],
                        lang_code,
                        morpho_t
                    )

                st.markdown("---")
                st.markdown("### Iteración Actual")
                
                # Campo para nueva versión con layout horizontal
                col1, col2 = st.columns([1, 1])
                with col1:
                    iteration_text = st.text_area(
                        "Modifique la oración",
                        value=st.session_state.morphosyntax_state['current_text'],
                        key=f"iteration_input_{st.session_state.morphosyntax_state['analysis_count']}",
                        placeholder="Ingresar solo una oración hasta el punto y aparte. Si es punto seguido, dejar así.",
                        height=100
                    )

                    analyze_button = st.button(
                        "Analizar Cambios",
                        type="primary",
                        icon="🔍",
                        key=f"analyze_{st.session_state.morphosyntax_state['analysis_count']}",
                        disabled=not bool(iteration_text.strip()),
                        use_container_width=True
                    )
                
                with col2:
                    if analyze_button and iteration_text.strip():
                        try:
                            with st.spinner("Procesando..."):
                                doc = nlp_models[lang_code](iteration_text)
                                analysis = perform_advanced_morphosyntactic_analysis(
                                    iteration_text,
                                    nlp_models[lang_code]
                                )
                                
                                current_analysis = {
                                    'doc': doc,
                                    'advanced_analysis': analysis
                                }
                                
                                # Guardar iteración
                                st.session_state.morphosyntax_state['iterations'].append({
                                    'text': iteration_text,
                                    'analysis': current_analysis,
                                    'timestamp': pd.Timestamp.now()
                                })
                                
                                st.session_state.morphosyntax_state['current_text'] = iteration_text
                                st.session_state.morphosyntax_state['analysis_count'] += 1
                                
                                if store_student_morphosyntax_result(
                                    username=st.session_state.username,
                                    text=iteration_text,
                                    arc_diagrams=analysis['arc_diagrams']
                                ):
                                    display_morphosyntax_results(
                                        current_analysis,
                                        lang_code,
                                        morpho_t
                                    )
                                else:
                                    st.error("Error al guardar el análisis")
                        except Exception as e:
                            logger.error(f"Error procesando iteración: {str(e)}")
                            st.error("Error al procesar los cambios")

                # Historial de iteraciones
                if st.session_state.morphosyntax_state['iterations']:
                    with st.expander("Historial de Iteraciones", expanded=False):
                        for idx, iteration in enumerate(reversed(st.session_state.morphosyntax_state['iterations'])):
                            col1, col2 = st.columns([1, 1])
                            with col1:
                                st.markdown(f"**Iteración {idx + 1} ({iteration['timestamp'].strftime('%H:%M:%S')})**")
                                st.text_area(
                                    f"Texto {idx + 1}",
                                    value=iteration['text'],
                                    disabled=True,
                                    height=100,
                                    key=f"hist_text_{idx}"
                                )
                            with col2:
                                display_morphosyntax_results(
                                    iteration['analysis'],
                                    lang_code,
                                    morpho_t
                                )
                            st.markdown("---")

    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.")

#########################################################################3
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}")