Update app.py

app.py

@@ -9,6 +9,7 @@ openai_client = OpenAI(base_url="https://openrouter.ai/api/v1",api_key=api_key)
 
 prompt_input_area = st.empty()
 output = st.empty()
+prompt_btn_area = st.empty()
 
 prompt_template = """Answer the question only according to the information provided below. Answer only the user's question, dont give additional information.
 ## Information : 
@@ -37,7 +38,7 @@ if st.session_state.stage == "crawl-input":
 
 if st.session_state.stage == "prompt-input":
     prompt = prompt_input_area.text_area("Ask the documentation a question :")
-    prompt_btn = st.button("Enter")
+    prompt_btn = prompt_btn_area.button("Enter")
     
     if prompt and prompt_btn:
         results = st.session_state.collection.query(query_texts=[prompt],n_results=5)