import requests
import os
import logging

def call_azure_function(operation, data):
    function_url = os.getenv('AZURE_FUNCTION_URL')
    api_key = os.getenv('AZURE_FUNCTION_API_KEY')

    if not function_url or not api_key:
        logging.error("AZURE_FUNCTION_URL or AZURE_FUNCTION_API_KEY is not set")
        raise ValueError("Azure Function URL or API Key is not configured")

    logging.info(f"Calling Azure Function: {function_url}")
    logging.info(f"Operation: {operation}")
    logging.info(f"Data: {data}")

    headers = {
        'Content-Type': 'application/json',
        'x-functions-key': api_key
    }

    try:
        response = requests.post(function_url, json={'operation': operation, 'data': data}, headers=headers)
        logging.info(f"Response status code: {response.status_code}")
        logging.info(f"Response content: {response.text}")

        if response.status_code == 200:
            return response.json()
        elif response.status_code == 401:
            logging.error("Authentication failed. Check if the API key is correct.")
            raise Exception("Authentication failed. Check if the API key is correct.")
        else:
            logging.error(f"Error calling Azure Function: {response.text}")
            raise Exception(f"Error calling Azure Function: {response.text}")
    except requests.exceptions.RequestException as e:
        logging.error(f"Request exception: {str(e)}")
        raise