Update app.py

app.py

@@ -15,7 +15,8 @@ prompt_template = """Answer the question only according to the information provi
 # {}"""
 
 try:
-    st.session_state.stage
+    if st.session_state.stage == "remove-crawl":
+        st.session_state.stage = "prompt-input"
 except Exception:
     st.session_state.stage = "crawl-input"
 
@@ -30,7 +31,7 @@ if st.session_state.stage == "crawl-input":
         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 = "prompt-input"
+        st.session_state.stage = "remove-crawl"
 
 if st.session_state.stage == "prompt-input":
     prompt = st.text_area("What would you like to chat about?")
@@ -53,4 +54,4 @@ if st.session_state.stage == "prompt-input":
         model="deepseek/deepseek-r1:free",
         messages=[{"role":"user","content":prompt}])
         
-        st.write((completion.choices[0].message.content)[7:-1])
+        st.write((completion.choices[0].message.content))