Update app.py

app.py

@@ -63,46 +63,29 @@ st.sidebar.title("Navigation")
 st.sidebar.write("Reclaim Your Mental Health")
 st.sidebar.markdown("[Visit us at](https://www.insighttherapysolutions.com/)")
 
-#rag_chain = RemoteRunnable("http://69.61.24.171:8000/rag_chain/")
-
-
-
-msgs = StreamlitChatMessageHistory(key="langchain_messages")
-
-# --- Main Content ---
-st.markdown("## 🔍 Chatbot For AI Engineer test:")
-
-if len(msgs.messages) == 0:
-    msgs.add_ai_message("Hi! How can I assist you today?")
-
-
-for msg in msgs.messages:
-    st.chat_message(msg.type).write(msg.content)
-
-if prompt := st.chat_input():
-    st.chat_message("human").write(prompt)
-
-    with st.chat_message("assistant"):
-        message_placeholder = st.empty()
-        full_response = ""
-
-        try:
-            _chat_history = st.session_state.langchain_messages[1:40]
-            _chat_history_tranform = list(
-                chunked([msg.content for msg in _chat_history], n=2)
-            )
-
-            response = rag_chain.stream(
-                {"question": prompt, "chat_history": _chat_history_tranform}
-            )
-
-            for res in response:
-                full_response += res or ""
-                message_placeholder.markdown(full_response + "|")
-                message_placeholder.markdown(full_response)
-
-            msgs.add_user_message(prompt)
-            msgs.add_ai_message(full_response)
-
-        except Exception as e:
-            st.error(f"An error occured. {e}")
+# Add some custom styling
+st.markdown(
+    """
+    <style>
+    .css-18e3th9 {
+        padding-top: 3rem;
+    }
+    .css-1d391kg {
+        text-align: center;
+    }
+    .stButton>button {
+        background-color: #4CAF50;
+        color: white;
+        border: none;
+        padding: 15px 32px;
+        text-align: center;
+        text-decoration: none;
+        display: inline-block;
+        font-size: 16px;
+        margin: 4px 2px;
+        cursor: pointer;
+        border-radius: 8px;
+    }
+    </style>
+    """, unsafe_allow_html=True
+)