Create app.py

app.py

@@ -0,0 +1,42 @@
+import streamlit as st
+from config import setup_environment
+from rag_initializer import initialize_rag_chain
+from chatbot import chatbot_response
+
+setup_environment()
+
+st.set_page_config(page_title="Mini AI Bot for Englishfirm.com", page_icon="🤖", layout="wide")
+
+def main():
+    st.title("Mini AI Bot for Englishfirm.com")
+    st.sidebar.info("Englishfirm is the one of the best PTE coaching academies in Sydney.  Among 52 PTE institutes in Sydney, Englishfirm is the only training centre in Sydney that offers 100% one-on-one coaching. Englishfirm has 2 branches in Sydney, operating 7 day a week. We operate from Sydney CBD campus (Pitt Street) and Parramatta.")
+    st.sidebar.warning("Disclaimer: This chatbot provides information related to Englishfirm's IELTS, PTE, and Spoken English coaching services. While we strive to offer accurate and up-to-date information, this should not be considered a substitute for professional educational advice. For personalized guidance and assistance, please contact our experts directly.")
+    rag_chain = initialize_rag_chain()
+    
+    if rag_chain is None:
+        st.error("Failed to initialize the chatbot. Please try again later.")
+        return
+    
+    st.write("Welcome to Englishfirm! I'm here to assist you with any questions regarding our IELTS, PTE, and Spoken English coaching. How can I help you today?")
+    if "messages" not in st.session_state:
+        st.session_state.messages = []
+
+    for message in st.session_state.messages:
+        with st.chat_message(message["role"]):
+            st.markdown(message["content"])
+
+    if prompt := st.chat_input("Your question:"):
+        st.session_state.messages.append({"role": "user", "content": prompt})
+        with st.chat_message("user"):
+            st.markdown(prompt)
+
+        with st.chat_message("assistant"):
+            with st.spinner("AI is thinking..."):
+                response = chatbot_response(prompt, rag_chain)
+            st.markdown(response)
+        
+        st.session_state.messages.append({"role": "assistant", "content": response})
+    
+    
+if __name__ == "__main__":
+    main()