joshuadunlop commited on
Commit
dff4535
·
1 Parent(s): 426b438

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -4
app.py CHANGED
@@ -236,9 +236,12 @@ class WorkerThread(threading.Thread):
236
  job = self.jobs.get()
237
  if job is None:
238
  break
239
- i, question = job
240
- result = generate_answer(question, openAI_key)
241
- self.results.put((i, result))
 
 
 
242
 
243
  if generate_all:
244
  questions = [st.session_state.get(f"question{i}", "") for i in range(row_count)]
@@ -262,4 +265,7 @@ if generate_all:
262
 
263
  while not results.empty():
264
  i, answer = results.get()
265
- st.session_state[f'session_answer{i}'] = answer
 
 
 
 
236
  job = self.jobs.get()
237
  if job is None:
238
  break
239
+ try:
240
+ i, question = job
241
+ result = generate_answer(question, openAI_key)
242
+ self.results.put((i, result))
243
+ except Exception as e:
244
+ self.results.put((i, str(e)))
245
 
246
  if generate_all:
247
  questions = [st.session_state.get(f"question{i}", "") for i in range(row_count)]
 
265
 
266
  while not results.empty():
267
  i, answer = results.get()
268
+ if isinstance(answer, str) and 'Error' in answer:
269
+ st.error(f"Error on row {i}: {answer}")
270
+ else:
271
+ st.session_state[f'session_answer{i}'] = answer