joshuadunlop commited on
Commit
2231f35
·
1 Parent(s): dff4535

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -3
app.py CHANGED
@@ -98,7 +98,10 @@ def load_recommender(path, start_page=1):
98
  return 'Corpus Loaded.'
99
 
100
  def generate_text(openAI_key,prompt, engine="text-davinci-003"):
101
- openai.api_key = openAI_key
 
 
 
102
  completions = openai.Completion.create(
103
  engine=engine,
104
  prompt=prompt,
@@ -188,7 +191,10 @@ for i in range(row_count):
188
  st.error('Question field is empty')
189
  else:
190
  glob_url = url
191
- download_pdf(glob_url, 'corpus.pdf')
 
 
 
192
  load_recommender('corpus.pdf')
193
 
194
  answer = generate_answer(question,openAI_key)
@@ -241,7 +247,7 @@ class WorkerThread(threading.Thread):
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)]
 
98
  return 'Corpus Loaded.'
99
 
100
  def generate_text(openAI_key,prompt, engine="text-davinci-003"):
101
+ try:
102
+ openai.api_key = openAI_key
103
+ except Exception as e:
104
+ raise ValueError("Failed to set OpenAI API key: " + str(e))
105
  completions = openai.Completion.create(
106
  engine=engine,
107
  prompt=prompt,
 
191
  st.error('Question field is empty')
192
  else:
193
  glob_url = url
194
+ try:
195
+ download_pdf(glob_url, 'corpus.pdf')
196
+ except Exception as e:
197
+ raise ValueError("Failed to download and process the PDF: " + str(e))
198
  load_recommender('corpus.pdf')
199
 
200
  answer = generate_answer(question,openAI_key)
 
247
  result = generate_answer(question, openAI_key)
248
  self.results.put((i, result))
249
  except Exception as e:
250
+ self.results.put((i, "Error (line number, error): (" + str(i) + ", " + str(e) + ")"))
251
 
252
  if generate_all:
253
  questions = [st.session_state.get(f"question{i}", "") for i in range(row_count)]