Update app.py

app.py

@@ -76,30 +76,27 @@ st.markdown(
 st.title("Fastest AI Chatbot")
 st.write("Ask a question and get a response.")
 
+# Function to display chat history
+def display_chat_history():
+    for chat in st.session_state.chat_history:
+        if chat["role"] == "user":
+            st.markdown(f"**You:** {chat['content']}")
+        elif chat["role"] == "assistant":
+            st.markdown(f"**AI:** {chat['content']}")
+
 # Display chat history
-for chat in st.session_state.chat_history:
-    if chat["role"] == "user":
-        st.markdown(f"**You:** {chat['content']}")
-    elif chat["role"] == "assistant":
-        st.markdown(f"**AI:** {chat['content']}")
+display_chat_history()
 
 # Text input for user's question
-user_input = st.text_input("Enter your question here:")
+user_input = st.text_input("Enter your question here:", key="input")
 
 # Button to trigger response
 if st.button("Get Response"):
     if user_input:
         # Fetch and display response
         response = fetch_response(user_input)
-        st.session_state.chat_history.append({"role": "user", "content": user_input})
-        st.session_state.chat_history.append({"role": "assistant", "content": response})
-        
-        # Display new chat history without refreshing
         st.experimental_rerun()
 
-# Display input field at the end
-st.text_input("Enter your question here:", key="input")
-
 # Footer
 st.markdown(
     """