Update app.py

app.py

@@ -4,17 +4,20 @@ import create_database
 import streamlit as st
 
 api_key = "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"
+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 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")
+    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 = "prompt-input"
 
 openai_client = OpenAI(base_url="https://openrouter.ai/api/v1",api_key=api_key)
 
@@ -25,24 +28,25 @@ prompt_template = """Answer the question only according to the information provi
 ## Question : 
 # {}"""
 
-prompt = st.chat_input("What would you like to chat about?")
-prompt_btn = st.button("Enter")
-
-if prompt and prompt_btn:
-    results = st.session_state.collection.query(query_texts=[prompt],n_results=5)
-    infos = results["documents"][0]
-
-    info_text = ""
-    for info in infos:
-        info_text += info + "\n---\n"
-    info_text = info_text.strip()
-
-    prompt = prompt_template.format(info_text,prompt)
-
-    completion = openai_client.chat.completions.create(
-    extra_headers={},
-    extra_body={},
-    model="deepseek/deepseek-r1:free",
-    messages=[{"role":"user","content":prompt}])
+if st.session_state.stage == "prompt-input"
+    prompt = st.chat_input("What would you like to chat about?")
+    prompt_btn = st.button("Enter")
 
-    st.write((completion.choices[0].message.content)[7:-1])
+    if prompt and prompt_btn:
+        results = st.session_state.collection.query(query_texts=[prompt],n_results=5)
+        infos = results["documents"][0]
+    
+        info_text = ""
+        for info in infos:
+            info_text += info + "\n---\n"
+        info_text = info_text.strip()
+    
+        prompt = prompt_template.format(info_text,prompt)
+    
+        completion = openai_client.chat.completions.create(
+        extra_headers={},
+        extra_body={},
+        model="deepseek/deepseek-r1:free",
+        messages=[{"role":"user","content":prompt}])
+    
+        st.write((completion.choices[0].message.content)[7:-1])