import json
import logging
from openai import OpenAI
from foodie_config import OPENAI_API_KEY, AUTHORS, LIGHT_TASK_MODEL
from datetime import datetime, timezone

logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")

client = OpenAI(api_key=OPENAI_API_KEY)

def generate_background(author):
    # Include the DOB in the prompt to contextualize the timeline
    dob = author.get('dob', '1980-01-01')  # Fallback DOB if not provided
    current_year = datetime.now().year
    birth_year = int(dob.split('-')[0])
    age = current_year - birth_year  # Calculate approximate age

    prompt = (
        f"Generate a fictional background for a food writer named {author['username']} with the persona '{author['persona']}'. "
        f"They were born on {dob} and are currently {age} years old. Use this to create a realistic timeline for their life events. "
        "Include: hometown, cultural influences, an early food memory (from their childhood, appropriate for their birth year), "
        "and how they became a food writer (considering their age and career timeline). "
        f"Match the tone of their bio: '{author['bio']}'. Keep it concise (100-150 words). "
        "Return as JSON: {'username': '...', 'hometown': '...', 'cultural_influences': '...', 'early_memory': '...', 'career_path': '...'}"
    )
    try:
        response = client.chat.completions.create(
            model=LIGHT_TASK_MODEL,
            messages=[
                {"role": "system", "content": prompt},
                {"role": "user", "content": f"Generate background for {author['username']}."}
            ],
            max_tokens=200
        )
        raw_result = response.choices[0].message.content.strip()
        cleaned_result = raw_result.replace('```json', '').replace('```', '').strip()
        return json.loads(cleaned_result)
    except Exception as e:
        logging.error(f"Failed to generate background for {author['username']}: {e}")
        return None

def main():
    backgrounds = []
    for author in AUTHORS:
        background = generate_background(author)
        if background:
            backgrounds.append(background)
            logging.info(f"Generated background for {author['username']}")
    
    with open('/home/shane/foodie_automator/author_backgrounds.json', 'w') as f:
        json.dump(backgrounds, f, indent=2)
    logging.info("Saved backgrounds to author_backgrounds.json")

if __name__ == "__main__":
    main()