Upload concordancer.py with huggingface_hub

concordancer.py

@@ -72,7 +72,7 @@ mark {
 )
 
 
-def _highlight(text, query):
+def _highlight(text, query, case_sensitive=False):
     if not text or not query:
         return html.escape(text)
     escaped_text = html.escape(text)
@@ -81,15 +81,17 @@ def _highlight(text, query):
         rf"({regex.escape(escaped_query)})",
         r'<mark style="background:#fef08a;padding:1px 2px;border-radius:2px">\1</mark>',
         escaped_text,
-        flags=regex.IGNORECASE | regex.V1,
+        flags=regex.V1 if case_sensitive else regex.IGNORECASE | regex.V1,
     )
 
 
-def _highlight_regex(text, pattern):
+def _highlight_regex(text, pattern, case_sensitive=False):
     if not text or not pattern:
         return html.escape(text)
     try:
-        compiled = regex.compile(pattern, flags=regex.IGNORECASE | regex.V1)
+        compiled = regex.compile(
+            pattern, flags=regex.V1 if case_sensitive else regex.IGNORECASE | regex.V1
+        )
     except regex.error:
         return html.escape(text)
 
@@ -173,6 +175,7 @@ with st.form("search_form", clear_on_submit=False):
         )
     with col2:
         use_regex = st.checkbox("Regex", value=False)
+        case_sensitive = st.checkbox("Case sensitive", value=False)
         submitted = st.form_submit_button("Submit", use_container_width=True)
     with col3:
         per_page = st.selectbox("Per page", [10, 25, 50, 100], index=1)
@@ -193,6 +196,7 @@ if law_id_filter:
 search_signature = (
     query,
     use_regex,
+    case_sensitive,
     per_page,
     cat_filter,
     law_id_filter,
@@ -210,6 +214,7 @@ if query:
         conn,
         query,
         use_regex=use_regex,
+        case_sensitive=case_sensitive,
         law_id=law_id_filter,
         category=cat_filter,
         article_no=article_filter,
@@ -264,11 +269,11 @@ if query:
             en_text = row["en_text"] or ""
 
             if query and use_regex:
-                zh_display = _highlight_regex(zh_text, query)
-                en_display = _highlight_regex(en_text, query)
+                zh_display = _highlight_regex(zh_text, query, case_sensitive=case_sensitive)
+                en_display = _highlight_regex(en_text, query, case_sensitive=case_sensitive)
             elif query and not use_regex:
-                zh_display = _highlight(zh_text, query)
-                en_display = _highlight(en_text, query)
+                zh_display = _highlight(zh_text, query, case_sensitive=case_sensitive)
+                en_display = _highlight(en_text, query, case_sensitive=case_sensitive)
             else:
                 zh_display = html.escape(zh_text)
                 en_display = html.escape(en_text)
@@ -305,11 +310,19 @@ if query:
                         para_zh = _join_sentences(para["sentences"], "zh")
                         para_en = _join_sentences(para["sentences"], "en")
                         if query and use_regex:
-                            para_zh_display = _highlight_regex(para_zh, query)
-                            para_en_display = _highlight_regex(para_en, query)
+                            para_zh_display = _highlight_regex(
+                                para_zh, query, case_sensitive=case_sensitive
+                            )
+                            para_en_display = _highlight_regex(
+                                para_en, query, case_sensitive=case_sensitive
+                            )
                         elif query and not use_regex:
-                            para_zh_display = _highlight(para_zh, query)
-                            para_en_display = _highlight(para_en, query)
+                            para_zh_display = _highlight(
+                                para_zh, query, case_sensitive=case_sensitive
+                            )
+                            para_en_display = _highlight(
+                                para_en, query, case_sensitive=case_sensitive
+                            )
                         else:
                             para_zh_display = html.escape(para_zh)
                             para_en_display = html.escape(para_en)
@@ -337,11 +350,19 @@ if query:
                             art_zh = _join_sentences(para["sentences"], "zh")
                             art_en = _join_sentences(para["sentences"], "en")
                             if query and use_regex:
-                                art_zh_display = _highlight_regex(art_zh, query)
-                                art_en_display = _highlight_regex(art_en, query)
+                                art_zh_display = _highlight_regex(
+                                    art_zh, query, case_sensitive=case_sensitive
+                                )
+                                art_en_display = _highlight_regex(
+                                    art_en, query, case_sensitive=case_sensitive
+                                )
                             elif query and not use_regex:
-                                art_zh_display = _highlight(art_zh, query)
-                                art_en_display = _highlight(art_en, query)
+                                art_zh_display = _highlight(
+                                    art_zh, query, case_sensitive=case_sensitive
+                                )
+                                art_en_display = _highlight(
+                                    art_en, query, case_sensitive=case_sensitive
+                                )
                             else:
                                 art_zh_display = html.escape(art_zh)
                                 art_en_display = html.escape(art_en)