Create optimizer.py

optimizer.py

@@ -0,0 +1,61 @@
+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)
+    }