Update app.py

app.py

@@ -92,9 +92,14 @@ def load_dataset_into_session():
 # Streamlit app main
 st.title("ChatCSV powered by LLM")
 
+# Ensure session state for dataframe
 if "df" not in st.session_state:
     st.session_state.df = pd.DataFrame()
 
+# Ensure session state for user query
+if "user_query" not in st.session_state:
+    st.session_state.user_query = ""
+
 st.header("Load Your Dataset")
 load_dataset_into_session()
 
@@ -104,13 +109,15 @@ if "df" in st.session_state and not st.session_state.df.empty:
     st.dataframe(st.session_state.df.head(num_rows))
 
     st.subheader("Chat with Your Dataset")
-    user_query = st.text_area("Enter your query:")
-
+    
+    # Text area for user query with session state persistence
+    st.session_state.user_query = st.text_area("Enter your query:", value=st.session_state.user_query)
+    
     if st.button("Run Query"):
-        if user_query.strip():
+        if st.session_state.user_query.strip():
             with st.spinner("Processing your query..."):
                 try:
-                    result = chat_with_csv(st.session_state.df, user_query)
+                    result = chat_with_csv(st.session_state.df, st.session_state.user_query)
                     st.success(result)
                 except Exception as e:
                     st.error(f"Error processing your query: {e}")