parent
15186846be
commit
a4df2954d4
3 changed files with 26 additions and 73 deletions
@ -1,55 +0,0 @@ |
||||
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() |
||||
Loading…
Reference in new issue