Update app.py

app.py

@@ -14,25 +14,26 @@ prompt_template = """Answer the question only according to the information provi
 ## Question : 
 # {}"""
 
-try:
-    if st.session_state.stage == "remove-crawl":
-        st.session_state.stage = "prompt-input"
-except Exception:
+if not "stage" in st.session_state:
     st.session_state.stage = "crawl-input"
+elif st.session_state.stage == "remove-crawl-input-widgets":
+    st.session_state.stage = "prompt-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!")
-
-    st.write("im here")
-    if not (st.session_state.stage == "remove-crawl") and 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"
+while True:
+    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!")
+    
+        st.write("im here")
+        if not (st.session_state.stage == "remove-crawl") and 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"
+            break
 
 if st.session_state.stage == "prompt-input":
     prompt = st.text_area("Ask the documentation a question :")