session state

app.py

@@ -15,15 +15,14 @@ os.environ["LANGCHAIN_ENDPOINT"]="https://api.smith.langchain.com"
 os.environ["LANGCHAIN_API_KEY"]=st.secrets["LANGCHAIN_API_KEY"]
 os.environ["LANGCHAIN_PROJECT"]=st.secrets["LANGCHAIN_PROJECT"]
 
-st.session_state['session_started'] = False
-
 @traceable(run_type="tool")
 def start_session(session_started):
+    st.session_state['session_started'] = session_started
     return session_started
 
 # change session_started to True
-if not st.session_state['session_started']:
-    st.session_state['session_started'] = start_session(True)
+if 'session_started' not in st.session_state:
+    start_session(True)
 
 st.write("This app helps you to extract data from HTML code using web scraping. It uses GPT-3.5-turbo to generate the code for you. \n *Contribute to this project on [GitHub](https://github.com/CognitiveLabs/GPT-auto-webscraping)*")