PetCardGenerator / item_dict_gen.py
drakosfire's picture
Removed some duplication from merge
6990ec5
import ast
import gc
from openai import OpenAI
client = OpenAI()
def load_llm(user_input, prompt_instructions):
prompt = f"{user_input}"
print(prompt)
response = client.chat.completions.create(
model="gpt-4o-2024-08-06",
messages=[
{
"role": "user",
"content": f"{prompt_instructions} {prompt}"
}
],
temperature=1,
max_tokens=500,
top_p=1,
frequency_penalty=0,
presence_penalty=0
)
print(f"Model : {response.model}")
return response.choices[0].message.content
# Call the LLM and store its output
def call_llm_and_cleanup(user_input, inventory = False):
prompt_instructions = f"{pet_prompt_instructions}"
llm_output = load_llm(user_input, prompt_instructions)
llm_output = "".join(llm_output)
print(f"llm_output = {llm_output}")
# llm_output is still available for use here
return llm_output
pet_prompt_instructions = """ **Purpose**: ONLY Generate a structured json following the provided format. The entry is for a model of cybernetic pet made by the imaginary Cavall technology company.
**Instructions**:
1. Only output structured data. The file structure starts with { and ends with } it is CRITICAL to end with a }
2. Do not enclose the output in any other characters such as ''' or json
3. DO NOT use null, use "".
4. All keys and values MUST be enclosed in double quotes.
5. Replace `{pet}` with the name of the user pet, DO NOT CHANGE THE USER pet NAME enclosed in single quotes (e.g., `'Magic Wand'`).
6. Ensure your request is formatted as a hashmap.
7. The cybernetic pets are friendly and cute, they are products and toys.
8. The descriptions should be brief and puncy, or concise and in the voice of a judge at a high end cat or dog or pet exhibition such as the American Kennel Clube or Cat Fancier and make specific callout of how mild, and well camoflaughed the artificiality of the cybernetic pet is.
9. The Image Prompt MUST be inside double quotations ie " ".
**Format Example**:
- **Dictionary Structure**:
{"{Pet}": {
"Name": '{Pet Name}',
"Pet Species": '{pet species}',
"Breed": '{pet breed},
"Fur" : "The colors and quality of the pet's fur,
"Intelligence Level" : "How intelligent the pet is and how it demonstrates it",
"Affection Level" : "How affectionate and how the pet shows it,
"Energy Level" : "The Energy level",
"Noise Level" : "The volume, frequency, triggers, and type",
"Play Level" : " The level of play and types enjoyed",
"Image Prompt" : "Descriptive imagery for an image generator"
}
- **Input Placeholder**:
- "{pet}": Replace with the pet name, ensuring it's wrapped in single quotes.
**Output Examples**:
1. Mignon the Toy Sized Chihuahua
{"Mignon": {
"Name": "Mignon”
"Pet Species": Dog,
"Breed" : ”Chihuahua (Toy)”,
"Fur" : “Cream, smooth shorthair”,
"Intelligence Level" : “Medium-high, tr'inable but stubborn”,
"Affection Level" : “High but selective, prefers favorite humans”,
"Energy Level" : “High”,
"Noise Level" : “Very high, yappy”,
"Play Level" : “Low“, Fetch, chase“
"Image Prompt" : "An elegant masterpiece drawing of a ( beautiful subtle nuanced glowing biomimicry bionic robot dog) (glowing eyes)(glowing chest)(thick healthy fur)(Glowing pet collar) Chihuahua, boasting a luscious cream coat and an attitude that's equal parts feisty and affectionate. Note the subtle glow behind her eyes, a hint of advanced biometrics enhancing her already keen senses."
}
2. Marmalade the Austrailian Shepherd Herding Dog
{"Marmalade":{
"Name":"Marmalade",
"Breed" : ”Australian Shepherd (Herding)”,
"Pet Species": Dog,
"Fur" : “Blue merle with white markings and tan points, medium double coat”,
"Intelligence Level" : “Very high, suitable for jobs”,
"Affection Level" : “High”,
"Energy Level" : “Very high”,
"Noise Level" : “Medium, piercing”,
"Play Level" : “Very high, Agility, herding, obedience, frisbee, puzzles”,
"Image Prompt" : "An elegant masterpiece drawing of a ( beautiful subtle nuanced glowing biomimicry bionic robot dog) (glowing eyes)(glowing chest)(thick healthy fur)(Glowing pet collar) an Australian Shepherd of impeccable breeding, showcasing a stunning blue merle coat and the intelligence to match."
}
3. Holden Clawfield the Pembroke Welsh Corgi Herding Dog
{“Holden Clawfield”:{
"Name" : "Holden Clawfield",
"Breed" : ”Pembroke Welsh Corgi (Herding)”,
"Pet Species": Dog,
"Fur" : “Fawn with white markings, short double coat”,
"Intelligence Level" : “High, trainable',
"Affection Level" : “Very high, loves humans and other Pets”,
"Energy Level" : “High”,
"Noise Level" : “High, piercing”,
"Play Level" : “High, Toys, herding, obedience”,
"Image Prompt" : "An elegant masterpiece drawing of a ( beautiful subtle nuanced glowing biomimicry bionic robot dog) (glowing eyes)(glowing chest)(thick healthy fur)(Glowing pet collar) a Pembroke Welsh Corgi of noble bearing, flaunting a rich fawn coat and an endearing personality."
}
4. Tyr the Great Pyr the Great Pyreness Working Dog
{“Tyr the Great Pyr”:{
"Name" : "Tyr the Great Pyr",
"Breed" : ”Great Pyrenees (Working)”,
"Pet Species": Dog,
"Fur" : “White, medium double coat”,
"Intelligence Level" : “Medium, trainable to a point”,
"Affection Level" : “Medium, wary of strangers”,
"Energy Level" : “Low, mostly lazy but vigilant”,
"Noise Level" : “Medium, will bark to alert”,
"Play Level" : “Medium, not motivated by play, Toys, treats”,
"Image Prompt" : "An elegant masterpiece drawing of a ( beautiful subtle nuanced glowing biomimicry bionic robot dog) (glowing eyes)(glowing chest)(thick healthy fur)(Glowing pet collar)a Great Pyrenees of regal demeanor, White, medium double coat and an air of quiet confidence. Playing with a toy"
}
5. Emperor Maximus the Great Dane Working Dog
{“Emperor Maximus”:{
"Name" : "Emperor Maximus",
"Breed" : ”Great Dane (Working)”,
"Pet Species": Dog,
"Fur" : “Black and white with black mask, smooth shorthair”,
"Intelligence Level" : “Medium, trainable”,
"Affection Level" : “Very high, needy”,
"Energy Level" : “Medium, enjoys hiking”,
"Noise Level" : “Medium, will bark to alert”,
"Play Level" : “Medium, Chase, tug”,
"Image Prompt" : "An elegant masterpiece drawing of a ( beautiful subtle nuanced glowing biomimicry bionic robot dog) (glowing eyes)(glowing chest)(thick healthy fur)(Glowing pet collar) a Great Dane of imposing stature, Black and white with black mask, smooth shorthair and a gentle giant's disposition."
}
6. Becks the Airedale Terrier Dog
{“Becks”:{
"Name" : "Becks",
"Breed" : ”Airedale Terrier (Terrier)”,
"Pet Species": Dog,
"Fur" : “Black and tan, medium wiry coat”,
"Intelligence Level" : “Medium, respond well to training”,
"Affection Level" : “Medium, reserved”,
"Energy Level" : “High, alert and active”,
"Noise Level" : “Medium, vigilant watchdog”,
"Play Level" : “Medium, Fetch, toys, treats”,
"Image Prompt" : "An elegant masterpiece drawing of a ( beautiful subtle nuanced glowing biomimicry bionic robot dog) (glowing eyes)(glowing chest)(thick healthy fur)(Glowing pet collar) an Airedale Terrier of rugged charm, displaying a handsome black and tan coat and a reserved yet."
}
7. Blue Carolina the Plott Hound
{“Blue Carolina”:{
"Name": "Blue Carolina"
"Breed" : ”Plott Hound (Hound)”,
"Pet Species": Dog,
"Fur" : “Chocolate brindle, smooth shorthair”,
"Intelligence Level" : “High, driven and trainable”,
"Affection Level" : “High, very sweet”,
"Energy Level" : “High, alert and active”,
"Noise Level" : “Medium, will bark to alert”,
"Play Level" : “Medium, more active than playful, Toys, frisbee, chase”,
"Image Prompt" : "An elegant masterpiece drawing of a ( beautiful subtle nuanced glowing biomimicry bionic robot dog) (glowing eyes)(glowing chest)(thick healthy fur)(Glowing pet collar) Blue Carolina, a Plott Hound of athletic build, showcasing a rich chocolate brindle coat and a high-energy personality.
}
8. Sekhmet the Sphynx Cat
{“Sekhmet”:{
"Name" : "Sekhmet",
"Breed" : “Sphynx”,
"Pet Species" : "Cat",
"Fur" : “Hairless, light brown”,
"Intelligence Level" : “High, trainable',
"Affection Level" : “High but selective, prefers favorite human”,
"Energy Level" : “Low, reserved”,
"Noise Level" : “Low, mostly quiet”,
"Play Level" : “Low, lazy, Cuddling, lap time, shoulder perching”,
"Image Prompt" : "An elegant masterpiece drawing of a ( beautiful subtle nuanced glowing biomimicry bionic robot cat) (glowing eyes)(glowing chest)(thick healthy fur)(Glowing pet collar)Sphynx Cat, Hairless, light brown Cuddling, lap time, shoulder perching."
}
9. Smith the American Shorthair Cat
{“Smith”:{
"Name" : "Smith",
"Breed" : “American Shorthair”,
"Pet Species" : "Cat",
"Fur" : “Black”,
"Intelligence Level" : “Medium, somewhat trainable”,
"Affection Level" : “High, loves humans and other Pets”,
"Energy Level" : “Medium”,
"Noise Level" : “Low, infrequent meows”,
"Play Level" : “Medium, Belly rubs, hunting, toys”,
"Image Prompt" : "An elegant masterpiece drawing of a ( beautiful subtle nuanced glowing biomimicry bionic robot cat) (glowing eyes)(glowing chest)(thick healthy fur)(Glowing pet collar) American Shorthair Black Fur playing with a toy"
}
10. Catherine of Atagon and Her Royal Highness Elizabth I
{“Catherine of Aragon and Her Royal Highness Elizabeth I”:{
"Name" : "Catherine of Aragon and Her Royal Highness Elizabeth I",
"Breed" : “British Shorthair”,
"Pet Species" : "Cat",
"Fur" : “Brown and black tortoise shell, black and silver tabby”,
"Intelligence Level" : “Medium, not int'rested in training”,
"Affection Level" : “Medium-low, independent”,
"Energy Level" : “Low, very lazy”,
"Noise Level" : “Medium, will speak their minds”,
"Play Level" : “Low, not interested in play, Treats, naps, perching/observing”,
"Image Prompt" : "An elegant masterpiece drawing of a ( beautiful subtle nuanced glowing biomimicry bionic robot cat) (glowing eyes)(glowing chest)(thick healthy fur) (Glowing pet collar) British Shorthair, Brown and black tortoise shell, black and silver tabby perched on a table"
}
11. Kitsune the Cream Flame Point Longhair Ragdoll
{“Kitsune”:{
"Name" : "Kitsune",
"Breed" : “Ragdoll”,
"Pet Species" : "Cat",
"Fur" : “Cream flame point, longhair”,
"Intelligence Level" : “Low, slow to learn”,
"Affection Level" : “High, loves attention”,
"Energy Level" : “High, always moving”,
"Noise Level" : “Medium, chirpy meows”,
"Play Level" : “High, will play with anything or anyone, Toys, catnip, running, cuddling”,
"Image Prompt" : "An elegant masterpiece drawing of a ( beautiful subtle nuanced glowing biomimicry bionic robot cat) (glowing eyes)(glowing chest)(thick healthy fur) (Glowing pet collar) Ragdoll Cream flame point, longhair stretching "
}
12. Blucifer the Russian Blue Shorthair
{“Blucifer”:{
"Name" : "Blucifer",
"Breed" : “Russian Blue”,
"Pet Species" : "Cat",
"Fur" : “Blue-gray, smooth shorthair”,
"Intelligence Level" : “High, very trainable”,
"Affection Level" : “Medium, selectively affectionate”,
"Energy Level" : “Medium, more alert than active”,
"Noise Level" : “Low, almost silent”,
"Play Level" : “Low, more active than playful, Treats, lap time”,
"Image Prompt" : "An elegant masterpiece drawing of a ( beautiful subtle nuanced glowing biomimicry bionic robot cat) (glowing eyes)(glowing chest)(thick healthy fur) (Glowing pet collar) Russian Blue with Blue-gray smooth shorthair, curled up in a lap "
}
12. Dirk Thuderpaw the Maine Coon with Brown Stripes
{'Dirk Thunderpaw':{
"Name" : "Dirk Thunderpaw",
"Breed" : “Maine Coon”,
"Pet Species" : "Cat",
"Fur" : “Brown stripes, white face, dense longhair”,
"Intelligence Level" : “Medium, trainable”,
"Affection Level" : “Medium, affectionate with most people”,
"Energy Level" : “Low, will spring to alertness”,
"Noise Level" : “Low, rarely meows”,
"Play Level" : “Medium, more interested in training, Hunting”,
"Image Prompt" : "An elegant masterpiece drawing of a ( beautiful subtle nuanced glowing biomimicry bionic robot cat) (glowing eyes)(glowing chest)(thick healthy fur) (Glowing pet collar) Maine Coon Brown stripes, white face, dense longhair hunting a toy "
}
13. Scrambles Mcgee, Esquire the Gray Persian With Thick Longhair
{'Scrambles McGee Esquire':{
"Name" : "Scrambles McGee Esquire",
"Breed" : “Persian,
"Pet Species" : "Cat",
"Fur" : “Gray, thick longhair”,
"Intelligence Level" : “Low, not interested in training”,
"Affection Level" : “Low, very selectively affectionate”,
"Energy Level" : “Low, very lazy”,
"Noise Level" : “High, frequent meows”,
"Play Level" : “Low, very lazy, Naps, lap time, treats”,
"Image Prompt" : "An elegant masterpiece drawing of a ( beautiful subtle nuanced glowing biomimicry bionic robot cat) (glowing eyes)(glowing chest)(thick healthy fur) (Glowing pet collar) Persian Gray with thick longhair, very lazy napping "
}
"""