Create semantic_mongo_live_db.py

modules/database/semantic_mongo_live_db.py

@@ -0,0 +1,93 @@
+# modules/database/semantic_mongo_live_db.py
+import logging
+from datetime import datetime, timezone
+import base64
+
+# Importaciones locales
+from .mongo_db import get_collection, insert_document, find_documents
+
+logger = logging.getLogger(__name__)
+COLLECTION_NAME = 'student_semantic_live_analysis'
+
+def store_student_semantic_live_result(username, text, analysis_result, lang_code='en'):
+    """
+    Guarda el resultado del análisis semántico en vivo en MongoDB.
+    """
+    try:
+        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
+        concept_graph_data = None
+        if 'concept_graph' in analysis_result and analysis_result['concept_graph'] is not None:
+            try:
+                if isinstance(analysis_result['concept_graph'], bytes):
+                    concept_graph_data = base64.b64encode(analysis_result['concept_graph']).decode('utf-8')
+                else:
+                    logger.warning("El gráfico conceptual no está en formato bytes")
+            except Exception as e:
+                logger.error(f"Error al codificar gráfico conceptual: {str(e)}")
+
+        # Crear documento para MongoDB
+        analysis_document = {
+            'username': username,
+            'timestamp': datetime.now(timezone.utc),
+            'text': text,
+            'analysis_type': 'semantic_live',
+            'key_concepts': analysis_result.get('key_concepts', []),
+            'concept_centrality': analysis_result.get('concept_centrality', {}),
+            'concept_graph': concept_graph_data,
+            'language': lang_code
+        }
+
+        # Insertar en MongoDB
+        result = insert_document(COLLECTION_NAME, analysis_document)
+        if result:
+            logger.info(f"Análisis semántico en vivo guardado para {username}")
+            return True
+        
+        logger.error("No se pudo insertar el documento en MongoDB")
+        return False
+
+    except Exception as e:
+        logger.error(f"Error al guardar el análisis semántico en vivo: {str(e)}")
+        return False
+
+def get_student_semantic_live_analysis(username, limit=10):
+    """
+    Recupera los análisis semánticos en vivo de un estudiante.
+    """
+    try:
+        query = {
+            "username": username,
+            "analysis_type": "semantic_live"
+        }
+        
+        projection = {
+            "timestamp": 1,
+            "text": 1,
+            "key_concepts": 1,
+            "concept_graph": 1,
+            "_id": 1
+        }
+        
+        results = find_documents(
+            COLLECTION_NAME,
+            query,
+            projection=projection,
+            sort=[("timestamp", -1)],
+            limit=limit
+        )
+        
+        logger.info(f"Recuperados {len(results)} análisis semánticos en vivo para {username}")
+        return results
+        
+    except Exception as e:
+        logger.error(f"Error recuperando análisis semántico en vivo: {str(e)}")
+        return []
+
+__all__ = [
+    'store_student_semantic_live_result',
+    'get_student_semantic_live_analysis'
+]