Update app.py

app.py

@@ -2486,10 +2486,10 @@ if problem and problem != st.session_state.last_submitted:
             answer = ask_ai_streaming(problem, sympy_result, st.session_state.messages)
             plot_graph(problem, sympy_result)
 
+    
     # Save to history
+    if not st.session_state.current_chat_id:
+        new_chat()  # create chat ID BEFORE appending messages
     st.session_state.messages.append({"role": "user", "content": problem})
     st.session_state.messages.append({"role": "assistant", "content": answer})
-    # Auto-save to chat history
-    if not st.session_state.current_chat_id:
-        new_chat()
     save_current_chat()