Sage commited on
Commit
8b0e4a2
1 Parent(s): 6fc7f47

settings changes

Browse files
Files changed (1) hide show
  1. app.py +11 -9
app.py CHANGED
@@ -32,6 +32,7 @@ def chat_gpt_image(content, context):
32
  completion = openai.ChatCompletion.create(
33
  model=gpt_model,
34
  user="1",
 
35
  messages=[
36
  {"role": "system", "content": prompt},
37
  {"role": "user", "content": final_content}
@@ -74,6 +75,7 @@ def chat_gpt_document(content, document_type, context):
74
  completion_1 = openai.ChatCompletion.create(
75
  model=gpt_model,
76
  user="1",
 
77
  messages=[
78
  {"role": "system", "content": prompt},
79
  {"role": "user", "content": content_1}
@@ -193,24 +195,24 @@ def batch_document(content, document_type, context, progress=gr.Progress()):
193
  data = json.loads(chat_gpt_document(detect_document(x),document_type,context))
194
  combined_data.append(data)
195
  break
196
- except openai.error.APIConnectionError:
197
- logging.error(f'Retry {i+1} failed: openai.error.APIConnectionError')
198
  if i < retries - 1:
199
  logging.error(f'Retrying in {timeout} seconds...')
200
  time.sleep(timeout)
201
  i += 1
202
- except openai.error.RateLimitError:
203
- logging.error(f'Retry {j+1} failed: openai.error.RateLimitError')
204
- if j < retries - 1:
205
- logging.error(f'Retrying in {timeout} seconds...')
206
- time.sleep(timeout)
207
- j += 1
208
  logging.info(combined_data)
209
  return save_json(combined_data, document_type)
210
 
211
  def save_json(text, filename):
212
  filename = filename+".json"
213
- with open(filename, "w") as outfile:
214
  json.dump(text, outfile)
215
  return filename
216
 
 
32
  completion = openai.ChatCompletion.create(
33
  model=gpt_model,
34
  user="1",
35
+ temperature="0.5",
36
  messages=[
37
  {"role": "system", "content": prompt},
38
  {"role": "user", "content": final_content}
 
75
  completion_1 = openai.ChatCompletion.create(
76
  model=gpt_model,
77
  user="1",
78
+ temperature="0.1",
79
  messages=[
80
  {"role": "system", "content": prompt},
81
  {"role": "user", "content": content_1}
 
195
  data = json.loads(chat_gpt_document(detect_document(x),document_type,context))
196
  combined_data.append(data)
197
  break
198
+ except Exception as e:
199
+ logging.error(f'Retry {i+1} failed: ', e)
200
  if i < retries - 1:
201
  logging.error(f'Retrying in {timeout} seconds...')
202
  time.sleep(timeout)
203
  i += 1
204
+ # except openai.error.RateLimitError:
205
+ # logging.error(f'Retry {j+1} failed: openai.error.RateLimitError')
206
+ # if j < retries - 1:
207
+ # logging.error(f'Retrying in {timeout} seconds...')
208
+ # time.sleep(timeout)
209
+ # j += 1
210
  logging.info(combined_data)
211
  return save_json(combined_data, document_type)
212
 
213
  def save_json(text, filename):
214
  filename = filename+".json"
215
+ with open(filename, "w", encoding='utf-8') as outfile:
216
  json.dump(text, outfile)
217
  return filename
218