Update app.py

app.py

@@ -1,6 +1,5 @@
 import streamlit as st
 import pandas as pd
-import numpy as np
 import os
 import requests
 import json
@@ -99,6 +98,8 @@ function_map = {
 # --- Conversation memory: Use Streamlit session state
 if "messages" not in st.session_state:
     st.session_state.messages = []
+if "temp_input" not in st.session_state:
+    st.session_state.temp_input = ""
 
 # If CSV is loaded, update the system prompt with current columns
 if df is not None:
@@ -136,10 +137,8 @@ for i, msg in enumerate(st.session_state.messages[1:]):  # Skip system message f
 
 # --- User input box at bottom (like ChatGPT)
 if df is not None:
-    if "user_input" not in st.session_state:
-        st.session_state.user_input = ""
-    user_input = st.text_input("Your message:", key="user_input")
-    send = st.button("Send", key="send_btn")
+    user_input = st.text_input("Your message:", value=st.session_state.temp_input, key="temp_input")
+    send = st.button("Send")
 else:
     user_input = None
     send = False
@@ -207,6 +206,7 @@ if send and user_input and user_input.strip():
         # No function call: Just add model's reply
         st.session_state.messages.append({"role": "assistant", "content": msg["content"]})
 
-    # Clear input after sending
-    st.session_state.user_input = ""
-    # The UI will update on the next rerender (no need for manual rerun)
+    # Clear input after sending (no session state error!)
+    st.session_state.temp_input = ""
+    # The UI will update automatically
+