Update app.py

app.py

@@ -73,12 +73,21 @@ class ArxivIDsModel(BaseModel):
     )
 
 if __name__ == '__main__':
+    if 'session_count' not in st.session_state:
+        st.session_state.session_count = 0
+        
+    demo_session_limit = 2
+    
+    if st.session_state.session_count > demo_session_limit:
+        st.write(f'{st.session_state.session_count} sessions running, this is a demo and only supports {demo_session_limit} parallel sessions, \n please try in sometime')
+    
     st.sidebar.image(Image.open('logo_landscape.png'), use_column_width = 'always')
     st.title('Auto-Research')
     st.write('#### A no-code utility to generate a detailed well-cited survey with topic clustered sections' 
              '(draft paper format) and other interesting artifacts from a single research query or a curated set of papers(arxiv ids).')
     st.write('##### Data Provider: arXiv Open Archive Initiative OAI')
     st.write('##### GitHub: https://github.com/sidphbot/Auto-Research')
+    st.write(f'Note: this is only a demo on cpu-13GB RAM, hence it supports limited number of papers & only {demo_session_limit} parallel user sessions ({st.session_state.session_count} running currently !!)')
     download_placeholder = st.container()
 
     with st.sidebar.form(key="survey_keywords_form"):
@@ -91,12 +100,16 @@ if __name__ == '__main__':
     run_kwargs = {'surveyor':get_surveyor_instance(_print_fn=st.sidebar.write, _survey_print_fn=st.write),
                   'download_placeholder':download_placeholder}
     if submit:
-        if session_data['research_keywords'] != '':
-            run_kwargs.update({'research_keywords':session_data['research_keywords'], 
-                               'max_search':session_data['max_search'], 
-                               'num_papers':session_data['num_papers']})
-        elif session_data['arxiv_ids'] != '':
-            run_kwargs.update({'arxiv_ids':[id.strip() for id in session_data['arxiv_ids'].split(',')]})
-        print(run_kwargs)
-        run_survey(**run_kwargs)
+        if st.session_state.session_count < demo_session_limit:
+            st.session_state.session_count = st.session_state.session_count + 1
+            if session_data['research_keywords'] != '':
+                run_kwargs.update({'research_keywords':session_data['research_keywords'], 
+                                   'max_search':session_data['max_search'], 
+                                   'num_papers':session_data['num_papers']})
+            elif session_data['arxiv_ids'] != '':
+                run_kwargs.update({'arxiv_ids':[id.strip() for id in session_data['arxiv_ids'].split(',')]})
+            print(run_kwargs)
+            run_survey(**run_kwargs)
+        else:
+            st.write(f'{st.session_state.session_count} sessions running, this is a demo and only supports 2 parallel sessions, \n please try in sometime')