Update modules/db_connection.py

modules/db_connection.py

@@ -1,17 +1,20 @@
-# modules/db_connection.py
+#db_connection.py
+import os
+from azure.identity import ClientSecretCredential
 import pymssql
-from azure.identity import DefaultAzureCredential
 
 def get_db_connection():
     server = 'aideatexdb.database.windows.net'
     database = 'textdb'
-    username = 'manuel.var.ale_aideatext.ai#EXT#@manuelvaraleaideatext.onmicrosoft.com'
-    
-    credential = DefaultAzureCredential()
+    client_id = os.getenv('AZURE_CLIENT_ID')
+    client_secret = os.getenv('AZURE_CLIENT_SECRET')
+    tenant_id = os.getenv('AZURE_TENANT_ID')
+
+    credential = ClientSecretCredential(tenant_id, client_id, client_secret)
     token = credential.get_token("https://database.windows.net/.default")
-    
+
     conn = pymssql.connect(server=server, 
                            database=database, 
-                           user=username, 
+                           user=client_id+'@'+server, 
                            password=token.token)
     return conn