Update modules/db_connection.py

modules/db_connection.py

@@ -1,20 +1,23 @@
-#db_connection.py
+import requests
 import os
-from azure.identity import ClientSecretCredential
-import pymssql
 
-def get_db_connection():
-    server = 'aideatexdb.database.windows.net'
-    database = 'textdb'
-    client_id = os.getenv('AZURE_CLIENT_ID')
-    client_secret = os.getenv('AZURE_CLIENT_SECRET')
-    tenant_id = os.getenv('AZURE_TENANT_ID')
+def call_azure_function(operation, data):
+    function_url = os.getenv('AZURE_FUNCTION_URL')
+    api_key = os.getenv('AZURE_FUNCTION_API_KEY')  # Si está utilizando una clave de API
 
-    credential = ClientSecretCredential(tenant_id, client_id, client_secret)
-    token = credential.get_token("https://database.windows.net/.default")
+    headers = {
+        'Content-Type': 'application/json',
+        'x-functions-key': api_key  # Si está utilizando una clave de API
+    }
 
-    conn = pymssql.connect(server=server, 
-                           database=database, 
-                           user=client_id+'@'+server, 
-                           password=token.token)
-    return conn
+    payload = {
+        'operation': operation,
+        'data': data
+    }
+
+    response = requests.post(function_url, json=payload, headers=headers)
+    
+    if response.status_code == 200:
+        return response.json()
+    else:
+        raise Exception(f"Error calling Azure Function: {response.text}")