Update app.py

app.py

@@ -24,6 +24,8 @@ if "qa_pipeline" not in st.session_state:
     st.session_state["qa_pipeline"] = None
 if "conversation" not in st.session_state:
     st.session_state["conversation"] = []
+if "user_input" not in st.session_state:
+    st.session_state["user_input"] = ""
 
 # Load Model
 if load_model_button:
@@ -58,26 +60,33 @@ with chat_container:
     st.subheader("Conversation")
     for idx, (speaker, message) in enumerate(st.session_state["conversation"]):
         if speaker == "You":
-            st.text_area(f"You ({idx}):", message, key=f"you_{idx}", disabled=False)
+            st.text_area(f"You ({idx}):", message, key=f"you_{idx}", disabled=True)
         else:
-            st.text_area(f"Model ({idx}):", message, key=f"model_{idx}", disabled=False)
+            st.text_area(f"Model ({idx}):", message, key=f"model_{idx}", disabled=True)
 
 # Input Area
 with input_container:
     if st.session_state["qa_pipeline"]:
-        user_input = st.text_input("Enter your query:", key="chat_input", label_visibility="visible")
+        user_input = st.text_input(
+            "Enter your query:",
+            value=st.session_state["user_input"],  # Use session state for persistence
+            key="chat_input",
+            label_visibility="visible",
+            on_change=lambda: st.session_state.update({"user_input": st.session_state.chat_input}),
+        )
         if st.button("Send", key="send_button"):
-            if user_input:
+            if st.session_state["user_input"]:
                 with st.spinner("Generating response..."):
                     try:
-                        response = st.session_state["qa_pipeline"](f"Q: {user_input}", max_length=300)
+                        response = st.session_state["qa_pipeline"](f"Q: {st.session_state['user_input']}", max_length=300)
                         generated_text = response[0]["generated_text"]
-                        st.session_state["conversation"].append(("You", user_input))
+                        st.session_state["conversation"].append(("You", st.session_state["user_input"]))
                         st.session_state["conversation"].append(("Model", generated_text))
+                        st.session_state["user_input"] = ""  # Clear input after submission
                     except Exception as e:
                         st.error(f"Error generating response: {e}")
 
 # Clear Conversation
 if clear_conversation_button:
     st.session_state["conversation"] = []
-    st.success("Conversation cleared.")
+    st.success("Conversation cleared.")