Increased token limit

app.py

@@ -100,22 +100,23 @@ 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:
-    role = "You" if message["role"] == "user" else "assistant"
-    st.markdown(f"**{role}:** {message['content']}")
+    with st.chat_message(message["role"]):
+        # st.markdown(remove_code_blocks(message["content"]))
+        # st.markdown(message["content"])
+        st.text(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}")
+    # st.markdown(f"**You:** {prompt}")
 
     with st.chat_message("assistant"):
         with st.spinner("Thinking..."):
@@ -134,8 +135,8 @@ if prompt := st.chat_input("Ask a follow-up question..."):
                 result = classifier.predict(image, user_input=prompt, reuse_embeddings=False)
 
             # st.markdown(remove_code_blocks(result["diagnosis"]))
-            st.markdown(result["diagnosis"])
-
+            # st.markdown(result["diagnosis"])
+            st.text(result["diagnosis"])
             st.session_state.messages.append({"role": "assistant", "content": result["diagnosis"]})
 
 if st.session_state.messages and st.button("📄 Download Chat as PDF"):