Update app.py

app.py

@@ -19,15 +19,15 @@ def mark_tokens_bold(string, tokens):
   return string
 
 
-def process_results(results):
+def process_results(results, query):
     if len(results) == 0:
         return """<br><p>No results retrieved.</p><br><hr>"""
 
     results_html = ""
     for result in results:
         text_html = result["text"]
-        # text_html = mark_tokens_bold(text_html, highlight_terms)
-        repository = '<a href=\"https://github.com/' + result['repository'] + '>' + result['repository'] + '</a>'
+        text_html = mark_tokens_bold(text_html, query.split(" "))
+        repository = result['repository']
         license = result["license"]
         language = result["language"]
         code_height = min(600, len(text_html.split('\n')) * 20) # limit to maximum height of 600px
@@ -64,7 +64,7 @@ def search(query, num_results=10):
     else:
         response = match_query(query, num_results=num_results)
     results = [{"text": nh3.clean(html.escape(hit.content)), "repository": f"{hit.repository}/{hit.path}", "license": hit.license[0], "language": hit.language} for hit in response]
-    return process_results(results)
+    return process_results(results, query)
 
 description = """# <p style="text-align: center;"><span style='color: #e6b800;'>StarCoder:</span> Dataset Search 🔍 </p>
 <span>When using <a href="https://huggingface.co/bigcode/large-model" style="color: #e6b800;">StarCoder</a> to generate code, it might produce exact copies of code in the pretraining dataset. \