##############
###modules/studentact/student_activities_v2.py

import streamlit as st
import re
import io
from io import BytesIO
import pandas as pd
import numpy as np
import time
import matplotlib.pyplot as plt
from datetime import datetime
from spacy import displacy
import random
import base64
import seaborn as sns
import logging

# Importaciones de la base de datos
from ..database.morphosintax_mongo_db import get_student_morphosyntax_analysis
from ..database.semantic_mongo_db import get_student_semantic_analysis
from ..database.discourse_mongo_db import get_student_discourse_analysis
from ..database.chat_mongo_db import get_chat_history
from ..database.current_situation_mongo_db import get_current_situation_analysis
from ..database.claude_recommendations_mongo_db import get_claude_recommendations

# Importar la función generate_unique_key
from ..utils.widget_utils import generate_unique_key

logger = logging.getLogger(__name__)

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

def display_student_activities(username: str, lang_code: str, t: dict):
    """
    Muestra todas las actividades del estudiante
    Args:
        username: Nombre del estudiante
        lang_code: Código del idioma
        t: Diccionario de traducciones
    """
    try:
        st.header(t.get('activities_title', 'Mis Actividades'))

        # Tabs para diferentes tipos de análisis
        tabs = st.tabs([
            t.get('current_situation_activities', 'Mi Situación Actual'),
            t.get('morpho_activities', 'Análisis Morfosintáctico'),
            t.get('semantic_activities', 'Análisis Semántico'),
            t.get('discourse_activities', 'Análisis del Discurso'),
            t.get('chat_activities', 'Conversaciones con el Asistente')
        ])

        # Tab de Situación Actual
        with tabs[0]:
            display_current_situation_activities(username, t)
        
        # Tab de Análisis Morfosintáctico
        with tabs[1]:
            display_morphosyntax_activities(username, t)

        # Tab de Análisis Semántico
        with tabs[2]:
            display_semantic_activities(username, t)

        # Tab de Análisis del Discurso
        with tabs[3]:
            display_discourse_activities(username, t)
            
        # Tab de Conversaciones del Chat
        with tabs[4]:
            display_chat_activities(username, t)

    except Exception as e:
        logger.error(f"Error mostrando actividades: {str(e)}")
        st.error(t.get('error_loading_activities', 'Error al cargar las actividades'))


