import streamlit as st
import core


st.set_page_config(
    page_title="arXiv Seach",
    layout="wide"
)

if 'last_query' not in st.session_state:
    st.session_state['last_query'] = None
    st.session_state['num_articles_to_show'] = 10

core.login()
st.markdown("## arXiv Search")

with st.form("Search form"):
    query = st.text_input("Enter a query")
    clicked = st.form_submit_button("Search")


if clicked:
    last_query = st.session_state['last_query']
    if last_query != query:
        st.session_state['num_articles_to_show'] = 10
    st.session_state['last_query'] = query

if st.session_state['last_query'] is not None:
    core.display_answers(st.session_state['last_query'])
    more_answers = st.button("More answers")
    if more_answers:
        st.session_state['num_articles_to_show'] += 10
        st.experimental_rerun()