Tonic commited on
Commit
a7d5408
·
1 Parent(s): 7fb4afb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -206,14 +206,14 @@ def retrieve_info(query):
206
 
207
  return final_response.choices[0].text
208
 
209
- def combined_interface(query, file, collection_name):
210
  if query:
211
  article_info = retrieve_info(query)
212
  return article_info
213
- elif file is not None and collection_name:
214
- return embed_pdf(file, collection_name)
215
  else:
216
- return "Please enter a query or upload a PDF file."
217
 
218
  iface = gr.Interface(
219
  fn=combined_interface,
 
206
 
207
  return final_response.choices[0].text
208
 
209
+ def combined_interface(query, file, filename, collection_name):
210
  if query:
211
  article_info = retrieve_info(query)
212
  return article_info
213
+ elif file is not None and filename and collection_name:
214
+ return embed_pdf(file, filename, collection_name)
215
  else:
216
+ return "Please enter a query or upload a PDF file and specify a collection name."
217
 
218
  iface = gr.Interface(
219
  fn=combined_interface,