Update app.py

app.py

@@ -22,6 +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
 
 # Title
 st.title("Legal Assistant")
@@ -66,18 +68,15 @@ def add_to_chat_history(question, answer):
         'answer': answer
     })
 
-# Clear chat function
-def clear_chat():
-    # Clear the last response
-    st.session_state.last_response = None
-    # Optional: You can also clear the specific key used for the text input
-    st.session_state.prompt1 = ""
-
 # Main input area with clear button
 col1, col2 = st.columns([3, 1])
 with col1:
     # Use a key to manage the text input state
-    prompt1 = st.text_input("Enter Your Question From Documents", key="prompt1")
+    # 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', '')
+    prompt1 = st.text_input("Enter Your Question From Documents", 
+                             value=default_text, 
+                             key="user_question")
 
 with col2:
     # Add clear chat button next to the input
@@ -86,12 +85,20 @@ with col2:
 
 # Clear chat functionality
 if clear_button:
-    clear_chat()
+    # Reset flags and last response
+    st.session_state.clear_chat = True
+    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:
     try:
+        # Store the current question
+        st.session_state.last_question = prompt1
+
         # Create document and retrieval chains
         document_chain = create_stuff_documents_chain(llm, prompt)
         retriever = st.session_state.vectors.as_retriever()