Update app.py

app.py

@@ -52,20 +52,30 @@ if user_input:
 
     # Prepare to get the response from Physics Master
     st.write('Physics Master is thinking...')
+
+    # Initialize an empty string to accumulate the response
+    full_response = ""
+
+    # Fetch response tokens and accumulate them
     response = llm.create_chat_completion(
         messages=st.session_state.messages,
         stream=True
     )
 
-    # Handle the response from the model and append it to the session state
     for chunk in response:
         delta = chunk['choices'][0]['delta']
         if 'role' in delta:
             st.session_state.messages.append({'role': delta['role'], 'content': ''})
         elif 'content' in delta:
             token = delta['content']
-            st.session_state.messages[-1]['content'] += token
-            st.write(token, end="", flush=True)
+            # Accumulate tokens into the full response
+            full_response += token
+
+    # Once the full response is received, append it to the chat history
+    st.session_state.messages[-1]['content'] = full_response
+
+    # Display the full response as a paragraph
+    st.write(f"**Physics Master:** {full_response}")
 
     # Reset user input box
     st.session_state.user_input = ""