fix ad_gen example

pull/380/head
evelynmitchell 1 year ago
parent da024708f6
commit 8f2e62d5ae

@ -42,7 +42,7 @@ class ProductAdConceptGenerator:
"serene", "serene",
"lasers,lightning", "lasers,lightning",
] ]
self.contexts = [ self.style = [
"in an everyday setting", "in an everyday setting",
"in a rave setting", "in a rave setting",
"in an abstract environment", "in an abstract environment",
@ -63,10 +63,21 @@ class ProductAdConceptGenerator:
self.contexts = [ self.contexts = [
"high realism product ad (extremely creative)" "high realism product ad (extremely creative)"
] ]
self.social_media_platforms = [
"Facebook",
"Instagram",
"Twitter",
"LinkedIn",
"Snapchat",
"TikTok",
"Pinterest",
"YouTube",
]
def generate_concept(self): def generate_concept(self):
theme = random.choice(self.themes) theme = random.choice(self.themes)
context = random.choice(self.contexts) context = random.choice(self.contexts)
style = random.choice(self.style)
return ( return (
f"{theme} inside a {style} {self.product_name}, {context}" f"{theme} inside a {style} {self.product_name}, {context}"
) )
@ -94,8 +105,7 @@ ad_copy_prompt = (
ad_copy = ad_copy_agent.run(task=ad_copy_prompt) ad_copy = ad_copy_agent.run(task=ad_copy_prompt)
# Output the results # Output the results
print("Creative Concept:", concept_result) print("Creative Concept:", concept_concept)
print("Design Ideas:", design_result)
print("Ad Copy:", copywriting_result) print("Ad Copy:", copywriting_result)
print( print(
"Image Path:", "Image Path:",

Loading…
Cancel
Save