v3 / modules /semantic /semantic_interface.py
AIdeaText's picture
Update modules/semantic/semantic_interface.py
0ae8910 verified
raw
history blame
8.5 kB
#modules/semantic/semantic_interface.py
import streamlit as st
from streamlit_float import *
from streamlit_antd_components import *
from streamlit.components.v1 import html
import spacy_streamlit
import io
from io import BytesIO
import base64
import matplotlib.pyplot as plt
import pandas as pd
import re
import logging
# Configuración del logger
logger = logging.getLogger(__name__)
# Importaciones locales
from .semantic_process import (
process_semantic_input,
format_semantic_results
)
from ..utils.widget_utils import generate_unique_key
from ..database.semantic_mongo_db import store_student_semantic_result
from ..database.chat_mongo_db import store_chat_history, get_chat_history
# from ..database.semantic_export import export_user_interactions
###############################
def display_semantic_interface(lang_code, nlp_models, semantic_t):
"""
Interfaz para el análisis semántico
Args:
lang_code: Código del idioma actual
nlp_models: Modelos de spaCy cargados
semantic_t: Diccionario de traducciones semánticas
"""
try:
# 1. Inicializar el estado de la sesión
if 'semantic_state' not in st.session_state:
st.session_state.semantic_state = {
'analysis_count': 0,
'last_analysis': None,
'current_file': None
}
# 2. Área de carga de archivo con mensaje informativo
st.info(semantic_t.get('initial_instruction',
'Para comenzar un nuevo análisis semántico, cargue un archivo de texto (.txt)'))
uploaded_file = st.file_uploader(
semantic_t.get('semantic_file_uploader', 'Upload a text file for semantic analysis'),
type=['txt'],
key=f"semantic_file_uploader_{st.session_state.semantic_state['analysis_count']}"
)
# 3. Columnas para los botones y mensajes
col1, col2 = st.columns([1,4])
# 4. Botón de análisis
with col1:
analyze_button = st.button(
semantic_t.get('semantic_analyze_button', 'Analyze'),
key=f"semantic_analyze_button_{st.session_state.semantic_state['analysis_count']}",
type="primary", # Nuevo en Streamlit 1.39.0
icon="🔍", # Nuevo en Streamlit 1.39.0
disabled=uploaded_file is None,
use_container_width=True
)
# 5. Procesar análisis
if analyze_button and uploaded_file is not None:
try:
with st.spinner(semantic_t.get('processing', 'Processing...')):
# Leer contenido del archivo
text_content = uploaded_file.getvalue().decode('utf-8')
# Realizar análisis
analysis_result = process_semantic_input(
text_content,
lang_code,
nlp_models,
semantic_t
)
if analysis_result['success']:
# Guardar resultado
st.session_state.semantic_result = analysis_result
st.session_state.semantic_state['analysis_count'] += 1
st.session_state.semantic_state['current_file'] = uploaded_file.name
# Guardar en base de datos
if store_student_semantic_result(
st.session_state.username,
text_content,
analysis_result['analysis']
):
st.success(
semantic_t.get('analysis_complete',
'Análisis completado y guardado. Para realizar un nuevo análisis, cargue otro archivo.')
)
# Mostrar resultados
display_semantic_results(
st.session_state.semantic_result,
lang_code,
semantic_t
)
else:
st.error(semantic_t.get('error_message', 'Error saving analysis'))
else:
st.error(analysis_result['message'])
except Exception as e:
logger.error(f"Error en análisis semántico: {str(e)}")
st.error(semantic_t.get('error_processing', f'Error processing text: {str(e)}'))
# 6. Mostrar resultados previos o mensaje inicial
elif 'semantic_result' in st.session_state and st.session_state.semantic_result is not None:
# Mostrar mensaje sobre el análisis actual
st.info(
semantic_t.get('current_analysis_message',
f'Mostrando análisis del archivo: {st.session_state.semantic_state["current_file"]}. '
'Para realizar un nuevo análisis, cargue otro archivo.')
)
display_semantic_results(
st.session_state.semantic_result,
lang_code,
semantic_t
)
else:
st.info(semantic_t.get('upload_prompt', 'Cargue un archivo para comenzar el análisis'))
except Exception as e:
logger.error(f"Error general en interfaz semántica: {str(e)}")
st.error(semantic_t.get('general_error', "Se produjo un error. Por favor, intente de nuevo."))
#######################################
def display_semantic_results(semantic_result, lang_code, semantic_t):
"""
Muestra los resultados del análisis semántico de conceptos clave.
Args:
semantic_result: Diccionario con los resultados del análisis
lang_code: Código del idioma actual
semantic_t: Diccionario de traducciones semánticas
"""
# Verificar resultado
if semantic_result is None or not semantic_result['success']:
st.warning(semantic_t.get('no_results', 'No results available'))
return
analysis = semantic_result['analysis']
# Crear contenedor para los resultados
col1, col2 = st.columns(2)
# Columna 1: Lista de conceptos clave
with col1:
st.subheader(semantic_t.get('key_concepts', 'Key Concepts'))
if 'key_concepts' in analysis and analysis['key_concepts']:
# Crear tabla de conceptos
df = pd.DataFrame(
analysis['key_concepts'],
columns=[
semantic_t.get('concept', 'Concept'),
semantic_t.get('frequency', 'Frequency')
]
)
st.dataframe(
df,
hide_index=True,
column_config={
semantic_t.get('frequency', 'Frequency'): st.column_config.NumberColumn(
format="%.2f"
)
}
)
else:
st.info(semantic_t.get('no_concepts', 'No key concepts found'))
# Columna 2: Gráfico de conceptos
with col2:
st.subheader(semantic_t.get('concept_graph', 'Concepts Graph'))
if 'concept_graph' in analysis and analysis['concept_graph'] is not None:
st.image(analysis['concept_graph'])
else:
st.info(semantic_t.get('no_graph', 'No concept graph available'))
'''
# Botón de exportación al final
if 'semantic_analysis_counter' in st.session_state:
col1, col2, col3 = st.columns([2,1,2])
with col2:
if st.button(
semantic_t.get('export_button', 'Export Analysis'),
key=f"semantic_export_{st.session_state.semantic_analysis_counter}",
use_container_width=True
):
pdf_buffer = export_user_interactions(st.session_state.username, 'semantic')
st.download_button(
label=semantic_t.get('download_pdf', 'Download PDF'),
data=pdf_buffer,
file_name="semantic_analysis.pdf",
mime="application/pdf",
key=f"semantic_download_{st.session_state.semantic_analysis_counter}"
)
'''