TEST-1

main.py

@@ -7,10 +7,6 @@ client = Groq(api_key="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"
 # Streamlit app UI
 st.set_page_config(page_title="💰 Finance & Banking Chatbot", layout="wide")
 
-# # Sidebar for chat history
-# st.sidebar.title("🗂 Chat History")
-# st.sidebar.write("Review your past finance-related conversations here.")
-
 # System prompt to enforce finance-related responses
 SYSTEM_PROMPT = (
     "You are an expert financial assistant. Your role is to answer ONLY finance-related topics, including banking, investments, "
@@ -25,22 +21,17 @@ if "messages" not in st.session_state:
         {"role": "assistant", "content": "Hello! I can help you with financial questions. How can I assist? 💳"}
     ]
 
-# # Display chat history in the sidebar
-# for msg in st.session_state.messages:
-#     if msg["role"] == "user":
-#         st.sidebar.write(f"👤 **You**: {msg['content']}")
-#     else:
-#         st.sidebar.write(f"🤖 **Bot**: {msg['content']}")
-
 # Main chat interface
 st.title("💳 Finance & Banking Chatbot 🤵")
 
 # Display previous messages in the chat area
 for msg in st.session_state.messages:
     if msg["role"] == "user":
-        st.chat_message("user", avatar="👦🏻").write(msg["content"])
+        with st.chat_message("user", avatar="👦🏻"):
+            st.markdown(f"<div style='text-align: right;'>{msg['content']}</div>", unsafe_allow_html=True)
     else:
-        st.chat_message("assistant", avatar="🤵").write(msg["content"])
+        with st.chat_message("assistant", avatar="🤵"):
+            st.markdown(f"<div style='text-align: left;'>{msg['content']}</div>", unsafe_allow_html=True)
 
 # User input field
 user_input = st.chat_input("Ask me about finance, banking, investments, etc. 📈")
@@ -48,7 +39,8 @@ user_input = st.chat_input("Ask me about finance, banking, investments, etc.
 if user_input:
     # Add user message to history
     st.session_state.messages.append({"role": "user", "content": user_input})
-    st.chat_message("user", avatar="👦🏻").write(user_input)
+    with st.chat_message("user", avatar="👦🏻"):
+        st.markdown(f"<div style='text-align: right;'>{user_input}</div>", unsafe_allow_html=True)
 
     # Get response from Groq API
     response = client.chat.completions.create(
@@ -61,4 +53,5 @@ if user_input:
 
     # Add bot response to history
     st.session_state.messages.append({"role": "assistant", "content": bot_reply})
-    st.chat_message("assistant", avatar="🤵").write(bot_reply)
+    with st.chat_message("assistant", avatar="🤵"):
+        st.markdown(f"<div style='text-align: left;'>{bot_reply}</div>", unsafe_allow_html=True)