###############################################################################################
def display_current_situation_activities(username: str, t: dict):
    """
    Muestra análisis de situación actual junto con las recomendaciones de Claude
    unificando la información de ambas colecciones.
    """
    try:
        # Recuperar datos de ambas colecciones
        logger.info(f"Recuperando análisis de situación actual para {username}")
        situation_analyses = get_student_current_situation(username)
        
        logger.info(f"Recuperando recomendaciones de Claude para {username}")
        claude_recommendations = get_claude_recommendations(username)
        
        # Verificar si hay algún tipo de análisis disponible
        if not situation_analyses and not claude_recommendations:
            logger.info("No se encontraron análisis de situación actual ni recomendaciones")
            st.info(t.get('no_current_situation', 'No hay análisis de situación actual registrados'))
            return
        
        # Crear un diccionario para indexar análisis por timestamp
        logger.info("Creando índice temporal de análisis")
        analyses_by_timestamp = {}
        
        # Indexar análisis de situación actual
        for analysis in situation_analyses:
            if 'timestamp' in analysis:
                timestamp_key = analysis['timestamp'].split('.')[0]
                if timestamp_key not in analyses_by_timestamp:
                    analyses_by_timestamp[timestamp_key] = {'situation': analysis}
        
        # Indexar recomendaciones de Claude 
        for recommendation in claude_recommendations:
            if 'timestamp' in recommendation:
                timestamp_key = recommendation['timestamp'].split('.')[0]
                if timestamp_key in analyses_by_timestamp:
                    analyses_by_timestamp[timestamp_key]['recommendation'] = recommendation
                else:
                    analyses_by_timestamp[timestamp_key] = {'recommendation': recommendation}
        
        # Si no hay análisis después de la indexación
        if not analyses_by_timestamp:
            st.info(t.get('no_paired_analyses', 'No hay análisis disponibles'))
            return
            
        # Convertir a lista ordenada por timestamp (más reciente primero)
        paired_analyses = sorted(
            analyses_by_timestamp.items(), 
            key=lambda x: x[0], 
            reverse=True
        )
        
        logger.info(f"Procesando {len(paired_analyses)} pares de análisis")
        
        # Mostrar cada par de análisis
        for i, (timestamp_key, analysis_pair) in enumerate(paired_analyses):
            try:
                # Obtener datos de situación y recomendación
                situation_data = analysis_pair.get('situation', {})
                recommendation_data = analysis_pair.get('recommendation', {})
                
                # Si no hay ningún dato, continuar al siguiente
                if not situation_data and not recommendation_data:
                    continue
                
                # Determinar qué texto mostrar
                text_to_show = situation_data.get('text', recommendation_data.get('text', ''))
                text_type = situation_data.get('text_type', recommendation_data.get('text_type', ''))
                
                # Formatear fecha para mostrar
                try:
                    timestamp_str = situation_data.get('timestamp', recommendation_data.get('timestamp', timestamp_key))
                    timestamp = datetime.fromisoformat(timestamp_str.replace('Z', '+00:00'))
                    formatted_date = timestamp.strftime("%d/%m/%Y %H:%M:%S")
                except Exception as date_error:
                    logger.error(f"Error formateando fecha: {str(date_error)}")
                    formatted_date = timestamp_key
                
                # Determinar el título del expander
                title = f"{t.get('analysis_date', 'Fecha')}: {formatted_date}"
                if text_type:
                    text_type_display = {
                        'academic_article': t.get('academic_article', 'Artículo académico'),
                        'student_essay': t.get('student_essay', 'Trabajo universitario'),
                        'general_communication': t.get('general_communication', 'Comunicación general')
                    }.get(text_type, text_type)
                    title += f" - {text_type_display}"
                
                # Mostrar el análisis en un expander
                with st.expander(title, expanded=False, key=f"situation_expander_{i}"):
                    # Mostrar texto analizado
                    st.subheader(t.get('analyzed_text', 'Texto analizado'))
                    st.text_area(
                        label="Texto analizado", 
                        value=text_to_show, 
                        height=100,
                        disabled=True,
                        key=f"situation_text_{i}",
                        label_visibility="hidden"
                    )
                    
                    # Crear tabs para separar diagnóstico y recomendaciones
                    diagnosis_tab, recommendations_tab = st.tabs([
                        t.get('diagnosis_tab', 'Diagnóstico'),
                        t.get('recommendations_tab', 'Recomendaciones')
                    ])
                    
                    # Tab de diagnóstico
                    with diagnosis_tab:
                        if situation_data and 'metrics' in situation_data:
                            metrics = situation_data['metrics']
                            
                            # Dividir en dos columnas
                            col1, col2 = st.columns(2)
                            
                            # Principales métricas en formato de tarjetas
                            with col1:
                                st.subheader(t.get('key_metrics', 'Métricas clave'))
                                
                                # Mostrar cada métrica principal
                                for j, (metric_name, metric_data) in enumerate(metrics.items()):
                                    if isinstance(metric_data, dict) and 'normalized_score' in metric_data:
                                        score = metric_data['normalized_score']
                                        
                                        # Determinar color y emoji basado en la puntuación
                                        if score < 0.5:
                                            emoji = "🔴"
                                            color = "#ffcccc"  # light red
                                        elif score < 0.75:
                                            emoji = "🟡"
                                            color = "#ffffcc"  # light yellow
                                        else:
                                            emoji = "🟢"
                                            color = "#ccffcc"  # light green
                                        
                                        # Mostrar la métrica con estilo
                                        st.markdown(f"""
                                        <div style="background-color:{color}; padding:10px; border-radius:5px; margin-bottom:10px;">
                                            <b>{emoji} {metric_name.capitalize()}:</b> {score:.2f}
                                        </div>
                                        """, unsafe_allow_html=True)
                            
                            # Mostrar detalles adicionales si están disponibles
                            with col2:
                                st.subheader(t.get('details', 'Detalles'))
                                
                                # Mostrar detalles como texto simple
                                for j, (metric_name, metric_data) in enumerate(metrics.items()):
                                    if isinstance(metric_data, dict) and 'details' in metric_data and metric_data['details']:
                                        st.markdown(f"**{metric_name.capitalize()}**")
                                        st.json(metric_data['details'])
                        else:
                            st.info(t.get('no_diagnosis', 'No hay datos de diagnóstico disponibles'))
                    
                    # Tab de recomendaciones
                    with recommendations_tab:
                        if recommendation_data and 'recommendations' in recommendation_data:
                            st.markdown(f"""
                            <div style="padding: 20px; border-radius: 10px; 
                                background-color: #f8f9fa; margin-bottom: 20px;">
                                {recommendation_data['recommendations']}
                            </div>
                            """, unsafe_allow_html=True)
                        elif recommendation_data and 'feedback' in recommendation_data:
                            st.markdown(f"""
                            <div style="padding: 20px; border-radius: 10px; 
                                background-color: #f8f9fa; margin-bottom: 20px;">
                                {recommendation_data['feedback']}
                            </div>
                            """, unsafe_allow_html=True)
                        else:
                            st.info(t.get('no_recommendations', 'No hay recomendaciones disponibles'))
            
            except Exception as e:
                logger.error(f"Error procesando par de análisis: {str(e)}")
                continue
        
    except Exception as e:
        logger.error(f"Error mostrando actividades de situación actual: {str(e)}")
        st.error(t.get('error_current_situation', 'Error al mostrar análisis de situación actual'))

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

