Update app.py

app.py

@@ -28,23 +28,23 @@ if st.session_state.stage == "crawl-input":
         st.session_state.stage = "prompt-input"
 else:
     prompt = st.text_area("What would you like to chat about?")
-        prompt_btn = st.button("Enter")
+    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]
+    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()
+        info_text = ""
+        for info in infos:
+            info_text += info + "\n---\n"
+        info_text = info_text.strip()
         
-            prompt = prompt_template.format(info_text,prompt)
+        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}])
+        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])
+        st.write((completion.choices[0].message.content)[7:-1])