Update app.py

app.py

@@ -56,10 +56,11 @@ if st.session_state.stage == "prompt-input":
         model="deepseek/deepseek-r1:free",
         messages=[{"role":"user","content":prompt}])
 
-        full_response = ""
-        message_placeholder = st.empty()
-        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)
+        with st.chat_message("assistant"):
+            full_response = ""
+            message_placeholder = st.empty()
+            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)