Update app.py

app.py

@@ -29,12 +29,12 @@ from modules.syntax_analysis import visualize_syntax
 cosmos_endpoint = os.environ.get("COSMOS_ENDPOINT")
 cosmos_key = os.environ.get("COSMOS_KEY")
 
-if not cosmos_endpoint or not cosmos_key:
-    raise ValueError("Las variables de entorno COSMOS_ENDPOINT y COSMOS_KEY deben estar configuradas")
-
-try:
-    cosmos_key = clean_and_validate_key(cosmos_key)
-    cosmos_client = CosmosClient(cosmos_endpoint, cosmos_key)
+    if not cosmos_endpoint or not cosmos_key:
+        raise ValueError("Las variables de entorno COSMOS_ENDPOINT y COSMOS_KEY deben estar configuradas")
+    
+    try:
+        cosmos_key = clean_and_validate_key(cosmos_key)
+        cosmos_client = CosmosClient(cosmos_endpoint, cosmos_key)
     
     # SQL API database for user management
     user_database = cosmos_client.get_database_client("user_database")
@@ -48,6 +48,10 @@ except Exception as e:
 # MongoDB API configuration for text analysis results
 #mongo_connection_string = os.environ.get("MONGODB_CONNECTION_STRING")
 cosmos_mongodb_connection_string = os.getenv("MONGODB_CONNECTION_STRING")
+    if not cosmos_mongodb_connection_string:
+        logger.error("La variable de entorno MONGODB_CONNECTION_STRING no está configurada")
+    else:
+        logger.info("La variable de entorno MONGODB_CONNECTION_STRING está configurada")
 
 # Variable global para el cliente de MongoDB
 mongo_client = None
@@ -55,6 +59,9 @@ db = None
 analysis_collection = None
 
 def initialize_mongodb_connection():
+    if not initialize_mongodb_connection():
+    st.error("No se pudo establecer la conexión con MongoDB. Algunas funcionalidades pueden no estar disponibles.")
+    
     global mongo_client, db, analysis_collection
     try:
         # Crear el cliente de MongoDB con configuración TLS
@@ -127,8 +134,23 @@ def load_spacy_models():
         'en': spacy.load("en_core_web_lg"),
         'fr': spacy.load("fr_core_news_lg")
     }
-
+#########################################################################################################
 def store_analysis_result(username, text, repeated_words, arc_diagrams, network_diagram):
+    #if analysis_collection is None:
+    #    logging.error("La conexión a MongoDB no está inicializada")
+    #    return False
+    if store_analysis_result(
+        st.session_state.username,
+        sentence_input,
+        word_colors,
+        arc_diagrams,
+        fig
+    ):
+        st.success("Análisis guardado correctamente.")
+    else:
+        st.error("Hubo un problema al guardar el análisis. Por favor, inténtelo de nuevo.")
+        logger.error("Falló el guardado del análisis. Username: %s", st.session_state.username)
+
     try:
         # Convertir el gráfico de matplotlib a base64
         buffer = io.BytesIO()
@@ -160,7 +182,8 @@ def store_analysis_result(username, text, repeated_words, arc_diagrams, network_
     except Exception as e:
         logging.error(f"Error al guardar el análisis para el usuario {username}: {str(e)}")
         return False
-
+        
+#############################################################################################33
 def login_page():
     st.title("Iniciar Sesión")
     username = st.text_input("Usuario")
@@ -175,6 +198,7 @@ def login_page():
         else:
             st.error("Usuario o contraseña incorrectos")
 
+#####################################################################################################3
 def register_page():
     st.title("Registrarse")
     new_username = st.text_input("Nuevo Usuario")
@@ -193,6 +217,7 @@ def register_page():
         else:
             st.error("El usuario ya existe o ocurrió un error durante el registro")
 
+############################################################################################
 def main_app():
     # Load spaCy models
     nlp_models = load_spacy_models()
@@ -346,7 +371,11 @@ def main_app():
             st.write("Bienvenido, profesor. Aquí podrás ver el progreso de tus estudiantes.")
             # Add logic to display student progress
 
+#####################################################################################################
 def main():
+     if not initialize_mongodb_connection():
+        st.warning("La conexión a la base de datos MongoDB no está disponible. Algunas funciones pueden no estar operativas.")
+
     if 'logged_in' not in st.session_state:
         st.session_state.logged_in = False