first commit

app.py

@@ -0,0 +1,83 @@
+import requests
+import streamlit as st
+from annotated_text import annotated_text
+
+
+@st.cache_data
+def send_autocomplete_request(q: str, lang: str, taxonomy_names: list[str], size: int):
+    return requests.get(
+        "http://localhost:8000/autocomplete",
+        params={
+            "q": q,
+            "lang": lang,
+            "taxonomy_names": ",".join(taxonomy_names),
+            "size": size,
+        },
+    ).json()
+
+
+def get_product(barcode: str):
+    r = requests.get(f"https://world.openfoodfacts.org/api/v2/product/{barcode}")
+
+    if r.status_code == 404:
+        return None
+
+    return r.json()["product"]
+
+
+def display_ner_tags(text: str, entities: list[dict]):
+    spans = []
+    previous_idx = 0
+    for entity in entities:
+        score = entity["score"]
+        lang = entity["lang"]["lang"]
+        start_idx = entity["start"]
+        end_idx = entity["end"]
+        spans.append(text[previous_idx:start_idx])
+        spans.append((text[start_idx:end_idx], f"score={score:.3f} | lang={lang}"))
+        previous_idx = end_idx
+    spans.append(text[previous_idx:])
+    annotated_text(spans)
+
+
+def run(
+    query: str,
+    taxonomy_names: list[str],
+    lang: str,
+    size: int,
+):
+    response = send_autocomplete_request(query, lang, taxonomy_names, size)
+    took = response["took"]
+    timed_out = response["timed_out"]
+    st.markdown(f"Took: {took} ms")
+
+    if timed_out:
+        st.warning("Request timed out!")
+
+    options = response["options"]
+    options = [
+        {"text": o["text"], "id": o["id"], "taxonomy": o["taxonomy_name"]}
+        for o in options
+    ]
+    st.table(options)
+
+    st.write(response["debug"])
+
+
+st.title("Search autocomplete demo")
+st.markdown(
+    "This is a demo of the search autocomplete feature, that searches in the taxonomy names and synonyms."
+)
+query = st.text_input("query", help="Query to search for", value="chocolate").strip()
+taxonomy_names = st.multiselect(
+    "taxonomy names",
+    ["category", "label", "ingredient", "allergen", "brand"],
+    default=["category", "label", "ingredient", "brand"],
+)
+lang = st.selectbox("language", ["en", "fr", "it", "es", "nl", "de"], index=0)
+size = st.number_input(
+    "number of results", value=10, min_value=1, max_value=500, step=1
+)
+
+if query:
+    run(query, taxonomy_names, lang, size)