add ability to specify strict or lenient

app.py

@@ -41,6 +41,10 @@ def search(term, limit=10, clean=True, strict=True):
             'Authorization': f'Bearer {SCITE_API_KEY}'
         }
     )
+    try:
+        req.json()
+    except:
+        return [], []
     return (
         [remove_html('\n'.join([cite['snippet'] for cite in doc['citations']])) for doc in req.json()['hits']],
         [(doc['doi'], doc['citations'], doc['title'])
@@ -80,6 +84,7 @@ def init_models():
 
 qa_model, reranker, stop, device = init_models()
 
+
 def clean_query(query, strict=True, clean=True):
     operator = ' '
     if strict:
@@ -136,6 +141,10 @@ st.markdown("""
 <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/bootstrap@4.0.0/dist/css/bootstrap.min.css" integrity="sha384-Gn5384xqQ1aoWXA+058RXPxPg6fy4IWvTNh0E263XmFcJlSAwiGgFAW/dAiS6JXm" crossorigin="anonymous">
 """, unsafe_allow_html=True)
 
+strict_mode = st.radio(
+    "Query mode? Strict means all words must match in source snippet. Lenient means only some words must match.",
+    ('strict', 'lenient'))
+
 def run_query(query):
     if device == 'cpu':
         limit = 50
@@ -143,7 +152,7 @@ def run_query(query):
     else:
         limit = 100
         context_limit = 25
-    contexts, orig_docs = search(query, limit=limit)
+    contexts, orig_docs = search(query, limit=limit, strict=strict_mode == 'strict')
     if len(contexts) == 0 or not ''.join(contexts).strip():
         return st.markdown("""
         <div class="container-fluid">