added suggested number of words slide bar

app.py

@@ -34,11 +34,8 @@ if len(api_key) != 51:
 
 max_results_current = 5
 max_results = max_results_current
-def search_click_callback(search_query, max_results):
+def search_click_callback(search_query, max_results, XRxiv_servers=[]):
     global pdf_info, pdf_citation
-    # pdf_info, pdf_citation = call_arXiv_API(f'{search_query}', max_results=max_results)
-    # download_pdf(pdf_info)
-    XRxiv_servers = ['rxiv']
     search_engines = XRxivQuery(search_query, max_results, XRxiv_servers=XRxiv_servers)
     pdf_info = search_engines.call_API()
     search_engines.download_pdf()
@@ -46,7 +43,7 @@ def search_click_callback(search_query, max_results):
     return pdf_info
 
 with st.form(key='columns_in_form', clear_on_submit = False):
-    c1, c2 = st.columns([8,1])
+    c1, c2, c3 = st.columns([6, 0.6, 3])
     with c1:
         search_query = st.text_input("Input search query here:", placeholder='Keywords for most relevant search...', value='CFD Modeling'
                                        )#search_query, max_results_current))
@@ -54,17 +51,40 @@ with st.form(key='columns_in_form', clear_on_submit = False):
     with c2:
         max_results = st.number_input("Max papers", value=max_results_current)
         max_results_current = max_results_current
+    with c3:
+        with st.container():
+            st.markdown('Pre-print server')
+            checks = st.columns(4)
+            with checks[0]:
+                ArXiv_check = st.checkbox('ArXiv')
+            with checks[1]:
+                ChemArXiv_check = st.checkbox('ChemArXiv')
+            with checks[2]:
+                BioArXiv_check = st.checkbox('BioArXiv')
+            with checks[3]:
+                MedrXiv_check = st.checkbox('MedrXiv')
+
     searchButton = st.form_submit_button(label = 'Search')
 
 if searchButton:
+    # checking which pre-print servers selected
+    XRxiv_servers = []
+    if ArXiv_check:
+        XRxiv_servers.append('rxiv')
+    if ChemArXiv_check:
+        XRxiv_servers.append('chemrxiv')
+    if BioArXiv_check:
+        XRxiv_servers.append('biorxiv')
+    if MedrXiv_check:
+        XRxiv_servers.append('medrxiv')
     global pdf_info
-    pdf_info =  search_click_callback(search_query, max_results)
+    pdf_info =  search_click_callback(search_query, max_results, XRxiv_servers=XRxiv_servers)
     if 'pdf_info' not in st.session_state:
         st.session_state.key = 'pdf_info'
     st.session_state['pdf_info'] = pdf_info
 
 
-def answer_callback(question_query):
+def answer_callback(question_query, word_count):
     import paperqa
     global docs
     if docs is None:
@@ -76,21 +96,29 @@ def answer_callback(question_query):
         for d, c in zip(pdf_paths, pdf_citations):
             docs.add(d, c)
     docs._build_faiss_index()
-    answer = docs.query(question_query)
+    answer = docs.query(question_query,  length_prompt=f'use {word_count:d} words')
     st.success('Voila! 😃')
     return answer.formatted_answer
 
-form = st.form(key='question_form')
-question_query = form.text_input("What do you wanna know from these papers?", placeholder='Input questions here...',
+with st.form(key='question_form', clear_on_submit = False):
+    c1, c2 = st.columns([6, 2])
+    with c1:
+        question_query = st.text_input("What do you wanna know from these papers?", placeholder='Input questions here...',
                 value='Write about CFD modeling')
-submitButton = form.form_submit_button('Submit')
+        # search_query = st.text_input("Input search query here:", placeholder='Keywords for most relevant search...', value='CFD Modeling'
+        #                                )#search_query, max_results_current))
+
+    with c2:
+        word_count = st.slider("Suggested number of words in your answer?", 30, 300, 100)
+    submitButton = st.form_submit_button('Submit')
 
 if submitButton:
     with st.expander("Found papers:", expanded=True):
         st.write(f"{st.session_state['all_reference_text']}")
     with st.spinner('⏳ Please wait...'):
         start = time.time()
-        final_answer = answer_callback(question_query)
+        print(word_count)
+        final_answer = answer_callback(question_query, word_count)
         length_answer = len(final_answer)
         st.text_area("Answer:", final_answer, height=max(length_answer//4, 100))
         end = time.time()