chnage colors

app.py

@@ -13,7 +13,7 @@ es = Elasticsearch(os.environ.get("host"), timeout=100, http_compress=True, maxs
 def mark_tokens_bold(string, tokens):
   for token in tokens:
     pattern = re.escape(token) #r"\b" + re.escape(token) + r"\b"
-    string = re.sub(pattern, "<span style='color: #ff75b3;'><b>" + token + "</b></span>", string)
+    string = re.sub(pattern, "<span style='color: #e6b800;'><b>" + token + "</b></span>", string)
   return string
 
 
@@ -26,11 +26,16 @@ def process_results(results):
         text_html = result["text"]
         # text_html = mark_tokens_bold(text_html, highlight_terms)
         repository = result["repository"]
-        
+        license = result["license"]
+        language = result["language"]
         results_html += """\
-            <p style='font-size:16px; text-align: left; color: white;'>Source: <span style='color: #727cd6;'>{}</span></p>
+            <p style='font-size:16px; text-align: left;'>Source: <span style='color: #00134d;'>{}</span></p>
             <br>
-            <pre style='height: 600px; overflow-y: scroll; overflow-x: hidden; color: #d9d9d9;border: 1px solid #ff75b3; padding: 10px'><code>{}</code></pre>
+            <p style='font-size:16px;> Language:<span style='color: #00134d;'>Python</span></p>
+            <br>
+            <p style='font-size:16px;> License:<span style='color: #00134d;'>MIT</span></p>
+            <br>
+            <pre style='height: 600px; overflow-y: scroll; overflow-x: hidden; color: #d9d9d9;border: 1px solid #e6b800; padding: 10px'><code>{}</code></pre>
             <br>
             <hr>
             <br>
@@ -59,19 +64,40 @@ def search(query, num_results=10):
         print(len(response))
     else:
         response = match_query(query, num_results=num_results)
-    results = [{"text": hit.content, "repository": f"{hit.repository}/{hit.path}"} for hit in response]
+    results = [{"text": hit.content, "repository": f"{hit.repository}/{hit.path}", "license": hit.license, "language": hit.language} for hit in response]
     return process_results(results)
 
-
-description = """# <p style="text-align: center; color: white;"><span style='color: #ff75b3;'>StarCoder:</span> Dataset Search 🔍 </p>
-<span style='color: white;'>When you use <a href="https://huggingface.co/bigcode/large-model" style="color: #ff75b3;">StarCoder</a> to generate code it might produce exact copies of code in the pretraining dataset. 
-In that case, the code license might have requirements to comply with. 
-With this search tool we aim to provide help to find out where the code came from, in order for the user to comply with licensing requirements in case the code produced by StarCoder belongs to an already existing repository. For exact matches, enclose your query in double quotes.</span>"""
+description = """# <p style="text-align: center; color: white;"><span style='color: #e6b800;'>StarCoder:</span> Dataset Search 🔍 </p>
+<span style='color: white;'>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. \
+In that case, the code license might have requirements to comply with. With this search tool, our aim is to help in identifying if the code belongs to an existing repository. For exact matches, enclose your query in double quotes.</span>"""
+
+theme = gr.themes.Monochrome(
+    primary_hue="indigo",
+    secondary_hue="blue",
+    neutral_hue="slate",
+    radius_size=gr.themes.sizes.radius_sm,
+    font=[
+        gr.themes.GoogleFont("Open Sans"),
+        "ui-sans-serif",
+        "system-ui",
+        "sans-serif",
+    ],
+)
+css = ".generating {visibility: hidden}"
+
+monospace_css = """
+#q-input textarea {
+    font-family: monospace, 'Consolas', Courier, monospace;
+}
+"""
+
+css = monospace_css + ".gradio-container {color: black}"
 
 
 if __name__ == "__main__":
     demo = gr.Blocks(
-        css=".gradio-container {background-color: #20233fff; color:white}"
+        theme=theme,
+        css=css,
     )
 
     with demo:
@@ -84,7 +110,7 @@ if __name__ == "__main__":
         with gr.Row():
             submit_btn = gr.Button("Submit")
         with gr.Row():
-            results = gr.HTML(label="Results", value="<img src='https://huggingface.co/datasets/bigcode/admin/resolve/main/bigcode_contact.png' alt='contact' style='display: block; margin: auto; max-width: 800px;'>")
+            results = gr.HTML(label="Results", value="")
 
         def submit(query, k, lang="en"):
             query = query.strip()