Update app.py

app.py

@@ -8,9 +8,14 @@ from dotenv import load_dotenv
 from sqlalchemy import create_engine, text
 import pandas as pd
 from typing import Dict
+import mysql.connector
 
-# Vérifier les drivers ODBC installés
-os.system("python check_odbc.py")
+import os
+from dotenv import load_dotenv
+from langchain_mistralai.chat_models import ChatMistralAI
+from langchain_community.utilities import SQLDatabase
+from sqlalchemy import create_engine
+import mysql.connector
 
 # 🔹 Charger les variables d'environnement
 load_dotenv()
@@ -35,8 +40,10 @@ llm = ChatMistralAI(
     verbose=True
 )
 
-# 🔹 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"
+# 🔹 Configuration de la base de données MySQL
+# Utiliser mysql-connector-python avec SQLAlchemy
+db_url = f"mysql+mysqlconnector://{user}:{password}@{server}/{database}"
+
 try:
     engine = create_engine(db_url)
     db = SQLDatabase.from_uri(db_url)
@@ -44,8 +51,17 @@ try:
 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)
+# Si tu veux tester une connexion directe sans SQLAlchemy
+try:
+    conn = mysql.connector.connect(
+        host=server,
+        user=user,
+        password=password,
+        database=database
+    )
+    print("Connexion réussie à la base de données MySQL !")
+except mysql.connector.Error as err:
+    print(f"Erreur de connexion : {err}")
 
 # 🔹 Définition du graphe LangGraph
 class QueryState(TypedDict):