Update modules/database.py

modules/database.py

@@ -23,6 +23,7 @@ mongo_client = None
 mongo_db = None
 analysis_collection = None
 
+#####################################################################################33
 def initialize_cosmos_sql_connection():
     global cosmos_client, user_database, user_container
     try:
@@ -42,6 +43,7 @@ def initialize_cosmos_sql_connection():
         logger.error(f"Error al conectar con Cosmos DB SQL API: {str(e)}")
         return False
 
+############################################################################################3        
 def initialize_mongodb_connection():
     global mongo_client, mongo_db, analysis_collection
     try:
@@ -69,8 +71,8 @@ def initialize_mongodb_connection():
         logger.error(f"Error al conectar con Cosmos DB MongoDB API: {str(e)}", exc_info=True)
         return False
 
+#######################################################################################################
 # Funciones para Cosmos DB SQL API (manejo de usuarios)
-
 def get_user(username):
     try:
         query = f"SELECT * FROM c WHERE c.id = '{username}'"
@@ -111,21 +113,22 @@ def get_student_data(username):
         
         # Formatear los datos
         formatted_data = {
-            "username": username,
-            "entries": [],
-            "entries_count": 0,
-            "word_count": {},
-            "arc_diagrams": [],
-            "network_diagrams": []
-        }
-        
-        for entry in cursor:
-            logger.debug(f"Procesando entrada: {entry}")
-            formatted_data["entries"].append({
-                "timestamp": entry["timestamp"],
-                "text": entry["text"]
-            })
-            formatted_data["entries_count"] += 1
+        "username": username,
+        "entries": [
+            {
+                "timestamp": entry["timestamp"].isoformat(),
+                "text": entry["text"],
+                "word_count": entry.get("word_count", {}),
+                "arc_diagrams": entry.get("arc_diagrams", []),
+                "network_diagram": entry.get("network_diagram", "")
+            }
+            for entry in cursor
+        ],
+        "entries_count": count,
+        "word_count": aggregated_word_count
+    }
+
+    return formatted_data
             
             # Agregar conteo de palabras
             for category, count in entry.get("word_count", {}).items():