import dashboard_text2image
import dashboard_image2image
import dashboard_featurefinder

import streamlit as st

PAGES = {
    "Retrieve Images given Text": dashboard_text2image,
    "Retrieve Images given Image": dashboard_image2image,
    "Find Feature in Image": dashboard_featurefinder,
}

st.sidebar.title("CLIP-RSICD")
st.sidebar.image("thumbnail.jpg")
st.sidebar.markdown("""
    We have fine-tuned the CLIP model (see [Model card](https://huggingface.co/flax-community/clip-rsicd-v2)) 
    using remote sensing images and captions from the [RSICD dataset](https://github.com/201528014227051/RSICD_optimal). 
    The CLIP model from OpenAI is trained in a self-supervised manner using contrastive learning to project images 
    and caption text onto a common embedding space.

    Please click here for [more information about our project](https://github.com/arampacha/CLIP-rsicd).

""")
selection = st.sidebar.radio("Go to", list(PAGES.keys()))
page = PAGES[selection]
page.app()