Update app.py

app.py

@@ -12,11 +12,11 @@ from elasticsearch_dsl import Search, Q
 
 # es = Elasticsearch(os.environ.get("host"), timeout=100, http_compress=True, maxsize=1000)
 es = Elasticsearch(os.environ.get("host"), http_compress=True, timeout=200)
-def mark_tokens_bold(text, query):
-    if query.startswith('"') and query.endswith('"'):
-        tokens = query[1:-1]
-    else:
-        tokens = query.split(" ")
+def mark_tokens_bold(text, tokens):
+    # if query.startswith('"') and query.endswith('"'):
+    #     tokens = query[1:-1]
+    # else:
+    #     tokens = query.split(" ")
     for token in tokens:
         pattern = re.escape(token) #r"\b" + re.escape(token) + r"\b"
         text = re.sub(pattern, "<span style='color: #e6b800;'><b>" + token + "</b></span>", text)
@@ -30,7 +30,7 @@ def process_results(results, query):
     results_html = ""
     for result in results:
         text_html = result["text"]
-        # text_html = mark_tokens_bold(text_html, query)
+        text_html = mark_tokens_bold(text_html, query.split(" "))
         repository = result['repository']
         license = result["license"]
         language = result["language"]