Update app.py

app.py

@@ -3,16 +3,20 @@ import crawl_the_site
 import create_database
 import streamlit as st
 
+st.session_state.collection = None
+
 api_key = "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"
 
 config = st.text_area("Enter the start URL and the limit of crawling (in this format : URL,limit):")
 config_btn = st.button("Start crawling!")
 
-start = "https://sern.dev/v4/cli/about/"
-limit = 10
-
-txt = crawl_the_site.crawl(start,limit)
-collection = create_database.create_database(txt)
+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.")
 
 openai_client = OpenAI(base_url="https://openrouter.ai/api/v1",api_key=api_key)
 
@@ -27,7 +31,7 @@ prompt = st.text_area("Enter a prompt (after the crawling is done):")
 prompt_btn = st.button("Enter")
 
 if prompt and prompt_btn:
-    results = collection.query(query_texts=[prompt],n_results=5)
+    results = st.session_state.collection.query(query_texts=[prompt],n_results=5)
     infos = results["documents"][0]
 
     info_text = ""