import streamlit as st

st.set_page_config(
    page_title="arXiv-GPT",
    page_icon="👋",
)

st.write("# Welcome to arXiv-GPT! 👋")

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(
    """
    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!

    ### Page 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.
    """
)

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

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!

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

"""
)