Update app.py

app.py

@@ -5,6 +5,15 @@ import streamlit as st
 
 api_key = "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"
 
+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 : 
+# {}"""
+
 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!")
@@ -17,35 +26,25 @@ if st.session_state.stage == "crawl-input":
         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)
-
-prompt_template = """Answer the question only according to the information provided below. Answer only the user's question, dont give additional information.
-## Information : 
-{}
-
-## Question : 
-# {}"""
-
-if st.session_state.stage == "prompt-input":
+else:
     prompt = st.text_area("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_btn = st.button("Enter")
     
-        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])
+        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])