Update app.py

app.py

@@ -19,22 +19,18 @@ if not "stage" in st.session_state:
 elif st.session_state.stage == "remove-crawl-input-widgets":
     st.session_state.stage = "prompt-input"
 
-def get_crawl_input():
-    if st.session_state.stage == "crawl-input":
-        config = st.text_area("Enter the start URL and the limit of crawling (in this format : URL,limit):")
-        config_btn = st.button("Start crawling!")
-    
-        if config and config_btn:
-            start = config.split(",")[0]
-            limit = int(config.split(",")[1])
-                
-            txt = crawl_the_site.crawl(start,limit)
-            st.session_state.collection = create_database.create_database(txt)
-            st.write("Crawling is done.")
-            st.session_state.stage = "remove-crawl-input-widgets"
-            return
-
-get_crawl_input()
+if st.session_state.stage == "crawl-input":
+    config = st.text_area("Enter the start URL and the limit of crawling (in this format : URL,limit):")
+    config_btn = st.button("Start crawling!")
+
+    if config and config_btn:
+        start = config.split(",")[0]
+        limit = int(config.split(",")[1])
+            
+        txt = crawl_the_site.crawl(start,limit)
+        st.session_state.collection = create_database.create_database(txt)
+        st.write("Crawling is done.")
+        st.session_state.stage = "remove-crawl-input-widgets"
 
 if st.session_state.stage == "prompt-input":
     prompt = st.text_area("Ask the documentation a question :")