Update app.py

app.py

@@ -5,17 +5,17 @@ import streamlit as st
 
 api_key = "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"
 
-config = st.text_area("enter the start URL and the limit of crawling :")
-
-if config:
-    pass
-
-start = "asdf"
-limit = 100
-
-print("")
-txt = crawl_the_site.crawl(start,limit)
-collection = create_database.create_database(txt)
+config = st.text_area("Enter the start URL and the limit of crawling (in this format : URL,limit):")
+btn = st.button("Start crawling!")
+
+if config and btn:
+    start = config.split(",")[0]
+    limit = int(config.split(",")[1])
+    
+    print("")
+    txt = crawl_the_site.crawl(start,limit)
+    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)
 
@@ -26,10 +26,10 @@ prompt_template = """Answer the question only according to the information provi
 ## Question : 
 # {}"""
 
-while True:
-    q = input("prompt : ")
+prompt = st.text_area("Enter a prompt (after the crawling is done):")
 
-    results = collection.query(query_texts=[q],n_results=5)
+if prompt:
+    results = collection.query(query_texts=[prompt],n_results=5)
     infos = results["documents"][0]
 
     info_text = ""
@@ -37,7 +37,7 @@ while True:
         info_text += info + "\n---\n"
     info_text = info_text.strip()
 
-    prompt = prompt_template.format(info_text,q)
+    prompt = prompt_template.format(info_text,prompt)
 
     completion = openai_client.chat.completions.create(
     extra_headers={},
@@ -45,4 +45,4 @@ while True:
     model="deepseek/deepseek-r1:free",
     messages=[{"role":"user","content":prompt}])
 
-    print((completion.choices[0].message.content)[7:-1])
+    st.write((completion.choices[0].message.content)[7:-1])