Update modules/database/current_situation_mongo_db.py

modules/database/current_situation_mongo_db.py

@@ -6,6 +6,8 @@ from .mongo_db import get_collection
 logger = logging.getLogger(__name__)
 COLLECTION_NAME = 'student_current_situation'
 
+# En modules/database/current_situation_mongo_db.py
+
 def store_current_situation_result(username, text, metrics, feedback):
     """
     Guarda los resultados del análisis de situación actual.
@@ -21,54 +23,32 @@ def store_current_situation_result(username, text, metrics, feedback):
             logger.error("No se pudo obtener la colección")
             return False
 
-        # Formatear métricas para MongoDB
-        formatted_metrics = {
-            'vocabulary': {
-                'score': metrics['vocabulary']['normalized_score'],
-                'details': metrics['vocabulary']['details']
-            },
-            'structure': {
-                'score': metrics['structure']['normalized_score'],
-                'details': metrics['structure']['details']
-            },
-            'cohesion': {
-                'score': metrics['cohesion']['normalized_score'],
-                'details': metrics['cohesion']['details']
-            },
-            'clarity': {
-                'score': metrics['clarity']['normalized_score'],
-                'details': metrics['clarity']['details']
-            }
-        }
-
         # Crear documento
         document = {
             'username': username,
             'timestamp': datetime.now(timezone.utc).isoformat(),
             'text': text,
-            'metrics': formatted_metrics,  # Usar las métricas formateadas
-            'feedback': feedback,
+            'metrics': metrics,
+            'feedback': feedback or {},
             'analysis_type': 'current_situation'
         }
 
-        # Insertar documento
+        # Insertar documento y verificar
         result = collection.insert_one(document)
         if result.inserted_id:
             logger.info(f"""
-                Análisis guardado exitosamente:
+                Análisis de situación actual guardado:
                 - Usuario: {username}
                 - ID: {result.inserted_id}
-                - Longitud del texto: {len(text)}
-                - Métricas: {formatted_metrics}
+                - Longitud texto: {len(text)}
             """)
             
             # Verificar almacenamiento
-            if verify_storage(username):
-                logger.info("Verificación de almacenamiento exitosa")
-                return True
-            else:
+            storage_verified = verify_storage(username)
+            if not storage_verified:
                 logger.warning("Verificación de almacenamiento falló")
-                return False
+            
+            return True
 
         logger.error("No se pudo insertar el documento")
         return False