Update app.py

app.py

@@ -11,24 +11,46 @@ import numpy as np
 from dotenv import load_dotenv
 load_dotenv()
 
+from modules.auth import clean_and_validate_key, register_user, authenticate_user, get_user_role
+from modules.morpho_analysis import get_repeated_words_colors, highlight_repeated_words, POS_COLORS, POS_TRANSLATIONS
+from modules.syntax_analysis import visualize_syntax
+
 # Azure Cosmos DB configuration
-cosmos_endpoint = "COSMOS_ENDPOINT"
-cosmos_key = "COSMOS_KEY"
-cosmos_client = CosmosClient(cosmos_endpoint, cosmos_key)
+cosmos_endpoint = os.environ.get("COSMOS_ENDPOINT")
+cosmos_key = os.environ.get("COSMOS_KEY")
 
-# SQL API database for user management
-user_database = cosmos_client.get_database_client("user_database")
-user_container = user_database.get_container_client("users")
+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")
+    user_container = user_database.get_container_client("users")
+    
+    print("Conexión a Cosmos DB SQL API exitosa")
+except Exception as e:
+    print(f"Error al conectar con Cosmos DB SQL API: {str(e)}")
+    raise
 
 # MongoDB API configuration for text analysis results
-mongo_connection_string = "MONGODB_CONNECTION_STRING"
-mongo_client = MongoClient(mongo_connection_string)
-mongo_db = mongo_client['aideatext_db']
-analysis_collection = mongo_db['text_analysis']
+mongo_connection_string = os.environ.get("MONGODB_CONNECTION_STRING")
+if not mongo_connection_string:
+    raise ValueError("La variable de entorno MONGODB_CONNECTION_STRING debe estar configurada")
 
-from modules.auth import register_user, authenticate_user, get_user_role
-from modules.morpho_analysis import get_repeated_words_colors, highlight_repeated_words, POS_COLORS, POS_TRANSLATIONS
-from modules.syntax_analysis import visualize_syntax
+try:
+    mongo_client = MongoClient(mongo_connection_string)
+    mongo_db = mongo_client['aideatext_db']
+    analysis_collection = mongo_db['text_analysis']
+    
+    # Prueba de conexión
+    mongo_client.server_info()
+    print("Conexión a MongoDB API exitosa")
+except Exception as e:
+    print(f"Error al conectar con MongoDB API: {str(e)}")
+    raise
 
 # Configure the page to use the full width
 st.set_page_config(