Update app.py

app.py

@@ -22,8 +22,8 @@ if 'chat_history' not in st.session_state:
     st.session_state.chat_history = []
 if 'last_response' not in st.session_state:
     st.session_state.last_response = None
-if 'clear_chat' not in st.session_state:
-    st.session_state.clear_chat = False
+if 'last_question' not in st.session_state:
+    st.session_state.last_question = ""
 
 # Title
 st.title("Legal Assistant")
@@ -71,11 +71,9 @@ def add_to_chat_history(question, answer):
 # Main input area with clear button
 col1, col2 = st.columns([3, 1])
 with col1:
-    # Use a key to manage the text input state
-    # If clear_chat is True, use an empty default value
-    default_text = "" if st.session_state.clear_chat else st.session_state.get('last_question', '')
+    # Text input with dynamic value management
     prompt1 = st.text_input("Enter Your Question From Documents", 
-                             value=default_text, 
+                             value=st.session_state.last_question, 
                              key="user_question")
 
 with col2:
@@ -85,13 +83,9 @@ with col2:
 
 # Clear chat functionality
 if clear_button:
-    # Reset flags and last response
-    st.session_state.clear_chat = True
+    # Reset last question and response
+    st.session_state.last_question = ""
     st.session_state.last_response = None
-    st.experimental_rerun()
-
-# Reset clear_chat flag after processing
-st.session_state.clear_chat = False
 
 # Process question and generate response
 if prompt1: