Restore working user message display from previous version

- Reverted to the working user message flow from commit bc17742
- User messages should now appear in blue immediately when typed
- Removed problematic immediate display logic that was causing delays
- Fixed chat history display to show all messages properly

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <[email protected]>

app.py

@@ -529,13 +529,8 @@ def show_custom_response(response):
 
 
 # Chat history
-# Display chat history (skip last user message if currently processing to avoid duplication)
+# Display chat history
 for response_id, response in enumerate(st.session_state.responses):
-    # Skip showing the last user message if we're processing (already shown above)
-    if (st.session_state.get("processing") and 
-        response_id == len(st.session_state.responses) - 1 and 
-        response["role"] == "user"):
-        continue
     status = show_custom_response(response)
     
     # Show feedback section for assistant responses
@@ -634,15 +629,6 @@ if prompt and not st.session_state.get("processing"):
             prompt = None
 
     if prompt:
-        # Show user message immediately with blue styling
-        st.markdown(f"""
-        <div style='display: flex; justify-content: flex-end; margin: 1rem 0;'>
-            <div class='user-message'>
-                {prompt}
-            </div>
-        </div>
-        """, unsafe_allow_html=True)
-        
         # Add user input to chat history
         user_response = get_from_user(prompt)
         st.session_state.responses.append(user_response)
@@ -651,6 +637,9 @@ if prompt and not st.session_state.get("processing"):
         st.session_state.processing = True
         st.session_state.current_model = model_name
         st.session_state.current_question = prompt
+        
+        # Rerun to show processing indicator
+        st.rerun()
 
 # Process the question if we're in processing state
 if st.session_state.get("processing"):

new_system_prompt.txt

@@ -34,9 +34,10 @@ DATA SAFETY:
 
 PLOTTING REQUIREMENTS:
 - Create plots for visualization requests: fig, ax = plt.subplots(figsize=(12, 7))
-- Display plots directly: st.pyplot(fig); plt.close()
+- Display plots directly with Streamlit: st.pyplot(fig); plt.close()
 - Store success message: answer = "Plot displayed successfully"
 - For non-plots: answer = "text result"
+- NEVER use plt.savefig() - always use st.pyplot(fig) for direct display
 
 BASIC ERROR PREVENTION:
 - Use try/except for complex operations

src.py

@@ -268,6 +268,7 @@ def ask_question(model_name, question):
 import pandas as pd
 import matplotlib.pyplot as plt
 import seaborn as sns
+import streamlit as st
 import uuid
 import calendar
 import numpy as np