Update app.py

app.py

@@ -30,7 +30,8 @@ if prompt := st.chat_input("How can I help you today?"):
     with st.chat_message("assistant"):
         message_placeholder = st.empty()
         full_response = ""
-        for response in openai.responses.create(
+        
+        response = openai.responses.create(
             model=st.session_state["openai_model"],
             instructions = grounding,
             input=[
@@ -38,11 +39,14 @@ if prompt := st.chat_input("How can I help you today?"):
                 for m in st.session_state.messages
             ],
             stream=True,
-        ):
-            for chunk in response:
-                if chunk.choices[0].delta.content is not None:
-                    full_response += chunk.choices[0].delta.content.replace('\\$','$').replace('$','\\$')
-                    message_placeholder.markdown(full_response + "▌")
+        )
+        
+        for chunk in response:
+            if chunk.choices[0].delta.content is not None:
+                full_response += chunk.choices[0].delta.content.replace('\\$','$').replace('$','\\$')
+                message_placeholder.markdown(full_response + "▌")
+        
         message_placeholder.markdown(full_response)
         print(full_response)
+        
     st.session_state.messages.append({"role": "assistant", "content": full_response})