TEST-2

main.py

@@ -26,12 +26,8 @@ st.title("💳 Finance & Banking Chatbot 🤵")
 
 # Display previous messages in the chat area
 for msg in st.session_state.messages:
-    if msg["role"] == "user":
-        with st.chat_message("user", avatar="👦🏻"):
-            st.markdown(f"<div style='text-align: right;'>{msg['content']}</div>", unsafe_allow_html=True)
-    else:
-        with st.chat_message("assistant", avatar="🤵"):
-            st.markdown(f"<div style='text-align: left;'>{msg['content']}</div>", unsafe_allow_html=True)
+    with st.chat_message(msg["role"], avatar=("👦🏻" if msg["role"] == "user" else "🤵")):
+        st.markdown(msg["content"])
 
 # User input field
 user_input = st.chat_input("Ask me about finance, banking, investments, etc. 📈")
@@ -40,7 +36,7 @@ if user_input:
     # Add user message to history
     st.session_state.messages.append({"role": "user", "content": user_input})
     with st.chat_message("user", avatar="👦🏻"):
-        st.markdown(f"<div style='text-align: right;'>{user_input}</div>", unsafe_allow_html=True)
+        st.markdown(user_input)
 
     # Get response from Groq API
     response = client.chat.completions.create(
@@ -54,4 +50,4 @@ if user_input:
     # Add bot response to history
     st.session_state.messages.append({"role": "assistant", "content": bot_reply})
     with st.chat_message("assistant", avatar="🤵"):
-        st.markdown(f"<div style='text-align: left;'>{bot_reply}</div>", unsafe_allow_html=True)
+        st.markdown(bot_reply)