Update src/streamlit_app.py

src/streamlit_app.py

@@ -66,13 +66,15 @@ def main():
     st.title("💬 ERNIE Chat - Hugging Face")
 
     init_session()
-    display_chat()
 
     user_prompt = st.chat_input("Say something...")
     if user_prompt:
-        # Append user message to session state
+        # Append user message BEFORE display_chat so it renders right away
         st.session_state.messages.append({"role": "user", "content": user_prompt})
 
+    display_chat()
+
+    if user_prompt:
         # Stream and collect assistant response
         model_name = "baidu/ERNIE-4.5-0.3B-PT"
         assistant_reply = stream_response(model_name, st.session_state.messages)
@@ -80,6 +82,5 @@ def main():
         # Append assistant message to session state
         st.session_state.messages.append({"role": "assistant", "content": assistant_reply})
 
-
 if __name__ == "__main__":
     main()