Increased token limit

app.py

@@ -28,6 +28,16 @@ warnings.filterwarnings("ignore")
 
 import re
 
+st.markdown("""
+<style>
+    [data-testid="stChatMessage"] {
+        background: none !important;
+        border: none !important;
+        padding: 0 !important;
+    }
+</style>
+""", unsafe_allow_html=True)
+
 def remove_code_blocks(text):
     # Remove triple backtick code blocks
     text = re.sub(r"```[\s\S]*?```", "", text)
@@ -104,6 +114,7 @@ for message in st.session_state.messages:
         # st.markdown(remove_code_blocks(message["content"]))
         st.markdown(message["content"])
 
+
 # === Chat Interface ===
 if prompt := st.chat_input("Ask a follow-up question..."):
     st.session_state.messages.append({"role": "user", "content": prompt})