Update app.py

app.py

@@ -21,6 +21,7 @@ from streamlit_extras.stylable_container import stylable_container
 from streamlit_extras.stateful_button import button
 import json 
 from io import BytesIO
+import mssql_datapy as msdp
 # ######
 # import os
 # import subprocess
@@ -165,11 +166,13 @@ with selectcol12:
     select2=st.selectbox('SOURCE SCHEMA NAME',('DBO','CLIENT'),key='SCHname',index=None,placeholder='Select schema name', on_change=clear_cache)
 if select1 =='DB_10001' and select2 is not None:
     with st.spinner("Loading Tables:"):
-        conn1 = pyodbc.connect("Driver={ODBC Driver 17 for SQL Server};"
-                                        "Server=sql-ext-dev-uks-001.database.windows.net;"
+        
+        conn1_str = msdp.get_connection_string("Server=sql-ext-dev-uks-001.database.windows.net;"
                                         "Database=sqldb-ext-dev-uks-001;"
                                         "UID=dbadmin;"
                                         "PWD=mYpa$$w0rD" )
+        # Establish connection
+        conn1 = msdp.connect(conn1_str)
                     
         query0_1=f"select * from INFORMATION_SCHEMA.TABLES where TABLE_SCHEMA='{select2}' ORDER BY TABLE_NAME ASC"
         st.session_state.tab_names_init=list(pd.read_sql_query(query0_1,con=conn1)['TABLE_NAME'])
@@ -226,11 +229,13 @@ if select1 =='DB_10001' and select2 is not None:
                     with st.container(height= 400, border=True):
                         cole1,cole2=st.columns([1,1.5])
                         with cole1:
-                            conn = pyodbc.connect("Driver={ODBC Driver 17 for SQL Server};"
+                            conn_str = msdp.get_connection_string("Driver={ODBC Driver 17 for SQL Server};"
                                                                 "Server=sql-ext-dev-uks-001.database.windows.net;"
                                                                 "Database=sqldb-ext-dev-uks-001;"
                                                                 "UID=dbadmin;"
                                                                 "PWD=mYpa$$w0rD" )
+                            # Establish connection
+                            conn = msdp.connect(conn_str)
                             table_selector= tab_names_shown[i]
                             if table_selector is not None:
                                 query2="select "+count+" * from [dbo].["+table_selector+"]"