Update main.py

main.py

@@ -1,41 +1,45 @@
-from openai import OpenAI
-import crawl_the_site
-import create_database
-
-api_key = input("enter your OpenRouter api key : ")
-start = input("enter the documentation link : ")
-limit = int(input("enter the limit for crawling (how many links before stopping) : "))
-
-print("")
-txt = crawl_the_site.crawl(start,limit)
-collection = create_database.create_database(txt)
-
-openai_client = OpenAI(base_url="https://openrouter.ai/api/v1",api_key=api_key)
-
-prompt_template = """Answer the question only according to the information provided below. Answer only the user's question, dont give additional information.
-## Information : 
-{}
-
-## Question : 
-# {}"""
-
-while True:
-    q = input("prompt : ")
-
-    results = collection.query(query_texts=[q],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,q)
-
-    completion = openai_client.chat.completions.create(
-    extra_headers={},
-    extra_body={},
-    model="deepseek/deepseek-r1:free",
-    messages=[{"role":"user","content":prompt}])
-
+from openai import OpenAI
+import crawl_the_site
+import create_database
+import streamlit as st
+
+api_key = "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"
+
+config = st.text_area("enter the start URL and the limit of crawling :")
+
+start = "asdf"
+limit = 100
+
+print("")
+txt = crawl_the_site.crawl(start,limit)
+collection = create_database.create_database(txt)
+
+openai_client = OpenAI(base_url="https://openrouter.ai/api/v1",api_key=api_key)
+
+prompt_template = """Answer the question only according to the information provided below. Answer only the user's question, dont give additional information.
+## Information : 
+{}
+
+## Question : 
+# {}"""
+
+while True:
+    q = input("prompt : ")
+
+    results = collection.query(query_texts=[q],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,q)
+
+    completion = openai_client.chat.completions.create(
+    extra_headers={},
+    extra_body={},
+    model="deepseek/deepseek-r1:free",
+    messages=[{"role":"user","content":prompt}])
+
     print((completion.choices[0].message.content)[7:-1])