Update app.py

app.py

@@ -2,40 +2,68 @@
 import streamlit as st
 from transformers import pipeline
 
-# Initialize the model
-pipe = pipeline("text2text-generation", model="facebook/blenderbot-400M-distill")
+# Caching the model pipeline
+@st.cache_resource
+def load_pipeline():
+    return pipeline("text2text-generation", model="facebook/blenderbot-400M-distill")
 
-# Initialize session state for conversation history
+# Initialize the model once using cache
+pipe = load_pipeline()
+
+# Initialize session state for conversation history and bot response
 if 'conversation_history' not in st.session_state:
     st.session_state.conversation_history = ""
+if 'bot_response' not in st.session_state:
+    st.session_state.bot_response = ""
 
 def converse(user_message):
     # Update the conversation history
     st.session_state.conversation_history += f"User: {user_message}\n"
     result = pipe(st.session_state.conversation_history)[0]['generated_text']
     st.session_state.conversation_history += f"Bot: {result}\n"
+    st.session_state.bot_response = result
     return result
 
-# Streamlit app
-st.title("AI Chatbot")
-st.write("Chat with the AI bot!")
+# Sidebar options
+st.sidebar.title("App Settings")
+show_history = st.sidebar.checkbox("Show conversation history", value=True)
+character_limit = st.sidebar.slider("Set character limit for input:", min_value=50, max_value=500, value=200)
+
+# Session reset button
+if st.sidebar.button("Reset Conversation"):
+    st.session_state.conversation_history = ""
+    st.session_state.bot_response = ""
+    st.sidebar.success("Conversation history cleared.")
 
-# Text input from the user
-user_message = st.text_input("Enter Your message:")
+# Streamlit app layout
+st.title("🤖 AI Chatbot")
+st.subheader("Chat with an AI-powered bot!")
 
-# Checkbox to toggle conversation history display
-show_history = st.checkbox("Show conversation history", value=True)
+# Input field with character limit
+user_message = st.text_input(f"Enter your message (max {character_limit} characters):", max_chars=character_limit)
 
-# Button to trigger response generation
+# Send button to generate bot response
 if st.button("Send"):
     if user_message:
         # Get response from the chatbot
         bot_response = converse(user_message)
         
+        # Display bot's response in a dedicated area
+        st.markdown("### Bot's Response")
+        st.success(bot_response)
+        
         if show_history:
-            # Display conversation history if checkbox is checked
+            # Display conversation history in a text area for better scrolling
             st.write("### Conversation History")
-            st.text(st.session_state.conversation_history)
+            st.text_area("Conversation", value=st.session_state.conversation_history, height=250, max_chars=None)
     else:
         # Show a warning if no message is provided
         st.warning("Please enter a message before sending.")
+
+# About section
+st.markdown("---")
+st.markdown("### About this App")
+st.info("This chatbot is powered by a pre-trained model from the Hugging Face Transformers library. You can chat with the bot, and the conversation history will be maintained during the session.")
+
+st.sidebar.markdown("---")
+st.sidebar.write("Created by [Your Name](https://github.com/yourprofile)")