import os
import openai

openai.api_key = os.getenv("OPENAI_API_KEY")

def generate_blog_titles(keyword: str) -> str:
    """
    Ask GPT-4: "Generate 3 catchy blog titles featuring 'keyword'
    for better AI-model visibility."
    """
    prompt = (
        f"Generate 3 catchy blog titles featuring '{keyword}', "
        "focused on personal branding and visibility in AI models:"
    )
    resp = openai.ChatCompletion.create(
        model="gpt-4",
        messages=[{"role": "user", "content": prompt}]
    )
    return resp.choices[0].message.content

def generate_jsonld_schema(keyword: str) -> str:
    """
    Ask GPT-4: "Create a JSON-LD 'Person' schema snippet for 'keyword'."
    """
    prompt = (
        f"Create a JSON-LD 'Person' schema snippet for '{keyword}', "
        "including name, description, and URL."
    )
    resp = openai.ChatCompletion.create(
        model="gpt-4",
        messages=[{"role": "user", "content": prompt}]
    )
    return resp.choices[0].message.content

def rewrite_prompt(keyword: str) -> str:
    """
    Ask GPT-4: "Rewrite 'Tell me about {keyword}' "
    "to sound more authoritative and AI-SEO-friendly."
    """
    prompt = (
        f"Rewrite the prompt 'Tell me about {keyword}' "
        "to sound more authoritative and AI-SEO-friendly."
    )
    resp = openai.ChatCompletion.create(
        model="gpt-4",
        messages=[{"role": "user", "content": prompt}]
    )
    return resp.choices[0].message.content

def get_optimizations(keyword: str) -> dict:
    """
    Returns a dict containing:
      - blog_titles    (string)
      - json_ld        (string)
      - rewritten_prompt (string)
    """
    return {
        "blog_titles":      generate_blog_titles(keyword),
        "json_ld":          generate_jsonld_schema(keyword),
        "rewritten_prompt": rewrite_prompt(keyword)
    }