Update app.py

app.py

@@ -79,11 +79,10 @@ if prompt := st.chat_input():
     st.session_state["messages"].append({"role": "user", "content": prompt})
     st.chat_message("user").write(prompt)
 
-    # Display a placeholder for the assistant's response
-    assistant_message_placeholder = st.chat_message("assistant").empty()
-
     with st.spinner("Translating Dubs language (Woof Woof!)"):
-        assistant_message_placeholder.write_stream(call_api(prompt))  # Update incrementally
+        assistant_message_placeholder = st.chat_message("assistant").empty()
+        response_stream = call_api_stream(prompt)
+        assistant_message_placeholder.write_stream(response_stream)
 
     # Append the final assistant's response to the chat history
     st.session_state["messages"].append({"role": "assistant", "content": bot_response})