Update modules/database.py

modules/database.py

@@ -127,72 +127,6 @@ def get_user(username):
         print(f"Error al obtener usuario {username}: {str(e)}")
         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:
@@ -307,4 +241,70 @@ def store_chat_history(username, messages):
         return True
     except Exception as e:
         logger.error(f"Error saving chat history for user {username}: {str(e)}")
-        return False
+        return False
+
+#######################################################################################################
+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