Update modules/database/semantic_mongo_db.py

modules/database/semantic_mongo_db.py

@@ -25,21 +25,24 @@ COLLECTION_NAME = 'student_semantic_analysis'
 ####################################################################
 # modules/database/semantic_mongo_db.py
 
-def store_student_semantic_result(username, text, analysis_result):
+# modules/database/semantic_mongo_db.py
+
+def store_student_semantic_result(username, text, analysis_result, lang_code='en'):
     """
     Guarda el resultado del análisis semántico en MongoDB.
     Args:
         username: Nombre del usuario
         text: Texto completo analizado
         analysis_result: Diccionario con los resultados del análisis
+        lang_code: Código de idioma (opcional, default 'en')
     """
     try:
-        # Verificar que los datos requeridos estén presentes
-        if not isinstance(analysis_result, dict):
-            logger.error("El resultado del análisis no es un diccionario")
+        # Verificar datos mínimos requeridos
+        if not username or not text or not analysis_result:
+            logger.error("Datos insuficientes para guardar el análisis")
             return False
 
-        # Preparar el gráfico conceptual (si existe)
+        # Preparar el gráfico conceptual
         concept_graph_data = None
         if 'concept_graph' in analysis_result and analysis_result['concept_graph'] is not None:
             try:
@@ -50,31 +53,24 @@ def store_student_semantic_result(username, text, analysis_result):
             except Exception as e:
                 logger.error(f"Error al codificar gráfico conceptual: {str(e)}")
 
-        # Crear documento con estructura consistente
+        # Crear documento para MongoDB
         analysis_document = {
             'username': username,
             'timestamp': datetime.now(timezone.utc),
             'text': text,
             'analysis_type': 'semantic',
-            'metrics': {
-                'key_concepts': analysis_result.get('key_concepts', []),
-                'concept_centrality': analysis_result.get('concept_centrality', {})
-            },
-            'graph_data': concept_graph_data
-            #'language': st.session_state.get('lang_code', 'en')
+            'key_concepts': analysis_result.get('key_concepts', []),
+            'concept_centrality': analysis_result.get('concept_centrality', {}),
+            'concept_graph': concept_graph_data,
+            'language': lang_code  # Usamos el parámetro directamente
         }
 
         # Insertar en MongoDB
-        collection = get_collection(COLLECTION_NAME)
-        if not collection:
-            logger.error("No se pudo obtener la colección de MongoDB")
-            return False
-
-        result = collection.insert_one(analysis_document)
-        if result.inserted_id:
-            logger.info(f"Análisis semántico guardado para {username} con ID: {result.inserted_id}")
+        result = insert_document(COLLECTION_NAME, analysis_document)
+        if result:
+            logger.info(f"Análisis semántico guardado para {username}")
             return True
-
+        
         logger.error("No se pudo insertar el documento en MongoDB")
         return False