Update app.py

app.py

@@ -17,6 +17,10 @@ database = os.getenv("DB_NAME")
 user = os.getenv("DB_USER")
 password = os.getenv("DB_PASSWORD")
 
+# Vérifier que les variables d'environnement sont chargées
+if not all([MISTRAL_API_KEY, server, database, user, password]):
+    raise ValueError("Veuillez vérifier les variables d'environnement dans les paramètres Hugging Face.")
+    
 # 🔹 Configuration du modèle Mistral
 model_name = "mistral-large-latest"
 
@@ -30,6 +34,13 @@ llm = ChatMistralAI(
 
 # 🔹 Configuration de la base de données SQL Server
 db_url = f"mssql+pyodbc://{user}:{password}@{server}/{database}?driver=ODBC+Driver+17+for+SQL+Server"
+try:
+    engine = create_engine(db_url)
+    db = SQLDatabase.from_uri(db_url)
+    print("Connexion à la base de données réussie !")
+except Exception as e:
+    print(f"Erreur lors de la connexion à la base de données : {e}")
+
 engine = create_engine(db_url)
 db = SQLDatabase.from_uri(db_url)