import json
import openai
import os

from .prompt import SYSTEM_PROMPT

# Load your API key from an environment variable or secret management service

def ai_language_detector(input_text:str) -> str | None:
    try:
        api_key = os.getenv("OPENAI_API_KEY")
        client = openai.OpenAI(api_key=api_key)

        completion = client.chat.completions.create(
            response_format={"type": "json_object"},
            model="gpt-4-0125-preview",
            messages=[
                {"role": "system", "content": SYSTEM_PROMPT},
                {"role": "user", "content": input_text},
            ],
        )

        
        content = json.loads(completion.choices[0].message.content or "{}")
        if "language" not in content:
            return None
    
        return content["language"]
    

    except Exception as e:
        print(f"An error occurred: {e}")
        return None


if __name__ == "__main__":
    # Example usage:
    print(ai_language_detector("hola como estas?"))