Update app.py

app.py

@@ -3,7 +3,7 @@ import pandas as pd
 import numpy as np
 
 from src.Surveyor import Surveyor
-from streamlit_tags import st_sidebar_tags
+
 
 
 @st.experimental_singleton(show_spinner=True, suppress_st_warning=True)
@@ -41,6 +41,39 @@ def show_survey_download(zip_file_name, survey_file_name, download_placeholder):
 
 def survey_space(surveyor, download_placeholder):
 
+    class KeywordsModel(BaseModel):
+        research_keywords: Optional[str] =  Field(
+            '', description="Enter your research keywords:"
+        )
+        max_search: int = Field(
+            10, ge=1, le=50, multiple_of=1,
+            description="num_papers_to_search:"
+        )
+        num_papers: int = Field(
+            3, ge=1, le=8, multiple_of=1, 
+            description="num_papers_to_select:"
+        )
+    class ArxivIDsModel(BaseModel):
+        arxiv_ids: Optional[List[str]] =  Field(
+            [], max_items=8, regex=r"^[0-9]+\.[0-9]+$", description="Enter arxiv ids for your curated set of papers (e.g. 2205.12755, 2205.10937, ...):"
+        )
+    
+    
+    with st.sidebar.form(key="survey_keywords_form"):
+        session_data = sp.pydantic_input(key="keywords_input_model", model=KeywordsModel)
+        st.write('or')
+        session_data.update(sp.pydantic_input(key="arxiv_ids_input_model", model=ArxivIDsModel))
+        submit = st.form_submit_button(label="Submit")
+        
+    run_kwargs = {'surveyor':surveyor, 'download_placeholder':download_placeholder}
+    if submit:
+        if session_data['research_keywords'] != '':
+            run_kwargs.update({'research_keywords':session_data['research_keywords'], 'max_search':max_search, 'num_papers':num_papers})
+        elif len(session_data['arxiv_ids']):
+            run_kwargs.update({'arxiv_ids':session_data['arxiv_ids']})
+        run_survey(**run_kwargs)
+    
+    '''
     form = st.sidebar.form(key='survey_form')
     research_keywords = form.text_input("Enter your research keywords:", key='research_keywords', value='')
     max_search = form.number_input("num_papers_to_search", help="maximium number of papers to glance through - defaults to 20", 
@@ -59,6 +92,7 @@ def survey_space(surveyor, download_placeholder):
                 
     submit = form.form_submit_button('Submit')
     
+    
     run_kwargs = {'surveyor':surveyor, 'download_placeholder':download_placeholder}
     if submit:
         if research_keywords != '':
@@ -66,6 +100,7 @@ def survey_space(surveyor, download_placeholder):
         elif len(arxiv_ids):
             run_kwargs.update({'arxiv_ids':arxiv_ids})
         run_survey(**run_kwargs)
+    '''