Add application file

.streamlit/config.toml

@@ -0,0 +1,6 @@
+[theme]
+primaryColor="#F63366"
+backgroundColor="#FFFFFF"
+secondaryBackgroundColor="#F0F2F6"
+textColor="#262730"
+font="sans serif"

app.py

@@ -10,7 +10,7 @@ import streamlit as st
 import streamlit.components.v1 as components
 import requests
 
-
+os.environ["address"] = "http://34.79.83.149:8080"
 pp = pprint.PrettyPrinter(indent=2)
 st.set_page_config(page_title="Gaia Search", layout="wide")
 
@@ -67,13 +67,14 @@ unsafe_allow_html=True,
 query = st.sidebar.text_input(label='Search query', value='')
 language = st.sidebar.selectbox(
     'Language',
-    ('Arabic', 'Catalan', 'Code', 'English', 'Spanish', 'French', 'Indonesian', 'Indic', 'Niger-Congo', 'Portuguese', 'Vietnamese', 'Chinese', 'Detect Language', 'All'))
+    ('Arabic', 'Catalan', 'Code', 'English', 'Spanish', 'French', 'Indonesian', 'Indic', 'Niger-Congo', 'Portuguese', 'Vietnamese', 'Chinese', 'Detect Language', 'All'),
+    index=3)
 max_results = st.sidebar.slider(
     "Maximum Number of Results",
     min_value=1,
     max_value=100,
     step=1,
-    value=1,
+    value=10,
     help="Maximum Number of Documents to return",
 )
 
@@ -122,7 +123,7 @@ def scisearch(query, language, num_results=10):
 
 def highlight_string(paragraph: str, highlight_terms: list) -> str:
     for term in highlight_terms:
-        paragraph = paragraph.replace(term, f"<mark>{term}</mark>")
+        paragraph = re.sub(f"\\b{term}\\b", f"<mark>{term}</mark>", paragraph, flags=re.I)
     return paragraph
 
 def process_results(hits: list, highlight_terms: list) -> str:
@@ -141,18 +142,19 @@ def process_results(hits: list, highlight_terms: list) -> str:
         res_head += f"""
                     <p>{highlight_string(hit['text'], highlight_terms)}</p>
                     </div>
+                    <hr>
                     """
         hit_list.append(res_head)
     return " ".join(hit_list)
 
 
 if st.sidebar.button("Search"):
-    hits, highlight_terms = scisearch(query, LANG_MAPPING[language])
+    hits, highlight_terms = scisearch(query, LANG_MAPPING[language], max_results)
     html_results = process_results(hits, highlight_terms)
     rendered_results = f"""
             <div id="searchresultsarea">
                 <br>
-                <p id="searchresultsnumber">About 155,000 results (0.56 seconds) </p>
+                <p id="searchresultsnumber">About {max_results} results</p>
                 {html_results}
             </div>
                         """
@@ -182,7 +184,7 @@ if st.sidebar.button("Search"):
         """,
         unsafe_allow_html=True)
     components.html(
-            """
+        """
         <style>
         #searchresultsarea {
             font-family: 'Arial';
@@ -197,7 +199,7 @@ if st.sidebar.button("Search"):
             font-size: 19px;
             line-height: 18px;
             font-weight: normal;
-            color: rgb(29, 1, 189);
+            color: rgb(7, 111, 222);
             margin-bottom: 0px;
             margin-top: 25px;
         }
@@ -218,7 +220,16 @@ if st.sidebar.button("Search"):
             border-width: 0px;
             background-color: white;
         }
-
+        
+        .dark-mode {
+          color: white;
+        }
         </style>
+        <script>
+        function myFunction() {
+           var element = document.body;
+           element.classList.toggle("dark-mode");
+        }
+        </script>
         """ + rendered_results, height=1000, scrolling=True
         )