Update modules/database.py

modules/database.py

@@ -25,6 +25,7 @@ user_container = None
 mongo_client = None
 mongo_db = None
 analysis_collection = None
+chat_collection = None  # Nueva variable global
 
 #####################################################################################33
 def initialize_cosmos_sql_connection():
@@ -71,6 +72,10 @@ def initialize_mongodb_connection():
         
         mongo_db = mongo_client['aideatext_db']
         analysis_collection = mongo_db['text_analysis']
+        chat_collection = mongo_db['chat_history']  # Inicializar la nueva colección
+
+        # Verificar la conexión
+        mongo_client.admin.command('ping')
         
         logger.info("Conexión a Cosmos DB MongoDB API exitosa")
         return True
@@ -238,6 +243,7 @@ def store_chat_history(username, messages):
         }
         result = chat_collection.insert_one(chat_document)
         logger.info(f"Chat history saved with ID: {result.inserted_id} for user: {username}")
+        logger.debug(f"Chat content: {messages}")
         return True
     except Exception as e:
         logger.error(f"Error saving chat history for user {username}: {str(e)}")
@@ -245,24 +251,24 @@ def store_chat_history(username, messages):
 
 #######################################################################################################
 def get_student_data(username):
-    if analysis_collection is None:
+    if analysis_collection is None or chat_collection is None:
         logger.error("La conexión a MongoDB no está inicializada")
         return None
 
+    formatted_data = {
+        "username": username,
+        "entries": [],
+        "entries_count": 0,
+        "word_count": {},
+        "semantic_analyses": [],
+        "discourse_analyses": [],
+        "chat_history": []
+    }
+
     try:
-        logger.info(f"Buscando datos para el usuario: {username}")
+        logger.info(f"Buscando datos de análisis para el usuario: {username}")
         cursor = analysis_collection.find({"username": username})
         
-        formatted_data = {
-            "username": username,
-            "entries": [],
-            "entries_count": 0,
-            "word_count": {},
-            "semantic_analyses": [],
-            "discourse_analyses": [],
-            "chat_history": []
-        }
-        
         for entry in cursor:
             formatted_entry = {
                 "timestamp": entry.get("timestamp", datetime.utcnow()),
@@ -297,14 +303,24 @@ def get_student_data(username):
         
         for entry in formatted_data["entries"]:
             entry["timestamp"] = entry["timestamp"].isoformat()
+    
+    except Exception as e:
+        logger.error(f"Error al obtener datos de análisis del estudiante {username}: {str(e)}")
+    
+    try:
+        logger.info(f"Buscando historial de chat para el usuario: {username}")
+        chat_cursor = chat_collection.find({"username": username})
+        for chat in chat_cursor:
+            formatted_chat = {
+                "timestamp": chat["timestamp"].isoformat(),
+                "messages": chat["messages"]
+            }
+            formatted_data["chat_history"].append(formatted_chat)
         
-        # Obtener el historial del chat
-        chat_cursor = mongo_db.chat_history.find({"username": username})
-        formatted_data["chat_history"] = list(chat_cursor)
-        
-        logger.info(f"Datos formateados para {username}: {formatted_data}")
-        return formatted_data
-
+        formatted_data["chat_history"].sort(key=lambda x: x["timestamp"], reverse=True)
+    
     except Exception as e:
-        logger.error(f"Error al obtener datos del estudiante {username}: {str(e)}")
-        return None
+        logger.error(f"Error al obtener historial de chat del estudiante {username}: {str(e)}")
+
+    logger.info(f"Datos formateados para {username}: {formatted_data}")
+    return formatted_data