Update app.py

app.py

@@ -7,10 +7,6 @@ api_key = "sk-or-v1-5b41e7106feb9b982d4ef5a6aa0959993387ba2e5fc9830df1279418776e
 
 openai_client = OpenAI(base_url="https://openrouter.ai/api/v1",api_key=api_key)
 
-prompt_input_area = st.empty()
-prompt_btn_area = st.empty()
-output = 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 : 
 {}
@@ -38,10 +34,11 @@ if st.session_state.stage == "crawl-input":
         st.session_state.stage = "remove-crawl-input-widgets"
 
 if st.session_state.stage == "prompt-input":
-    prompt = prompt_input_area.text_area("Ask the documentation a question :")
-    prompt_btn = prompt_btn_area.button("Enter")
+    prompt = st.chat_input("Ask the documentation a question!")
     
-    if prompt and prompt_btn:
+    if prompt:
+        with st.chat_message("user"):
+            st.markdown(prompt)
         results = st.session_state.collection.query(query_texts=[prompt],n_results=5)
         infos = results["documents"][0]
         
@@ -57,5 +54,9 @@ if st.session_state.stage == "prompt-input":
         extra_body={},
         model="deepseek/deepseek-r1:free",
         messages=[{"role":"user","content":prompt}])
-        
-        output.write((completion.choices[0].message.content))
+
+        for chunk in completion.choices[0].message.content.split():
+            full_response += chunk + " "
+            time.sleep(0.05)
+            message_placeholder.markdown(full_response + "▌")
+        message_placeholder.markdown(full_response)