Update function.py

function.py

@@ -116,7 +116,24 @@ def ai_generated_content(topic, audience, benefit, date_time):
 
 
 
+    # Define the template for the benefits section with the specific format
+    benefits_template = (
+        "List 6 benefits of {topic} for {audience}. Each benefit should be written in the format: "
+        "‘[Mini-Headline]: [Detailed Benefit Description].’ The mini-headline should highlight a key feature like a number or "
+        "a unique aspect of the {topic}, and the detailed benefit should explain the value or outcome in a single line."
+    )
+
+    # Format the benefits template using PromptTemplate
+    benefits_prompt = PromptTemplate(template=benefits_template).format(
+        topic=topic,
+        audience=audience
+    )
+
+    # Generate the benefits using the LLM
+    benefits_response = llm.invoke(benefits_prompt)
 
+    # Split the benefits into individual lines
+    benefits = benefits_response.content.split('\n')
 
 
 
@@ -143,7 +160,8 @@ def ai_generated_content(topic, audience, benefit, date_time):
             "subheadline":subheadline_response.content,
             "price_statement": price_statement_response.content,
             "cta_button": cta_button_response.content,
-            "urgency_statement": urgency_statement_response.content
+            "urgency_statement": urgency_statement_response.content,
+            "benefits":benefits
            }