Update ideaGen.py
Browse files- ideaGen.py +2 -2
ideaGen.py
CHANGED
|
@@ -60,7 +60,7 @@ def ideagen(content_type, brand_name, product_name, target_audience, core_essenc
|
|
| 60 |
|
| 61 |
#for _ in range(loop):
|
| 62 |
response = client.chat.completions.create(
|
| 63 |
-
model='gpt-
|
| 64 |
max_tokens=2000,
|
| 65 |
temperature=0.7,
|
| 66 |
messages=messages_base
|
|
@@ -68,7 +68,7 @@ def ideagen(content_type, brand_name, product_name, target_audience, core_essenc
|
|
| 68 |
completed_text = response.choices[0].message.content
|
| 69 |
total_prompt_tokens_used += response.usage.prompt_tokens
|
| 70 |
total_completion_tokens_used += response.usage.completion_tokens
|
| 71 |
-
price = total_prompt_tokens_used*0.
|
| 72 |
|
| 73 |
full_text += completed_text + "\n\n----------\n\n"
|
| 74 |
total_price += price
|
|
|
|
| 60 |
|
| 61 |
#for _ in range(loop):
|
| 62 |
response = client.chat.completions.create(
|
| 63 |
+
model='gpt-4o',
|
| 64 |
max_tokens=2000,
|
| 65 |
temperature=0.7,
|
| 66 |
messages=messages_base
|
|
|
|
| 68 |
completed_text = response.choices[0].message.content
|
| 69 |
total_prompt_tokens_used += response.usage.prompt_tokens
|
| 70 |
total_completion_tokens_used += response.usage.completion_tokens
|
| 71 |
+
price = total_prompt_tokens_used*0.005/1000 + total_completion_tokens_used*0.015/1000
|
| 72 |
|
| 73 |
full_text += completed_text + "\n\n----------\n\n"
|
| 74 |
total_price += price
|