import streamlit as st

st.set_page_config(
    page_title="Pathfinder",
    page_icon="👋",
)

# st.write("# Welcome to Pathfinder! 👋")
st.image('local_files/pathfinder_logo.png',caption="Pathfinder: LLM enabled literature search")
st.sidebar.success("Select a function above.")
st.sidebar.markdown("Current functions include visualizing papers in the arxiv embedding, searching for similar papers to an input paper or prompt phrase, or answering quick questions.")

st.markdown(
    """
    Pathfinder (formerly called arXiv+GPT) is a framework for searching and
    visualizing papers on the [arXiv](https://arxiv.org/) using the context
    sensitivity from modern large language models (LLMs) to better link paper contexts.

    **👈 Select a tool from the sidebar** to see some examples
    of what this framework can do!

    ### Tool summary:
    - `Paper search` looks for relevant papers given an arxiv id or a question.
    - `Arxiv embedding` shows the landscape of current galaxy evolution papers (astro-ph.GA)
    - `QA sources` brings it all together to give concise answers to questions with primary sources and relevant papers.

    This is not meant to be a replacement to existing tools like the
    [ADS](https://ui.adsabs.harvard.edu/),
    [arxivsorter](https://www.arxivsorter.org/), but rather a supplement to find papers
    that otherwise might be missed during a literature survey.
    It is also only trained on astro-ph.GA (astrophysics of galaxies) papers currently,
    if you are interested in extending it please reach out!

    The image below shows a representation of all the astro-ph.GA papers that can be explored in more detail
    using the `Arxiv embedding` page. The papers tend to cluster together by similarity, and result in an
    atlas that shows well studied (forests) and currently uncharted areas (water).
    """
)

# st.image('https://drive.google.com/uc?id=1yQQCdlgnFzi-_yOMplGIqEyPKJhIsZpO&export=download')
st.image('local_files/galaxy_worldmap_kiyer-min.png')

st.markdown(
    """
    ### Coming soon:
    - [AstroLLaMA](https://huggingface.co/spaces/universeTBD/astrollama) embeddings!
    - export results
    - daily updates to repo
    - other fields apart from `astro-ph.GA`

    ### Want to learn more?
    - Check out `AstroLLaMA` [paper](https://huggingface.co/papers/2309.06126)
    - Check out `chaotic_neural` [(link)](http://chaotic-neural.readthedocs.io/)
    - Jump into our [documentation](https://docs.streamlit.io)
    - Contribute!

    Pathfinder is developed and maintained by [UniverseTBD](https://universetbd.org/). Updates on [huggingface](https://huggingface.co/universeTBD) or [twitter](https://twitter.com/universe_tbd).

"""
)