Update app.py

app.py

@@ -1,10 +1,9 @@
 import streamlit as st
-import pyodbc
+from sqlalchemy import create_engine, text
 
 # Fixed SQL Server details
 SERVER = "indsr-foundry.database.windows.net"
 DATABASE = "foundry_db_monarch"
-DRIVER = "ODBC Driver 17 for SQL Server"  # Use the correct driver name
 
 st.set_page_config(page_title="SQL Server Connection Test", layout="centered")
 
@@ -15,18 +14,16 @@ username = st.text_input("Database Username", placeholder="Enter your username")
 password = st.text_input("Database Password", type="password", placeholder="Enter your password")
 
 def test_connection(username, password):
-    """Attempts to connect to the SQL Server and returns success or failure message."""
+    """Attempts to connect to the SQL Server using SQLAlchemy + pytds."""
     try:
-        conn_str = (
-            f"DRIVER={{{DRIVER}}};"
-            f"SERVER={SERVER};"
-            f"DATABASE={DATABASE};"
-            f"UID={username};"
-            f"PWD={password};"
-        )
-        conn = pyodbc.connect(conn_str, timeout=5)
-        conn.close()
+        conn_str = f"mssql+pytds://{username}:{password}@{SERVER}/{DATABASE}"
+        engine = create_engine(conn_str, pool_pre_ping=True)
+        
+        with engine.connect() as connection:
+            result = connection.execute(text("SELECT 1"))
+        
         return "✅ Connection successful! You are connected to SQL Server."
+    
     except Exception as e:
         return f"❌ Connection failed: {str(e)}"