pydantic test

app.py

@@ -1,6 +1,7 @@
+from typing import List, Optional
 import streamlit as st
-import pandas as pd
-import numpy as np
+import streamlit_pydantic as sp
+from pydantic import BaseModel, Field
 
 from src.Surveyor import Surveyor
 
@@ -39,26 +40,27 @@ 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 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, ...):"
-        )
-    
-    
+
+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, ...):"
+    )
+
+
+def survey_space(surveyor, download_placeholder):
     with st.sidebar.form(key="survey_keywords_form"):
         session_data = sp.pydantic_input(key="keywords_input_model", model=KeywordsModel)
         st.write('or')
@@ -68,7 +70,9 @@ def survey_space(surveyor, download_placeholder):
     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})
+            run_kwargs.update({'research_keywords':session_data['research_keywords'], 
+                               'max_search':session_data['research_keywords'], 
+                               'num_papers':session_data['research_keywords']})
         elif len(session_data['arxiv_ids']):
             run_kwargs.update({'arxiv_ids':session_data['arxiv_ids']})
         run_survey(**run_kwargs)

requirements.txt

@@ -23,7 +23,7 @@ scispacy
 https://s3-us-west-2.amazonaws.com/ai2-s2-scispacy/releases/v0.5.0/en_core_sci_scibert-0.5.0.tar.gz
 https://s3-us-west-2.amazonaws.com/ai2-s2-scispacy/releases/v0.5.0/en_core_sci_lg-0.5.0.tar.gz
 streamlit
-streamlit-tags
+streamlit-pydantic
 summarizer
 tabula
 tabula_py