Update strategyGen.py

strategyGen.py

@@ -49,6 +49,21 @@ def ideagen(content_type , brand_name, product_list, core_essence, brand_target,
         messages=messages_base
     )
 
+    completed_text = response.choices[0].message.content
+    total_prompt_tokens_used += response.usage.prompt_tokens
+    total_completion_tokens_used += response.usage.completion_tokens
+
+    price = total_prompt_tokens_used*0.005/1000 + total_completion_tokens_used*0.015/1000
+    price_str = "price:" + str(price) + "$"
+
+    #full_text = completed_text + "\n" + price
+    full_text = completed_text
+
+    #print(full_text)
+
+    #return response.choices[0].message.content
+    return full_text
+
 def ideagen_adv(content_type , brand_name, product_list, core_essence, brand_target, target_customer, strength_focus, weakness_focus, opportunities_focus, threats_focus, additional):
     total_prompt_tokens_used = 0
     total_completion_tokens_used = 0