def display_morphosyntax_activities(username: str, t: dict):
    """Muestra actividades de análisis morfosintáctico"""
    try:
        analyses = get_student_morphosyntax_analysis(username)
        if not analyses:
            st.info(t.get('no_morpho_analyses', 'No hay análisis morfosintácticos registrados'))
            return

        for analysis in analyses:
            with st.expander(
                f"{t.get('analysis_date', 'Fecha')}: {analysis['timestamp']}", 
                expanded=False
            ):
                st.text(f"{t.get('analyzed_text', 'Texto analizado')}:")
                st.write(analysis['text'])
                
                if 'arc_diagrams' in analysis:
                    st.subheader(t.get('syntactic_diagrams', 'Diagramas sintácticos'))
                    for diagram in analysis['arc_diagrams']:
                        st.write(diagram, unsafe_allow_html=True)

    except Exception as e:
        logger.error(f"Error mostrando análisis morfosintáctico: {str(e)}")
        st.error(t.get('error_morpho', 'Error al mostrar análisis morfosintáctico'))


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

def display_semantic_activities(username: str, t: dict):
    """Muestra actividades de análisis semántico"""
    try:
        logger.info(f"Recuperando análisis semántico para {username}")
        analyses = get_student_semantic_analysis(username)
        
        if not analyses:
            logger.info("No se encontraron análisis semánticos")
            st.info(t.get('no_semantic_analyses', 'No hay análisis semánticos registrados'))
            return

        logger.info(f"Procesando {len(analyses)} análisis semánticos")
        
        for analysis in analyses:
            try:
                # Verificar campos necesarios
                if not all(key in analysis for key in ['timestamp', 'concept_graph']):
                    logger.warning(f"Análisis incompleto: {analysis.keys()}")
                    continue
                
                # Formatear fecha
                timestamp = datetime.fromisoformat(analysis['timestamp'].replace('Z', '+00:00'))
                formatted_date = timestamp.strftime("%d/%m/%Y %H:%M:%S")
                
                # Crear expander
                with st.expander(f"{t.get('analysis_date', 'Fecha')}: {formatted_date}", expanded=False):
                    # Procesar y mostrar gráfico
                    if analysis.get('concept_graph'):
                        try:
                            # Convertir de base64 a bytes
                            logger.debug("Decodificando gráfico de conceptos")
                            image_data = analysis['concept_graph']
                            
                            # Si el gráfico ya es bytes, usarlo directamente
                            if isinstance(image_data, bytes):
                                image_bytes = image_data
                            else:
                                # Si es string base64, decodificar
                                image_bytes = base64.b64decode(image_data)
                            
                            logger.debug(f"Longitud de bytes de imagen: {len(image_bytes)}")
                            
                            # Mostrar imagen
                            st.image(
                                image_bytes,
                                caption=t.get('concept_network', 'Red de Conceptos'),
                                use_column_width=True
                            )
                            logger.debug("Gráfico mostrado exitosamente")
                            
                        except Exception as img_error:
                            logger.error(f"Error procesando gráfico: {str(img_error)}")
                            st.error(t.get('error_loading_graph', 'Error al cargar el gráfico'))
                    else:
                        st.info(t.get('no_graph', 'No hay visualización disponible'))

            except Exception as e:
                logger.error(f"Error procesando análisis individual: {str(e)}")
                continue

    except Exception as e:
        logger.error(f"Error mostrando análisis semántico: {str(e)}")
        st.error(t.get('error_semantic', 'Error al mostrar análisis semántico'))


