Update app.py

app.py

@@ -19,7 +19,7 @@ if not "stage" in st.session_state:
 elif st.session_state.stage == "remove-crawl-input-widgets":
     st.session_state.stage = "prompt-input"
 
-for i in range(1):
+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!")
@@ -32,7 +32,9 @@ for i in range(1):
             st.session_state.collection = create_database.create_database(txt)
             st.write("Crawling is done.")
             st.session_state.stage = "remove-crawl-input-widgets"
-            break
+            return
+
+get_crawl_input()
 
 if st.session_state.stage == "prompt-input":
     prompt = st.text_area("Ask the documentation a question :")