Update app.py

app.py

@@ -112,6 +112,11 @@ st.title("Medical Chatbot")
 if "history" not in st.session_state:
     st.session_state.history = []
 
+# Display chat history using streamlit-chat
+for i, chat in enumerate(st.session_state.history):
+    message(chat['question'], is_user=True, key=f"user_{i}")
+    message(chat['answer'], key=f"bot_{i}")
+
 user_input = st.text_input("Ask a question:", key="input")
 
 if st.button("Submit"):
@@ -119,8 +124,15 @@ if st.button("Submit"):
         result = st.session_state.qa_chain(user_input)
         answer = result["result"]
         st.session_state.history.append({"question": user_input, "answer": answer})
-
-# Display chat history using streamlit-chat
-for i, chat in enumerate(st.session_state.history):
-    message(chat['question'], is_user=True, key=f"user_{i}")
-    message(chat['answer'], key=f"bot_{i}")
+        # Clear input text after submission
+        st.session_state.input = ""
+
+# Scroll to the bottom to display the latest message
+st.markdown(
+    """
+    <script>
+    window.scrollTo(0, document.body.scrollHeight);
+    </script>
+    """,
+    unsafe_allow_html=True,
+)