###################################################################################################
def display_discourse_activities(username: str, t: dict):
    """Muestra actividades de análisis del discurso"""
    try:
        logger.info(f"Recuperando análisis del discurso para {username}")
        analyses = get_student_discourse_analysis(username)
        
        if not analyses:
            logger.info("No se encontraron análisis del discurso")
            st.info(t.get('no_discourse_analyses', 'No hay análisis del discurso registrados'))
            return

        logger.info(f"Procesando {len(analyses)} análisis del discurso")
        for analysis in analyses:
            try:
                # Verificar campos mínimos necesarios
                if not all(key in analysis for key in ['timestamp', 'combined_graph']):
                    logger.warning(f"Análisis incompleto: {analysis.keys()}")
                    continue

                # Formatear fecha
                timestamp = datetime.fromisoformat(analysis['timestamp'].replace('Z', '+00:00'))
                formatted_date = timestamp.strftime("%d/%m/%Y %H:%M:%S")
                
                with st.expander(f"{t.get('analysis_date', 'Fecha')}: {formatted_date}", expanded=False):
                    if analysis['combined_graph']:
                        logger.debug("Decodificando gráfico combinado")
                        try:
                            image_bytes = base64.b64decode(analysis['combined_graph'])
                            st.image(image_bytes, use_column_width=True)
                            logger.debug("Gráfico mostrado exitosamente")
                        except Exception as img_error:
                            logger.error(f"Error decodificando imagen: {str(img_error)}")
                            st.error(t.get('error_loading_graph', 'Error al cargar el gráfico'))
                    else:
                        st.info(t.get('no_visualization', 'No hay visualización comparativa disponible'))

            except Exception as e:
                logger.error(f"Error procesando análisis individual: {str(e)}")
                continue

    except Exception as e:
        logger.error(f"Error mostrando análisis del discurso: {str(e)}")
        st.error(t.get('error_discourse', 'Error al mostrar análisis del discurso'))

#################################################################################   
def display_chat_activities(username: str, t: dict):
    """
    Muestra historial de conversaciones del chat
    """
    try:
        # Obtener historial del chat
        chat_history = get_chat_history(
            username=username,
            analysis_type='sidebar',
            limit=50
        )
        
        if not chat_history:
            st.info(t.get('no_chat_history', 'No hay conversaciones registradas'))
            return

        for chat in reversed(chat_history):  # Mostrar las más recientes primero
            try:
                # Convertir timestamp a datetime para formato
                timestamp = datetime.fromisoformat(chat['timestamp'].replace('Z', '+00:00'))
                formatted_date = timestamp.strftime("%d/%m/%Y %H:%M:%S")
                
                with st.expander(
                    f"{t.get('chat_date', 'Fecha de conversación')}: {formatted_date}",
                    expanded=False
                ):
                    if 'messages' in chat and chat['messages']:
                        # Mostrar cada mensaje en la conversación
                        for message in chat['messages']:
                            role = message.get('role', 'unknown')
                            content = message.get('content', '')
                            
                            # Usar el componente de chat de Streamlit
                            with st.chat_message(role):
                                st.markdown(content)
                                
                            # Agregar separador entre mensajes
                            st.divider()
                    else:
                        st.warning(t.get('invalid_chat_format', 'Formato de chat no válido'))
                        
            except Exception as e:
                logger.error(f"Error mostrando conversación: {str(e)}")
                continue

    except Exception as e:
        logger.error(f"Error mostrando historial del chat: {str(e)}")
        st.error(t.get('error_chat', 'Error al mostrar historial del chat'))
        
#################################################################################        
def display_discourse_comparison(analysis: dict, t: dict):
    """Muestra la comparación de análisis del discurso"""
    st.subheader(t.get('comparison_results', 'Resultados de la comparación'))
    
    col1, col2 = st.columns(2)
    with col1:
        st.markdown(f"**{t.get('concepts_text_1', 'Conceptos Texto 1')}**")
        df1 = pd.DataFrame(analysis['key_concepts1'])
        st.dataframe(df1)
    
    with col2:
        st.markdown(f"**{t.get('concepts_text_2', 'Conceptos Texto 2')}**")
        df2 = pd.DataFrame(analysis['key_concepts2'])
        st.dataframe(df2)