test lists

app.py

@@ -4,7 +4,6 @@ import streamlit_pydantic as sp
 from pydantic import BaseModel, Field
 
 from src.Surveyor import Surveyor
-from temp_showcase_model import ShowcaseModel
 
 
 
@@ -56,8 +55,8 @@ class KeywordsModel(BaseModel):
 
 
 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, ...):"
+    arxiv_ids: Optional[str] =  Field(
+        '', description="Enter comma_separated arxiv ids for your curated set of papers (e.g. 2205.12755, 2205.10937, ...):"
     )
 
 
@@ -74,8 +73,9 @@ def survey_space(surveyor, download_placeholder):
             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']})
+        elif session_data['arxiv_ids'] != '':
+            run_kwargs.update({'arxiv_ids':[id.strip() for id in session_data['arxiv_ids'].split(',')]})
+        st.json(run_kwargs)
         run_survey(**run_kwargs)
     
     '''
@@ -112,8 +112,6 @@ def survey_space(surveyor, download_placeholder):
 
 if __name__ == '__main__':
     st.title('Auto-Research V0.1 - Automated Survey generation from research keywords')
-    with st.sidebar.form(key="dummy_form"):
-        dummy_session_data = sp.pydantic_input(key="dummy_model", model=ShowcaseModel)
     std_col, survey_col = st.columns(2)
     std_col.header('execution log:')
     survey_col.header('Generated_survey:')

temp_showcase_model.py

@@ -1,93 +0,0 @@
-import datetime
-from enum import Enum
-from typing import Dict, List, Literal, Optional, Set
-
-import streamlit as st
-from pydantic import BaseModel, Field, SecretStr
-
-import streamlit_pydantic as sp
-from streamlit_pydantic.types import FileContent
-
-
-class SelectionValue(str, Enum):
-    FOO = "foo"
-    BAR = "bar"
-
-
-class OtherData(BaseModel):
-    text: str
-    integer: int
-
-
-class ShowcaseModel(BaseModel):
-    short_text: str = Field(..., max_length=60, description="Short text property")
-    password: SecretStr = Field(..., description="Password text property")
-    long_text: str = Field(
-        ..., format="multi-line", description="Unlimited text property"
-    )
-    integer_in_range: int = Field(
-        20,
-        ge=10,
-        le=30,
-        multiple_of=2,
-        description="Number property with a limited range. Optional because of default value.",
-    )
-    positive_integer: int = Field(
-        ..., ge=0, multiple_of=10, description="Positive integer with step count of 10."
-    )
-    float_number: float = Field(0.001)
-    date: Optional[datetime.date] = Field(
-        datetime.date.today(),
-        description="Date property. Optional because of default value.",
-    )
-    time: Optional[datetime.time] = Field(
-        datetime.datetime.now().time(),
-        description="Time property. Optional because of default value.",
-    )
-    boolean: bool = Field(
-        False,
-        description="Boolean property. Optional because of default value.",
-    )
-    read_only_text: str = Field(
-        "Lorem ipsum dolor sit amet",
-        description="This is a ready only text.",
-        readOnly=True,
-    )
-    file_list: Optional[List[FileContent]] = Field(
-        None,
-        description="A list of files. Optional property.",
-    )
-    single_file: Optional[FileContent] = Field(
-        None,
-        description="A single file. Optional property.",
-    )
-    single_selection: SelectionValue = Field(
-        ..., description="Only select a single item from a set."
-    )
-    single_selection_with_literal: Literal["foo", "bar"] = Field(
-        "foo", description="Only select a single item from a set."
-    )
-    multi_selection: Set[SelectionValue] = Field(
-        ..., description="Allows multiple items from a set."
-    )
-    multi_selection_with_literal: Set[Literal["foo", "bar"]] = Field(
-        ["foo", "bar"], description="Allows multiple items from a set."
-    )
-    single_object: OtherData = Field(
-        ...,
-        description="Another object embedded into this model.",
-    )
-    string_list: List[str] = Field(
-        ..., max_items=20, description="List of string values"
-    )
-    int_list: List[int] = Field(..., description="List of int values")
-    string_dict: Dict[str, str] = Field(
-        ..., description="Dict property with string values"
-    )
-    float_dict: Dict[str, float] = Field(
-        ..., description="Dict property with float values"
-    )
-    object_list: List[OtherData] = Field(
-        ...,
-        description="A list of objects embedded into this model.",
-    )