Update app.py

app.py

@@ -4,12 +4,13 @@ from transformers import pipeline
 # Configure the Hugging Face API key
 HF_API_KEY = st.secrets['huggingface_api_key']
 
-# Initialize the Hugging Face model
-generator = pipeline('text-generation', model='gpt-2', tokenizer='gpt-2', api_key=HF_API_KEY)
+# Initialize the Hugging Face model with the API key
+generator = pipeline('text-generation', model='gpt2', tokenizer='gpt2', use_auth_token=HF_API_KEY)
 
 # Function to get response from the Hugging Face model
 def get_chatbot_response(user_input):
     try:
+        # Generate the response using the Hugging Face model
         response = generator(user_input, max_length=100, num_return_sequences=1)
         return response[0]['generated_text']
     except Exception as e:
@@ -47,7 +48,6 @@ st.markdown("""
         border-radius: 10px 10px 10px 10px;
     }
     .chat-header {
-        # text-align: center;
         font-size: 35px;
         font-weight: bold;
         margin-bottom: 20px;
@@ -60,23 +60,29 @@ st.markdown("""
     </style>
 """, unsafe_allow_html=True)
 
-st.markdown('<div class="chat-header">AI Chatbot-Your Companion 💻</div>', unsafe_allow_html=True)
+# Chat header and intro
+st.markdown('<div class="chat-header">AI Chatbot - Your Companion 💻</div>', unsafe_allow_html=True)
 st.write("Powered by Hugging Face AI for smart, engaging conversations. 🤖")
 
+# Initialize session state for conversation history if not already initialized
 if "history" not in st.session_state:
     st.session_state["history"] = []
 
+# Create the chat form
 with st.form(key="chat_form", clear_on_submit=True):
     user_input = st.text_input("Your message here... ✍️", max_chars=2000, label_visibility="collapsed")
     submit_button = st.form_submit_button("Send 🚀")
 
     if submit_button:
         if user_input:
+            # Get response from the chatbot
             response = get_chatbot_response(user_input)
+            # Store user input and bot response in session state history
             st.session_state.history.append((user_input, response))
         else:
             st.warning("Please Enter A Prompt 😅")
 
+# Display chat history
 if st.session_state["history"]:
     st.markdown('<div class="chat-container">', unsafe_allow_html=True)
     for user_input, response in st.session_state["history"]: