code updaed

app.py

@@ -1,7 +1,7 @@
 import os
 import gradio as gr
 from huggingface_hub import InferenceClient
-import pyodbc
+from sqlalchemy import create_engine
 
 # Clients for each model provider
 llama_client = InferenceClient(provider="sambanova", api_key=os.environ["HF_TOKEN"])
@@ -11,26 +11,21 @@ mistral_client = InferenceClient(provider="together", api_key=os.environ["HF_TOK
 # Global objects
 db_connection = None  
 
-def connect_sql_pyodbc():
+def get_sqlalchemy_connection():
     server = os.getenv("SQL_SERVER")
     database = os.getenv("SQL_DATABASE")
     username = os.getenv("SQL_USERNAME")
     password = os.getenv("SQL_PASSWORD")
 
-    conn_str = (
-        f"Driver={{ODBC Driver 17 for SQL Server}};"
-        f"Server={server};"
-        f"Database={database};"
-        f"UID={username};"
-        f"PWD={password};"
-    )
+    connection_url = f"mssql+pymssql://{username}:{password}@{server}/{database}"
 
     try:
-        conn = pyodbc.connect(conn_str)
-        print("✅ pyodbc connection successful")
+        engine = create_engine(connection_url)
+        conn = engine.connect()
+        print("✅ SQLAlchemy + pymssql connection successful")
         return conn
     except Exception as e:
-        print(f"❌ pyodbc connection failed: {e}")
+        print(f"❌ SQLAlchemy connection failed: {e}")
         return None
 
 def get_sql_connection():
@@ -45,7 +40,7 @@ def get_sql_connection():
             db_connection = None  # reset if broken
 
     # Reconnect if needed
-    db_connection = connect_sql_pyodbc()
+    db_connection = get_sqlalchemy_connection()
     return db_connection
     
 # Format chat history for Markdown display

requirements.txt

@@ -2,4 +2,4 @@ transformers>=4.41.0
 torch>=2.2.0
 gradio>=4.24.0
 huggingface_hub>=0.22.2
-pyodbc>=5.2.0
+sqlalchemy>=2.0.41