Update app.py

app.py

@@ -674,12 +674,11 @@ if prompt :
         # Display user message
         with st.chat_message("user"):
             st.markdown(prompt)
+
+        # Generate AI response
         with st.chat_message("assistant"):
             add_dot_typing_animation()
             display_typing_indicator()
-    
-        # Generate AI response
-        with st.chat_message("assistant"):
             
             full_response = ""