Update app.py

app.py

@@ -137,6 +137,10 @@ if "uploaded_file_name" not in st.session_state:
 if "uploader_key" not in st.session_state:
     st.session_state.uploader_key = 0
 
+# Initialize question key to track input state
+if "question_submitted" not in st.session_state:
+    st.session_state.question_submitted = ""
+
 # ---------------- FUNCTIONS ----------------
 def clear_chat_history():
     st.session_state.chat = []
@@ -279,23 +283,25 @@ if uploaded and uploaded.name != st.session_state.uploaded_file_name:
     
     st.rerun()
 
-# Chat Input
+# Chat Input with form for better control
 disabled_input = st.session_state.uploaded_file_name is None or client is None
-question = st.text_input(
-    "Ask a question about the loaded PDF:",
-    key="question_input",
-    disabled=disabled_input,
-    label_visibility="collapsed",
-    placeholder="Type your question here..."
-)
 
-col1, col2 = st.columns([6, 1])
-with col1:
-    pass  # Empty column for spacing
-with col2:
-    send_clicked = st.button("Send", disabled=disabled_input, use_container_width=True)
+# Use a form to handle input submission and clearing
+with st.form(key="chat_form", clear_on_submit=True):
+    col1, col2 = st.columns([6, 1])
+    with col1:
+        question = st.text_input(
+            "Ask a question about the loaded PDF:",
+            key="question_input",
+            disabled=disabled_input,
+            label_visibility="collapsed",
+            placeholder="Type your question here..."
+        )
+    with col2:
+        submit_button = st.form_submit_button("Send", disabled=disabled_input, use_container_width=True)
 
-if send_clicked and question:
+# Handle form submission
+if submit_button and question:
     # Add user query to chat history
     st.session_state.chat.append(("user", question))
     
@@ -309,8 +315,7 @@ if send_clicked and question:
         else:
             st.session_state.chat.append(("bot", f"🔴 **Error:** {error}"))
     
-    # Clear the input field
-    st.session_state.question_input = ""
+    # The form already clears the input (clear_on_submit=True)
     st.rerun()
 
 # Display Chat History in scrollable container