Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -7,19 +7,23 @@ from datetime import datetime
|
|
7 |
# Put your OpenAI API key here
|
8 |
openai.api_key = os.getenv('OpenAPI')
|
9 |
|
10 |
-
def gpt3_query(prompt, engine='gpt-3.5-turbo-16k', max_tokens=15050, temperature=0.5):
|
11 |
-
|
12 |
-
|
13 |
-
|
14 |
-
|
15 |
-
|
16 |
-
|
17 |
-
|
18 |
-
|
19 |
-
|
20 |
-
|
21 |
-
|
22 |
-
|
|
|
|
|
|
|
|
|
23 |
|
24 |
def read_file(file_path):
|
25 |
with open(file_path, 'r') as file:
|
|
|
7 |
# Put your OpenAI API key here
|
8 |
openai.api_key = os.getenv('OpenAPI')
|
9 |
|
10 |
+
def gpt3_query(prompt, engine='gpt-3.5-turbo-16k', max_tokens=15050, temperature=0.5, retries=3):
|
11 |
+
for i in range(retries):
|
12 |
+
try:
|
13 |
+
response = openai.ChatCompletion.create(
|
14 |
+
model=engine,
|
15 |
+
messages=[{"role": "system", "content": "You are a creative writer who specializes in crafting engaging stories to sell products or services"},
|
16 |
+
{"role": "user", "content": prompt}],
|
17 |
+
max_tokens=max_tokens,
|
18 |
+
n=1,
|
19 |
+
temperature=temperature
|
20 |
+
)
|
21 |
+
return response.choices[0].message['content'].strip()
|
22 |
+
except Exception as e:
|
23 |
+
if i < retries - 1: # i is zero indexed
|
24 |
+
continue
|
25 |
+
else:
|
26 |
+
return f"Error in gpt3_query after {retries} attempts: {str(e)}", None
|
27 |
|
28 |
def read_file(file_path):
|
29 |
with open(file_path, 'r') as file:
|