Bug fix

app.py

@@ -18,7 +18,7 @@ def process(query_type, index_desc, **kwargs):
     if API_URL is None:
         raise ValueError(f'API_URL envvar is not set!')
     try:
-        response = requests.post(API_URL, json=data, timeout=10)
+        response = requests.post(API_URL, json=data, timeout=30)
     except requests.exceptions.Timeout:
         raise ValueError('Web request timed out. Please try again later.')
     except requests.exceptions.RequestException as e:
@@ -61,10 +61,10 @@ def creativity(index_desc, query):
     htmls = []
     for n in range(NGRAM_LEN_MIN, NGRAM_LEN_MAX + 1):
         html = ''
-        highlighted = highlighteds_by_n[n]
+        highlighteds = highlighteds_by_n[n]
         line_len = 0
         for i, (token, highlighted) in enumerate(zip(tokens, highlighteds)):
-            if line_len >= 100 and token.startswith('▁'):
+            if line_len >= MAX_DISP_CHARS_PER_LINE and token.startswith('▁'):
                 html += '<br/>'
                 line_len = 0
             color = '0, 0, 255, 0.5'

constants.py

@@ -12,9 +12,9 @@ INDEX_DESCS = list(INDEX_BY_DESC.keys())
 
 # API limits and defaults
 MAX_QUERY_CHARS = int(os.environ.get('MAX_QUERY_CHARS', 1000))
-NGRAM_LEN_DEFAULT = int(os.environ.get('NGRAM_LEN_DEFAULT', 8))
 NGRAM_LEN_MIN = int(os.environ.get('NGRAM_LEN_MIN', 5))
 NGRAM_LEN_MAX = int(os.environ.get('NGRAM_LEN_MAX', 11))
+MAX_DISP_CHARS_PER_LINE = int(os.environ.get('MAX_DISP_CHARS_PER_LINE', 100))
 
 # HF demo
 API_URL = os.environ.get('API_URL', None)