Update modules/database/semantic_mongo_live_db.py

modules/database/semantic_mongo_live_db.py

@@ -23,96 +23,83 @@ COLLECTION_NAME = 'student_semantic_live_analysis'
 def store_student_semantic_live_result(username, text, analysis_result, lang_code='en'):
     """
     Versión corregida con:
-    - Mejor verificación de colección
-    - Logs más detallados
-    - Validación de inserción mejorada
+    - Verificación correcta de colección
+    - Manejo de proyección alternativo
+    - Mejor manejo de errores
     """
     try:
-        # 1. Verificación de colección
+        # 1. Obtener colección con verificación correcta
         collection = get_collection(COLLECTION_NAME)
-        if not collection:
-            logger.error(f"No se pudo acceder a la colección {COLLECTION_NAME}")
+        if collection is None:  # Cambiado de 'if not collection'
+            logger.error(f"No se pudo obtener la colección {COLLECTION_NAME}")
             return False
 
-        # 2. Validación de parámetros extendida
+        # 2. Validación de parámetros
         if not all([username, text, analysis_result]):
             logger.error("Parámetros incompletos para guardar análisis")
             return False
 
-        # 3. Preparación del documento con más logging
+        # 3. Preparar documento
         analysis_document = {
-            'username': str(username),
+            'username': username,
             'timestamp': datetime.now(timezone.utc),
-            'text': str(text)[:50000],  # Limitar tamaño
+            'text': text[:50000],
             'analysis_type': 'semantic_live',
-            'language': str(lang_code),
+            'language': lang_code,
             'key_concepts': analysis_result.get('key_concepts', [])[:50],
-            'concept_centrality': analysis_result.get('concept_centrality', {}),
-            'metadata': {
-                'version': '1.1',
-                'source': 'live_interface_v2'
-            }
+            'concept_centrality': analysis_result.get('concept_centrality', {})
         }
 
-        # 4. Manejo del gráfico con más controles
-        if 'concept_graph' in analysis_result:
+        # 4. Manejo del gráfico
+        if 'concept_graph' in analysis_result and analysis_result['concept_graph']:
             try:
-                graph_data = analysis_result['concept_graph']
-                if isinstance(graph_data, bytes):
-                    analysis_document['concept_graph'] = base64.b64encode(graph_data).decode('utf-8')
-                elif isinstance(graph_data, str) and graph_data:
-                    analysis_document['concept_graph'] = graph_data
+                if isinstance(analysis_result['concept_graph'], bytes):
+                    analysis_document['concept_graph'] = base64.b64encode(
+                        analysis_result['concept_graph']).decode('utf-8')
             except Exception as e:
-                logger.warning(f"Error procesando gráfico: {str(e)}")
+                logger.error(f"Error procesando gráfico: {str(e)}")
 
-        # 5. Inserción con verificación explícita
+        # 5. Insertar documento
         try:
             result = collection.insert_one(analysis_document)
             if result.inserted_id:
                 logger.info(f"Análisis guardado. ID: {result.inserted_id}")
                 return True
-            
             logger.error("Inserción fallida - Sin ID devuelto")
             return False
-            
         except PyMongoError as e:
             logger.error(f"Error de MongoDB: {str(e)}")
             return False
 
     except Exception as e:
-        logger.error(f"Error crítico: {str(e)}", exc_info=True)
+        logger.error(f"Error inesperado: {str(e)}", exc_info=True)
         return False
 
 ##########################################
 ##########################################
 def get_student_semantic_live_analysis(username, limit=10):
     """
-    Recupera los análisis semánticos en vivo de un estudiante.
-    Versión corregida con consulta adecuada.
+    Versión corregida sin usar projection
     """
     try:
-        query = {
-            "username": username,
-            "analysis_type": "semantic_live"  # Corregido a semantic_live
-        }
+        collection = get_collection(COLLECTION_NAME)
+        if collection is None:
+            logger.error("No se pudo obtener la colección")
+            return []
+
+        query = {"username": username, "analysis_type": "semantic_live"}
         
-        projection = {
+        # Versión alternativa sin projection
+        cursor = collection.find(query, {
             "timestamp": 1,
-            "text": {"$substr": ["$text", 0, 200]},  # Solo primeros 200 chars
+            "text": 1,
             "key_concepts": 1,
             "concept_graph": 1,
             "_id": 1
-        }
-        
-        results = find_documents(
-            COLLECTION_NAME,
-            query,
-            projection=projection,
-            sort=[("timestamp", -1)],
-            limit=limit
-        )
+        }).sort("timestamp", -1).limit(limit)
         
-        logger.info(f"Recuperados {len(results)} análisis en vivo para {username}")
+        results = list(cursor)
+        logger.info(f"Recuperados {len(results)} análisis para {username}")
         return results
         
     except PyMongoError as e:
@@ -122,6 +109,8 @@ def get_student_semantic_live_analysis(username, limit=10):
         logger.error(f"Error inesperado: {str(e)}")
         return []
 
+#######################################################
+#######################################################
 def update_student_semantic_live_analysis(analysis_id, update_data):
     """Actualiza un análisis existente con manejo de errores"""
     try:
@@ -132,6 +121,8 @@ def update_student_semantic_live_analysis(analysis_id, update_data):
         logger.error(f"Error al actualizar: {str(e)}")
         return False
 
+#######################################################
+#######################################################
 def delete_student_semantic_live_analysis(analysis_id):
     """Elimina un análisis con manejo de errores"""
     try:
@@ -141,6 +132,8 @@ def delete_student_semantic_live_analysis(analysis_id):
         logger.error(f"Error al eliminar: {str(e)}")
         return False
 
+#######################################################
+#######################################################
 def get_student_semantic_live_data(username):
     """
     Obtiene todos los análisis semánticos en vivo de un estudiante.
@@ -176,6 +169,10 @@ def get_student_semantic_live_data(username):
             'error': str(e)
         }
 
+
+#######################################################
+#######################################################
+
 __all__ = [
     'store_student_semantic_live_result',
     'get_student_semantic_live_analysis',