import json
import os
from openai import OpenAI
from backend.config import config
from backend.utils import write_json_file, ensure_directory

PROJECT_ROOT = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
output_dir = os.path.join(PROJECT_ROOT, "output")
ensure_directory(output_dir)

client = OpenAI(
    api_key=config.get('api_key'),
    base_url=config.get('base_url'),
)

def get_completion(prompt: str):
    response = client.chat.completions.create(
        model=config.get('model'),
        messages=[
            {"role": "system", "content": config.get('lesson_instruction')},
            {"role": "user", "content": prompt},
        ],
        response_format={"type": "json_object"},
    )

    try:
        data = json.loads(response.choices[0].message.content)
        filename = f"{output_dir}/lesson.json"
        
        if write_json_file(filename, data):
            print(f"JSON data saved to {filename}")
        else:
            print("Failed to save data")
        return data
            
    except json.JSONDecodeError as e:
        print(f"Error decoding JSON: {e}")
        return None