Add search buttons

app.py

@@ -27,11 +27,16 @@ st.sidebar.image("./assets/logo.png", width=100)
 st.sidebar.empty()
 st.sidebar.markdown("""Search for medical images with natural language powered by a CLIP model [[Model Card]](https://huggingface.co/flax-community/medclip-roco) finetuned on the
  [Radiology Objects in COntext (ROCO) dataset](https://github.com/razorx89/roco-dataset).""")
-st.sidebar.markdown("""Example queries:
-* `ultrasound scans`🔍
-* `pathology`🔍
-* `pancreatic carcinoma`🔍
-* `PET scan`🔍""")
+st.sidebar.markdown("Example queries:")
+# * `ultrasound scans`🔍
+# * `pathology`🔍
+# * `pancreatic carcinoma`🔍
+# * `PET scan`🔍""")
+ex1_button = st.sidebar.button("🔍 pathology")
+ex2_button = st.sidebar.button("🔍 ultrasound scans")
+ex3_button = st.sidebar.button("🔍 pancreatic carcinoma")
+ex4_button = st.sidebar.button("🔍 PET scan")
+
 k_slider = st.sidebar.slider("Number of images", min_value=1, max_value=10, value=5)
 st.sidebar.markdown("Kaushalya Madhawa, 2021")
 
@@ -44,11 +49,21 @@ st.title("MedCLIP 🩺")
 # * `pathology`🔍
 # * `pancreatic carcinoma`🔍
 # * `PET scan`🔍""")
+text_value = ''
+if ex1_button:
+    text_value = 'pathology'
+elif ex2_button:
+    text_value = 'ultrasound scans'
+elif ex3_button:
+    text_value = 'pancreatic carcinoma'
+elif ex4_button:
+    text_value = 'PET scan'
+
 
 image_list, image_embeddings = load_image_embeddings()
 model, processor = load_model()
 
-query = st.text_input("Enter your query here:")
+query = st.text_input("Enter your query here:", value=text_value)
 dot_prod = None
 
 if st.button("Search") or k_slider: