Update app.py

app.py

@@ -42,12 +42,12 @@ if prompt := st.chat_input("What is up?"):
             max_tokens=1024,
             stream=True,
         )
-        response = ""
+        response_chunks = []
         for chunk in stream:
             if chunk.choices[0].delta.content is not None:
-                chunk_text = chunk.choices[0].delta.content
-                response += chunk_text
-                st.write(chunk_text, end="")
+                response_chunks.append(chunk.choices[0].delta.content)
+        response = "".join(response_chunks)
+        st.markdown(response)
 
     # Store the assistant response in the session state
     st.session_state.messages.append({"role": "assistant", "content": response})