Modified chatbot.py

chatbot.py

@@ -7,6 +7,7 @@ import datetime
 import random
 from streamlit_extras.add_vertical_space import add_vertical_space
 from streamlit_extras.let_it_rain import rain
+from streamlit_modal import Modal
 
 # Load BERT Model & Tokenizer
 MODEL_PATH = "Trained Model/chatbot_model"  # Path where your model & tokenizer are saved
@@ -59,6 +60,8 @@ def get_response(intent_label):
 # Initialize session state for chat history
 if "chat_history" not in st.session_state:
     st.session_state.chat_history = []
+if "chat_input" not in st.session_state:
+    st.session_state.chat_input = ""
 
 # Sidebar menu
 with st.sidebar:
@@ -66,15 +69,33 @@ with st.sidebar:
     add_vertical_space(1)
     if st.button("🆕 New Chat"):
         st.session_state.chat_history = []
+        st.session_state.chat_input = ""
         st.rerun()
     if st.button("📜 Chat History"):
+        st.session_state.show_history = True
+    if st.button("ℹ️ About"):
+        st.session_state.show_about = True
+
+# Modals for Chat History and About
+history_modal = Modal("Chat History", key="history_modal")
+about_modal = Modal("About", key="about_modal")
+
+if "show_history" in st.session_state and st.session_state.show_history:
+    with history_modal.container():
         st.subheader("🕰 Chat History")
         for chat in st.session_state.chat_history:
             st.write(chat)
-    if st.button("ℹ️ About"):
+        if st.button("Close", key="close_history"):
+            st.session_state.show_history = False
+            st.rerun()
+
+if "show_about" in st.session_state and st.session_state.show_about:
+    with about_modal.container():
+        st.subheader("ℹ️ About")
         st.info("This is an intent-based chatbot powered by BERT, built using Streamlit with an elegant UI.")
-    add_vertical_space(2)
-    st.caption("🚀 Built with ❤️ by AI Enthusiast")
+        if st.button("Close", key="close_about"):
+            st.session_state.show_about = False
+            st.rerun()
 
 # Chat UI
 st.title("🤖 AI Chatbot")
@@ -88,26 +109,29 @@ rain(
 )
 
 # User input
-user_input = st.text_input("You:", "", key="input")
-if user_input:
-    intent_label = predict_intent(user_input)
+st.session_state.chat_input = st.text_input("You:", st.session_state.chat_input, key="input")
+if st.session_state.chat_input:
+    intent_label = predict_intent(st.session_state.chat_input)
     response = get_response(intent_label)
     chat_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-    st.session_state.chat_history.append(f"[{chat_time}] You: {user_input}")
+    st.session_state.chat_history.append(f"[{chat_time}] You: {st.session_state.chat_input}")
     st.session_state.chat_history.append(f"[{chat_time}] Bot: {response}")
 
     # Display conversation
     with st.chat_message("user"):
-        st.markdown(f"**You:** {user_input}")
+        st.markdown(f"**You:** {st.session_state.chat_input}")
         time.sleep(0.5)
     with st.chat_message("assistant"):
         st.markdown(f"**Bot:** {response}")
 
+    # Clear input box after sending
+    st.session_state.chat_input = ""
+
 # Style customization
 st.markdown("""
 <style>
     .stChatMessage {padding: 10px; border-radius: 10px; background-color: #000000;}
-    .stChatMessageUser {background-color: #A9A9A9;}
-    .stChatMessageAssistant {background-color: #A9A9A9;}
+    .stChatMessageUser {background-color: #000000;}
+    .stChatMessageAssistant {background-color: #000000;}
 </style>
-""", unsafe_allow_html=True)
+""", unsafe_allow_html=True)