Update modules/database.py

modules/database.py

@@ -128,7 +128,72 @@ def get_user(username):
         return None
 
 #######################################################################################################
+def get_student_data(username):
+    if analysis_collection is None:
+        logger.error("La conexión a MongoDB no está inicializada")
+        return None
+
+    try:
+        logger.info(f"Buscando datos 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["timestamp"],
+                "text": entry["text"],
+                "analysis_type": entry.get("analysis_type", "morphosyntax")
+            }
+            
+            if formatted_entry["analysis_type"] == "morphosyntax":
+                formatted_entry.update({
+                    "word_count": entry.get("word_count", {}),
+                    "arc_diagrams": entry.get("arc_diagrams", [])
+                })
+                for category, count in formatted_entry["word_count"].items():
+                    formatted_data["word_count"][category] = formatted_data["word_count"].get(category, 0) + count
+            
+            elif formatted_entry["analysis_type"] == "semantic":
+                formatted_entry["network_diagram"] = entry.get("network_diagram", "")
+                formatted_data["semantic_analyses"].append(formatted_entry)
+            
+            elif formatted_entry["analysis_type"] == "discourse":
+                formatted_entry.update({
+                    "text1": entry.get("text1", ""),
+                    "text2": entry.get("text2", ""),
+                    "combined_graph": entry.get("combined_graph", "")
+                })
+                formatted_data["discourse_analyses"].append(formatted_entry)
+            
+            formatted_data["entries"].append(formatted_entry)
+        
+        formatted_data["entries_count"] = len(formatted_data["entries"])
+        formatted_data["entries"].sort(key=lambda x: x["timestamp"], reverse=True)
+        
+        for entry in formatted_data["entries"]:
+            entry["timestamp"] = entry["timestamp"].isoformat()
+        
+        # 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
 
+    except Exception as e:
+        logger.error(f"Error al obtener datos del estudiante {username}: {str(e)}")
+        return None
+
+#######################################################################################################
 def store_morphosyntax_result(username, text, repeated_words, arc_diagrams):
     if analysis_collection is None:
         logger.error("La conexión a MongoDB no está inicializada")