Increased token limit

app.py

@@ -28,15 +28,6 @@ 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
@@ -109,17 +100,22 @@ if uploaded_file is not None and uploaded_file != st.session_state.current_image
         print("result in app : ", result["diagnosis"])
     st.session_state.messages.append({"role": "assistant", "content": result["diagnosis"]})
 
-for message in st.session_state.messages:
-    with st.chat_message(message["role"]):
-        # st.markdown(remove_code_blocks(message["content"]))
-        st.markdown(message["content"])
+# for message in st.session_state.messages:
+#     with st.chat_message(message["role"]):
+#         # st.markdown(remove_code_blocks(message["content"]))
+#         st.markdown(message["content"])
 
+for message in st.session_state.messages:
+    role = "You" if message["role"] == "user" else "assistant"
+    st.markdown(f"**{role}:** {message['content']}")
 
 # === Chat Interface ===
 if prompt := st.chat_input("Ask a follow-up question..."):
     st.session_state.messages.append({"role": "user", "content": prompt})
-    with st.chat_message("user"):
-        st.markdown(remove_code_blocks(prompt))
+    # with st.chat_message("user"):
+    #     st.markdown(remove_code_blocks(prompt))
+
+    st.markdown(f"**You:** {prompt}")
 
     with st.chat_message("assistant"):
         with st.spinner("Thinking..."):
@@ -139,6 +135,7 @@ if prompt := st.chat_input("Ask a follow-up question..."):
 
             # st.markdown(remove_code_blocks(result["diagnosis"]))
             st.markdown(result["diagnosis"])
+
             st.session_state.messages.append({"role": "assistant", "content": result["diagnosis"]})
 
 if st.session_state.messages and st.button("📄 Download Chat as PDF"):